bootstrap-vz/base/fs/partitions/partition.py

61 lines
1.9 KiB
Python
Raw Normal View History

from common.tools import log_check_call
from abstractpartition import AbstractPartition
class Partition(AbstractPartition):
events = [{'name': 'create', 'src': 'nonexistent', 'dst': 'unmapped'},
{'name': 'map', 'src': 'unmapped', 'dst': 'mapped'},
{'name': 'map', 'src': 'unmapped_fmt', 'dst': 'formatted'},
{'name': 'format', 'src': 'mapped', 'dst': 'formatted'},
{'name': 'mount', 'src': 'formatted', 'dst': 'mounted'},
{'name': 'unmount', 'src': 'mounted', 'dst': 'formatted'},
{'name': 'unmap', 'src': 'formatted', 'dst': 'unmapped_fmt'},
{'name': 'unmap', 'src': 'mapped', 'dst': 'unmapped'},
]
def __init__(self, size, filesystem, previous, callbacks={}):
self.previous = previous
callbacks.update({'onbeforecreate': self._create,
'onbeforemap': self._map,
'onbeforeunmap': self._unmap,
})
super(Partition, self).__init__(size, filesystem, callbacks=callbacks)
def get_index(self):
if self.previous is None:
return 1
else:
return self.previous.get_index()+1
def get_start(self):
if self.previous is None:
return 0
else:
return self.previous.get_start() + self.previous.size
def create(self, volume):
self.fsm.create(volume=volume)
def _create(self, e):
start = self.get_start()
# maybe use `parted -- name` to set partition name
log_check_call(['/sbin/parted', '--script', '--align', 'optimal', e.volume.device_path,
'--', 'mkpart', 'primary', str(start), str(start + self.size)])
def map(self, device_path):
self.fsm.map(device_path=device_path)
def _map(self, e):
self.device_path = e.device_path
def _unmap(self, e):
self.device_path = None
# Partition flags:
# boot, root, swap, hidden, raid, lvm, lba, legacy_boot, palo
# Partition tables
# bsd, dvh, gpt, loop, mac, msdos, pc98, sun