bootstrap-vz/bootstrapvz/common/fs/__init__.py
Andrew Bogott 43b321ce91 Replace a bunch of \\ escapes with raw strings
There are a few cases where changing to a raw string
will change the behavior; I'll address that in a later patch.
2019-03-05 20:05:59 +00:00

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)