diff options
Diffstat (limited to 'vmdb/plugins/partition_plugin.py')
-rw-r--r-- | vmdb/plugins/partition_plugin.py | 121 |
1 files changed, 0 insertions, 121 deletions
diff --git a/vmdb/plugins/partition_plugin.py b/vmdb/plugins/partition_plugin.py deleted file mode 100644 index 60d4225..0000000 --- a/vmdb/plugins/partition_plugin.py +++ /dev/null @@ -1,121 +0,0 @@ -# Copyright 2017 Lars Wirzenius -# -# 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 3 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 <http://www.gnu.org/licenses/>. -# -# =*= License: GPL-3+ =*= - - - -import os -import stat - -import cliapp - -import vmdb - - -class PartitionPlugin(cliapp.Plugin): - - def enable(self): - self.app.step_runners.add(MklabelStepRunner()) - self.app.step_runners.add(MkpartStepRunner()) - self.app.step_runners.add(KpartxStepRunner()) - - -class MklabelStepRunner(vmdb.StepRunnerInterface): - - def get_required_keys(self): - return ['mklabel', 'device'] - - def run(self, step, settings, state): - label_type = step['mklabel'] - device = step['device'] - vmdb.runcmd(['parted', '-s', device, 'mklabel', label_type]) - - -class MkpartStepRunner(vmdb.StepRunnerInterface): - - def get_required_keys(self): - return ['mkpart', 'device', 'start', 'end'] - - def run(self, step, settings, state): - part_type = step['mkpart'] - device = step['device'] - start = step['start'] - end = step['end'] - tag = step.get('tag') or step.get('part-tag') - if tag is None: - tag = step['part-tag'] - fs_type = step.get('fs-type', 'ext2') - - orig = self.list_partitions(device) - vmdb.runcmd(['parted', '-s', device, 'mkpart', part_type, fs_type, start, end]) - - state.tags.append(tag) - if self.is_block_dev(device): - new = self.list_partitions(device) - diff = self.diff_partitions(orig, new) - assert len(diff) == 1 - vmdb.progress('remembering partition {} as {}'.format(diff[0], tag)) - state.tags.set_dev(tag, diff[0]) - - def is_block_dev(self, filename): - st = os.lstat(filename) - return stat.S_ISBLK(st.st_mode) - - def list_partitions(self, device): - output = vmdb.runcmd(['parted', '-m', device, 'print']) - output = output.decode('UTF-8') - partitions = [ - line.split(':')[0] - for line in output.splitlines() - if ':' in line - ] - return [ - word if word.startswith('/') else '{}{}'.format(device, word) - for word in partitions - ] - - def diff_partitions(self, old, new): - return [ - line - for line in new - if line not in old - ] - - -class KpartxStepRunner(vmdb.StepRunnerInterface): - - def get_required_keys(self): - return ['kpartx'] - - def run(self, step, settings, state): - device = step['kpartx'] - tags = state.tags.get_tags() - devs = self.kpartx(device) - for tag, dev in zip(tags, devs): - vmdb.progress('remembering {} as {}'.format(dev, tag)) - state.tags.set_dev(tag, dev) - - def kpartx(self, device): - output = vmdb.runcmd(['kpartx', '-asv', device]).decode('UTF-8') - for line in output.splitlines(): - words = line.split() - if words[0] == 'add': - name = words[2] - yield '/dev/mapper/{}'.format(name) - - def teardown(self, step, settings, state): - device = step['kpartx'] - vmdb.runcmd(['kpartx', '-dsv', device]) |