2013-09-18 00:46:58 +02:00
|
|
|
from abstract import AbstractPartitionMap
|
|
|
|
|
from ..partitions.gpt import GPTPartition
|
|
|
|
|
from ..partitions.gpt_swap import GPTSwapPartition
|
|
|
|
|
from common.tools import log_check_call
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class GPTPartitionMap(AbstractPartitionMap):
|
|
|
|
|
|
|
|
|
|
def __init__(self, data):
|
2013-10-05 20:41:05 +02:00
|
|
|
self.partitions = []
|
2013-09-18 00:46:58 +02:00
|
|
|
if 'boot' in data:
|
|
|
|
|
self.boot = GPTPartition(data['boot']['size'], data['boot']['filesystem'], 'boot', None)
|
2013-10-05 20:41:05 +02:00
|
|
|
self.partitions.append(self.boot)
|
2013-10-27 08:52:26 +01:00
|
|
|
self.root = GPTPartition(data['root']['size'], data['root']['filesystem'], 'root',
|
|
|
|
|
getattr(self, 'boot', None))
|
2013-10-05 20:41:05 +02:00
|
|
|
self.partitions.append(self.root)
|
2013-09-18 00:46:58 +02:00
|
|
|
if 'swap' in data:
|
|
|
|
|
self.swap = GPTSwapPartition(data['swap']['size'], self.root)
|
2013-10-05 20:41:05 +02:00
|
|
|
self.partitions.append(self.swap)
|
2013-09-18 00:46:58 +02:00
|
|
|
|
2013-09-22 21:20:09 +02:00
|
|
|
super(GPTPartitionMap, self).__init__()
|
|
|
|
|
|
2013-09-25 23:27:31 +02:00
|
|
|
def _before_create(self, event):
|
|
|
|
|
volume = event.volume
|
2013-09-18 00:46:58 +02:00
|
|
|
log_check_call(['/sbin/parted', '--script', '--align', 'none', volume.device_path,
|
|
|
|
|
'--', 'mklabel', 'gpt'])
|
|
|
|
|
for partition in self.partitions:
|
|
|
|
|
partition.create(volume)
|
|
|
|
|
|
|
|
|
|
boot_idx = self.root.get_index()
|
2013-10-05 20:41:05 +02:00
|
|
|
if hasattr(self, 'boot'):
|
2013-09-18 00:46:58 +02:00
|
|
|
boot_idx = self.boot.get_index()
|
|
|
|
|
log_check_call(['/sbin/parted', '--script', volume.device_path,
|
|
|
|
|
'--', 'set ' + str(boot_idx) + ' boot on'])
|
2013-09-25 23:27:31 +02:00
|
|
|
log_check_call(['/sbin/parted', '--script', volume.device_path,
|
|
|
|
|
'--', 'set ' + str(boot_idx) + ' bios_grub on'])
|
