bootstrap-vz/common/fs/__init__.py

def get_partitions():
	import re
	regexp = re.compile('^ *(?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


def remount(volume, fn):
	from base.fs.partitionmaps.none import NoPartitions

	p_map = volume.partition_map
	volume.unmount_specials()
	if hasattr(p_map, 'boot'):
		boot_dir = p_map.boot.mount_dir
		p_map.boot.unmount()
	root_dir = p_map.root.mount_dir
	p_map.root.unmount()
	if not isinstance(p_map, NoPartitions):
		p_map.unmap(volume)
		fn()
		p_map.map(volume)
	else:
		fn()
	p_map.root.mount(root_dir)
	if hasattr(p_map, 'boot'):
		p_map.boot.mount(boot_dir)
	volume.mount_specials()