X-Git-Url: https://code.grnet.gr/git/snf-image-creator/blobdiff_plain/979096ddfddb0cc13151c7e223be64e80ec8db44..6ca928ac8e5d26fe49b22ed3b7406d240193b9ad:/image_creator/os_type/__init__.py diff --git a/image_creator/os_type/__init__.py b/image_creator/os_type/__init__.py index 17ca22e..c0fa505 100644 --- a/image_creator/os_type/__init__.py +++ b/image_creator/os_type/__init__.py @@ -31,11 +31,27 @@ # interpreted as representing official policies, either expressed # or implied, of GRNET S.A. -from image_creator.util import output +from image_creator.util import FatalError +import textwrap import re +def os_cls(distro, osfamily): + module = None + classname = None + try: + module = __import__("image_creator.os_type.%s" % distro, + fromlist=['image_creator.os_type']) + classname = distro.capitalize() + except ImportError: + module = __import__("image_creator.os_type.%s" % osfamily, + fromlist=['image_creator.os_type']) + classname = osfamily.capitalize() + + return getattr(module, classname) + + def add_prefix(target): def wrapper(self, *args): prefix = args[0] @@ -43,11 +59,98 @@ def add_prefix(target): return wrapper +def sysprep(enabled=True): + def wrapper(func): + func.sysprep = True + func.enabled = enabled + return func + return wrapper + + class OSBase(object): """Basic operating system class""" - def __init__(self, rootdev, ghandler): + + def __init__(self, rootdev, ghandler, output): self.root = rootdev self.g = ghandler + self.out = output + + # Collect metadata about the OS + self.meta = {} + self.meta['ROOT_PARTITION'] = "%d" % self.g.part_to_partnum(self.root) + self.meta['OSFAMILY'] = self.g.inspect_get_type(self.root) + self.meta['OS'] = self.g.inspect_get_distro(self.root) + self.meta['DESCRIPTION'] = self.g.inspect_get_product_name(self.root) + + def _is_sysprep(self, obj): + return getattr(obj, 'sysprep', False) and callable(obj) + + def list_syspreps(self): + + objs = [getattr(self, name) for name in dir(self) + if not name.startswith('_')] + + return [x for x in objs if self._is_sysprep(x)] + + def sysprep_info(self, obj): + assert self._is_sysprep(obj), "Object is not a sysprep" + + return (obj.__name__.replace('_', '-'), textwrap.dedent(obj.__doc__)) + + def get_sysprep_by_name(self, name): + """Returns the sysprep object with the given name""" + error_msg = "Syprep operation %s does not exist for %s" % \ + (name, self.__class__.__name__) + + method_name = name.replace('-', '_') + method = None + try: + method = getattr(self, method_name) + except AttributeError: + raise FatalError(error_msg) + + if not self._is_sysprep(method): + raise FatalError(error_msg) + + return method + + def enable_sysprep(self, obj): + """Enable a system preparation operation""" + setattr(obj.im_func, 'enabled', True) + + def disable_sysprep(self, obj): + """Disable a system preparation operation""" + setattr(obj.im_func, 'enabled', False) + + def print_syspreps(self): + """Print enabled and disabled system preparation operations.""" + + syspreps = self.list_syspreps() + enabled = filter(lambda x: x.enabled, syspreps) + disabled = filter(lambda x: not x.enabled, syspreps) + + wrapper = textwrap.TextWrapper() + wrapper.subsequent_indent = '\t' + wrapper.initial_indent = '\t' + wrapper.width = 72 + + self.out.output("Enabled system preparation operations:") + if len(enabled) == 0: + self.out.output("(none)") + else: + for sysprep in enabled: + name = sysprep.__name__.replace('_', '-') + descr = wrapper.fill(textwrap.dedent(sysprep.__doc__)) + self.out.output(' %s:\n%s\n' % (name, descr)) + + self.out.output("Disabled system preparation operations:") + if len(disabled) == 0: + self.out.output("(none)") + else: + for sysprep in disabled: + name = sysprep.__name__.replace('_', '-') + descr = wrapper.fill(textwrap.dedent(sysprep.__doc__)) + self.out.output(' %s:\n%s\n' % (name, descr)) @add_prefix def ls(self, directory): @@ -101,46 +204,20 @@ class OSBase(object): if has_ftype(f, ftype): action(full_path) - def get_metadata(self): - """Returns some descriptive metadata about the OS.""" - meta = {} - meta['ROOT_PARTITION'] = "%d" % self.g.part_to_partnum(self.root) - meta['OSFAMILY'] = self.g.inspect_get_type(self.root) - meta['OS'] = self.g.inspect_get_distro(self.root) - meta['DESCRIPTION'] = self.g.inspect_get_product_name(self.root) - - return meta - - def data_cleanup(self): - """Cleanup sensitive data out of the OS image.""" - - output('Cleaning up sensitive data out of the OS image:') - - is_cleanup = lambda x: x.startswith('data_cleanup_') and \ - callable(getattr(self, x)) - tasks = [getattr(self, x) for x in dir(self) if is_cleanup(x)] - size = len(tasks) - cnt = 0 - for task in tasks: - cnt += 1 - output(('(%d/%d)' % (cnt, size)).ljust(7), False) - task() - output() - - def sysprep(self): + def do_sysprep(self): """Prepere system for image creation.""" - output('Preparing system for image creation:') + self.out.output('Preparing system for image creation:') + + tasks = self.list_syspreps() + enabled = filter(lambda x: x.enabled, tasks) - is_sysprep = lambda x: x.startswith('sysprep_') and \ - callable(getattr(self, x)) - tasks = [getattr(self, x) for x in dir(self) if is_sysprep(x)] - size = len(tasks) + size = len(enabled) cnt = 0 - for task in tasks: + for task in enabled: cnt += 1 - output(('(%d/%d)' % (cnt, size)).ljust(7), False) + self.out.output(('(%d/%d)' % (cnt, size)).ljust(7), False) task() - output() + self.out.output() # vim: set sta sts=4 shiftwidth=4 sw=4 et ai :