ec7f56c1b2
During the creation of a disk image (e.g. for a VM), there is the need to create, setup, configure and afterwards detach some kind of storage where the newly installed OS can be copied to or directly installed in. This patch implements partitioning handling. Change-Id: I0ca6a4ae3a2684d473b44e5f332ee4225ee30f8c Signed-off-by: Andreas Florath <andreas@florath.net>
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
# Copyright 2016 Andreas Florath (andreas@florath.net)
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import argparse
|
|
from diskimage_builder.block_device.blockdevice import BlockDevice
|
|
from diskimage_builder import logging_config
|
|
import logging
|
|
|
|
|
|
def val_else_none(s):
|
|
return s if s is None or len(s) > 0 else None
|
|
|
|
|
|
def generate_phase_doc():
|
|
phase_doc = ""
|
|
bdattrs = dir(BlockDevice)
|
|
for attr in bdattrs:
|
|
if attr.startswith("cmd_"):
|
|
phase_doc += " '" + attr[4:] + "'\n"
|
|
method = getattr(BlockDevice, attr, None)
|
|
# The first line is the line that is used
|
|
phase_doc += " " + method.__doc__.split("\n")[0] + "\n"
|
|
return phase_doc
|
|
|
|
|
|
def main():
|
|
logging_config.setup()
|
|
logger = logging.getLogger(__name__)
|
|
phase_doc = generate_phase_doc()
|
|
|
|
parser = argparse.ArgumentParser(
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
description="Create block device layer",
|
|
epilog="Available phases:\n" + phase_doc)
|
|
parser.add_argument('--phase', required=True,
|
|
help="phase to execute")
|
|
parser.add_argument('--config', required=False,
|
|
help="configuration for block device "
|
|
"layer as JSON object")
|
|
parser.add_argument('--build-dir', required=True,
|
|
help="path to temporary build dir")
|
|
parser.add_argument('--image-size', required=False,
|
|
help="default image size")
|
|
parser.add_argument('--image-dir', required=False,
|
|
help="default image directory")
|
|
args = parser.parse_args()
|
|
|
|
logger.info("phase [%s]" % args.phase)
|
|
logger.info("config [%s]" % args.config)
|
|
logger.info("build_dir [%s]" % args.build_dir)
|
|
|
|
bd = BlockDevice(val_else_none(args.config),
|
|
val_else_none(args.build_dir),
|
|
val_else_none(args.image_size),
|
|
val_else_none(args.image_dir))
|
|
|
|
# Check if the method is available
|
|
method = getattr(bd, "cmd_" + args.phase, None)
|
|
if callable(method):
|
|
# If so: call it.
|
|
return method()
|
|
else:
|
|
logger.error("phase [%s] does not exists" % args.phase)
|
|
return 1
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|