Synnefo » snf-image-creator

import image_creator.os_type

def get_os_class(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)

# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :

        self.ostype = self.g.inspect_get_type(self.root)

        self.distro = self.g.inspect_get_distro(self.root)

        self.g.close()

from image_creator import get_os_class

        osclass = get_os_class(dev.distro, dev.ostype)

        image_os = osclass(dev.root, dev.g)

        metadata = image_os.get_metadata()

#!/usr/bin/env python

class OSBase(object):

    def __init__(self, rootdev, ghandler):

        self.root = rootdev

        self.g = ghandler

    def get_metadata(self):

        meta = {}

        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 mount_all(self):

        mps = g.inspect_get_mountpoints(self.root)

        # Sort the keys to mount the fs in a correct order.

        # / should be mounted befor /boot, etc

        def compare (a, b):

            if len(a[0]) > len(b[0]): return 1

            elif len(a[0]) == len(b[0]): return 0

            else: return -1

        mps.sort(compare)

        for mp, dev in mps:

            try:

                self.g.mount(dev, mp)

            except RuntimeError as msg:

                print "%s (ignored)" % msg

    def cleanup_sensitive_data(self):

        raise NotImplementedError

# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :

from image_creator.os_type.unix import Unix

class Freebsd(Unix):

	pass

# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :

from image_creator.os_type.unix import Unix

class Hard(Unix):

	pass

# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :

from image_creator.os_type.unix import Unix

class Linux(Unix):

	pass

# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :

from image_creator.os_type.unix import Unix

class Netbsd(Unix):

	pass

# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :

#!/usr/bin/env python

import re

from image_creator.os_type import OSBase

class Unix(OSBase):

    def get_metadata(self):

        meta = super(Unix, self).get_metadata()

        meta["USERS"] = " ".join(self.get_passworded_users())

        return meta

    def get_passworded_users(self):

        users = []

        regexp = re.compile('(\S+):((?:!\S+)|(?:[^!*]\S+)|):(?:\S*:){6}')

        for line in open('/etc/shadow', 'r').readlines():

            match = regexp.match(line)

            if not match:

                continue

            user, passwd = match.groups()

            if len(passwd) > 0 and passwd[0] == '!':

                print "Warning: %s is locked" % user

            else:

                users.append(user)

        return users

    def cleanup_sensitive_data(self):

        cleanup_userdata()

        cleanup_tmp()

        cleanup_log()

# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :

from image_creator.os_type import OSBase

class Windows(OSBase):

	pass

# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :