diff options
author | Lars Wirzenius <liw@liw.fi> | 2017-03-28 21:49:13 +0300 |
---|---|---|
committer | Lars Wirzenius <liw@liw.fi> | 2017-03-28 21:49:13 +0300 |
commit | 818fbc9c0de0902f60435cc4d41b8ced56dc9319 (patch) | |
tree | 4608954782f25d4c1502b8ee550c60d7eb6abd07 | |
parent | 7c1ad52dd01c1f58fc7d08c0a4880ab51960718d (diff) | |
download | vmdb2-818fbc9c0de0902f60435cc4d41b8ced56dc9319.tar.gz |
Add plugin for partitioning
-rw-r--r-- | vmdb/plugins/partition_plugin.py | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/vmdb/plugins/partition_plugin.py b/vmdb/plugins/partition_plugin.py new file mode 100644 index 0000000..cb0833e --- /dev/null +++ b/vmdb/plugins/partition_plugin.py @@ -0,0 +1,63 @@ +# 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 logging +import sys + +import cliapp + +import vmdb + + +class PartitionPlugin(cliapp.Plugin): + + def enable(self): + self.app.step_runners.add(MklabelStepRunner()) + self.app.step_runners.add(MkpartStepRunner()) + + +class MklabelStepRunner(vmdb.StepRunnerInterface): + + def get_required_keys(self): + return ['mklabel'] + + def run(self, step_spec, settings, state): + label_type = step_spec['mklabel'] + device = step_spec['device'] + sys.stdout.write( + 'Creating partition table ({}) on {}\n'.format(label_type, device)) + cliapp.runcmd(['parted', device, 'mklabel', label_type], stderr=None) + state.parts = [] + + +class MkpartStepRunner(vmdb.StepRunnerInterface): + + def get_required_keys(self): + return ['mkpart'] + + def run(self, step_spec, settings, state): + part_type = step_spec['mkpart'] + device = step_spec['device'] + start = step_spec['start'] + end = step_spec['end'] + + sys.stdout.write( + 'Creating partition ({}) on {} ({} to {})\n'.format( + part_type, device, start, end)) + cliapp.runcmd(['parted', '-s', device, 'mkpart', part_type, start, end]) |