Synnefo » snf-image-creator

import parted

from image_creator.util import get_command

from image_creator.util import FatalError

        if not os.path.isfile(f):

            raise FatalError("Unable to open: `%s'. File is missing." % f)

                                     'dev mpoint fs opts freq passno')

        with open(f) as table:

            for line in iter(table):

                entry = line.split('#')[0].strip().split()

                if len(entry) != 6:

                    continue

    def _get_root_partition(self):

        for entry in self._read_fstable('/etc/fstab'):

        mount_options = self._get_mount_options(

                self.disk.getPartitionBySector(last.start).path)

            stat = os.statvfs(mount_options.mpoint)

            # Shrink the last partition. The new size should be the size of the

            # occupied blocks

            # Leave 2048 blocks at the end.

            new_end = new_size + 2048

            if last.type == parted.PARTITION_LOGICAL:

                # Fix the extended partition

                extended = disk.getExtendedPartition()

        for entry in self._read_fstable('/proc/mounts'):

            if entry.mpoint.startswith(os.path.abspath(target)):

                    mpoints.append(entry.mpoint)

        mpoints.sort()

        for mpoint in reversed(mpoints):

            umount(mpoint)

    def _create_filesystems(self, image):

        partitions = self._get_partitions(parted.Disk(parted.Device(image)))

        filesystems = {}

        for p in self.disk.partitions:

        mapped = {}

        try:

            for p in mounted:

                mapped[i] = self._map_partition(loop, i, p.start, p.end)

            # Create the file systems

                absmpoints = self._mount(target,

                    [(mapped[i], filesystems[i].mpoint) for i in mapped.keys()]

                )

            finally:

                self._umount_all(target)