mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 09:50:37 +00:00

There are a few cases where changing to a raw string will change the behavior; I'll address that in a later patch.
33 lines
976 B
Python
33 lines
976 B
Python
from contextlib import contextmanager
|
|
|
|
|
|
def get_partitions():
|
|
import re
|
|
regexp = re.compile(r'^ *(?P<major>\d+) *(?P<minor>\d+) *(?P<num_blks>\d+) (?P<dev_name>\S+)$')
|
|
matches = {}
|
|
path = '/proc/partitions'
|
|
with open(path) as partitions:
|
|
next(partitions)
|
|
next(partitions)
|
|
for line in partitions:
|
|
match = regexp.match(line)
|
|
if match is None:
|
|
raise RuntimeError('Unable to parse {line} in {path}'.format(line=line, path=path))
|
|
matches[match.group('dev_name')] = match.groupdict()
|
|
return matches
|
|
|
|
|
|
@contextmanager
|
|
def unmounted(volume):
|
|
from bootstrapvz.base.fs.partitionmaps.none import NoPartitions
|
|
|
|
p_map = volume.partition_map
|
|
root_dir = p_map.root.mount_dir
|
|
p_map.root.unmount()
|
|
if not isinstance(p_map, NoPartitions):
|
|
p_map.unmap(volume)
|
|
yield
|
|
p_map.map(volume)
|
|
else:
|
|
yield
|
|
p_map.root.mount(destination=root_dir)
|