# # sysutils.py # # Copyright (C) 2009 Red Hat, Inc. # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . # # Red Hat Author(s): Martin Gracik # __all__ = ["joinpaths", "touch", "replace", "create_loop_dev", "remove_loop_dev", "create_dm_dev", "remove_dm_dev"] import sys import os import re import fileinput import subprocess def joinpaths(*args, **kwargs): path = os.path.sep.join(args) if kwargs.get("follow_symlinks"): return os.path.realpath(path) else: return path def touch(fname): with open(fname, "w") as fobj: pass def replace(fname, find, replace): fin = fileinput.input(fname, inplace=1) pattern = re.compile(find) for line in fin: line = pattern.sub(replace, line) sys.stdout.write(line) fin.close() def create_loop_dev(fname): cmd = ["losetup", "-f"] p = subprocess.Popen(cmd, stdout=subprocess.PIPE) rc = p.wait() if not rc == 0: return None loopdev = p.stdout.read().strip() cmd = ["losetup", loopdev, fname] p = subprocess.Popen(cmd, stdout=subprocess.PIPE) rc = p.wait() if not rc == 0: return None return loopdev def remove_loop_dev(dev): cmd = ["losetup", "-d", dev] p = subprocess.Popen(cmd, stdout=subprocess.PIPE) rc = p.wait() def create_dm_dev(name, size, loopdev): table = '0 {0} linear {1} 0'.format(size, loopdev) cmd = ["dmsetup", "create", name, "--table", table] p = subprocess.Popen(cmd, stdout=subprocess.PIPE) rc = p.wait() if not rc == 0: return None return joinpaths("/dev/mapper", name) def remove_dm_dev(dev): cmd = ["dmsetup", "remove", dev] p = subprocess.Popen(cmd, stdout=subprocess.PIPE) rc = p.wait()