bootstrap-vz/bootstrapvz/base/fs/__init__.py

35 lines
1.6 KiB
Python
Raw Normal View History

2014-01-19 01:02:29 +01:00
def load_volume(data, bootloader):
2014-03-23 16:04:03 +01:00
"""Instantiates a volume that corresponds to the data in the manifest
Args:
data (dict): The 'volume' section from the manifest
bootloader (str): Name of the bootloader the system will boot with
Returns:
Volume. The volume that represents all information pertaining to the volume we bootstrap on
"""
from bootstrapvz.common.fs.loopbackvolume import LoopbackVolume
from bootstrapvz.providers.ec2.ebsvolume import EBSVolume
from bootstrapvz.common.fs.virtualdiskimage import VirtualDiskImage
from bootstrapvz.common.fs.virtualmachinedisk import VirtualMachineDisk
2014-03-23 16:04:03 +01:00
# Create a mapping between valid partition maps in the manifest and their corresponding classes
from partitionmaps.gpt import GPTPartitionMap
from partitionmaps.msdos import MSDOSPartitionMap
from partitionmaps.none import NoPartitions
partition_maps = {'none': NoPartitions,
'gpt': GPTPartitionMap,
'msdos': MSDOSPartitionMap,
}
2014-03-23 16:04:03 +01:00
# Instantiate the partition map
2014-01-19 01:02:29 +01:00
partition_map = partition_maps.get(data['partitions']['type'])(data['partitions'], bootloader)
2014-03-23 16:04:03 +01:00
# Create a mapping between valid volume backings in the manifest and their corresponding classes
volume_backings = {'raw': LoopbackVolume,
2013-10-27 12:55:03 +01:00
's3': LoopbackVolume,
'vdi': VirtualDiskImage,
2013-12-28 14:02:28 +01:00
'vmdk': VirtualMachineDisk,
'ebs': EBSVolume
}
2014-03-23 16:04:03 +01:00
# Create the volume with the partition map as an argument
return volume_backings.get(data['backing'])(partition_map)