Revision e7cbfb0a image_creator/os_type/__init__.py
| b/image_creator/os_type/__init__.py | ||
|---|---|---|
| 97 | 97 |
|
| 98 | 98 |
return (obj.__name__.replace('_', '-'), textwrap.dedent(obj.__doc__))
|
| 99 | 99 |
|
| 100 |
def _sysprep_change_status(self, name, status):
|
|
| 101 |
|
|
| 100 |
def get_sysprep_by_name(self, name):
|
|
| 101 |
"""Returns the sysprep object with the given name""" |
|
| 102 | 102 |
error_msg = "Syprep operation %s does not exist for %s" % \ |
| 103 |
(name, self.__class__.__name__) |
|
| 103 |
(name, self.__class__.__name__)
|
|
| 104 | 104 |
|
| 105 | 105 |
method_name = name.replace('-', '_')
|
| 106 | 106 |
method = None |
| ... | ... | |
| 112 | 112 |
if not self._is_sysprep(method): |
| 113 | 113 |
raise FatalError(error_msg) |
| 114 | 114 |
|
| 115 |
setattr(method.im_func, 'enabled', status)
|
|
| 115 |
return method
|
|
| 116 | 116 |
|
| 117 |
def enable_sysprep(self, name):
|
|
| 117 |
def enable_sysprep(self, obj):
|
|
| 118 | 118 |
"""Enable a system preperation operation""" |
| 119 |
self._sysprep_change_status(name, True)
|
|
| 119 |
setattr(obj.im_func, 'enabled', True)
|
|
| 120 | 120 |
|
| 121 |
def disable_sysprep(self, name):
|
|
| 121 |
def disable_sysprep(self, obj):
|
|
| 122 | 122 |
"""Disable a system preperation operation""" |
| 123 |
self._sysprep_change_status(name, False)
|
|
| 123 |
setattr(obj.im_func, 'enabled', False)
|
|
| 124 | 124 |
|
| 125 | 125 |
def print_syspreps(self): |
| 126 | 126 |
"""Print enabled and disabled system preperation operations.""" |
Also available in: Unified diff