sig_cloud/diskimage-builder
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Create BlockDeviceCmd object

Browse Source 
This creates a BlockDeviceCmd object to hold the main() function.
This doesn't really do anything different right now, but sets a base
for using argparse subparsers to handle the command-line

Change-Id: I4acf95ff4d554a3b4e7e2244ab1706631b98458f
This commit is contained in:
Ian Wienand 2017-05-02 14:24:35 +10:00
parent a06c610a8c
commit ff5b30db8a
1 changed files with 68 additions and 61 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
diskimage_builder/block_device/cmd.py
View File

@ -21,8 +21,12 @@ import yaml
from diskimage_builder.block_device.blockdevice import BlockDevice from diskimage_builder.block_device.blockdevice import BlockDevice
from diskimage_builder import logging_config from diskimage_builder import logging_config
logger = logging.getLogger(__name__)
def generate_phase_doc():
class BlockDeviceCmd(object):
def generate_phase_doc(self):
phase_doc = "" phase_doc = ""
bdattrs = dir(BlockDevice) bdattrs = dir(BlockDevice)
for attr in bdattrs: for attr in bdattrs:
@ -33,11 +37,9 @@ def generate_phase_doc():
phase_doc += " " + method.__doc__.split("\n")[0] + "\n" phase_doc += " " + method.__doc__.split("\n")[0] + "\n"
return phase_doc return phase_doc
def main(self):
def main():
logging_config.setup() logging_config.setup()
logger = logging.getLogger(__name__) phase_doc = self.generate_phase_doc()
phase_doc = generate_phase_doc()
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
formatter_class=argparse.RawDescriptionHelpFormatter, formatter_class=argparse.RawDescriptionHelpFormatter,
@ -46,8 +48,9 @@ def main():
parser.add_argument('--phase', required=True, parser.add_argument('--phase', required=True,
help="phase to execute") help="phase to execute")
parser.add_argument('--params', required=False, parser.add_argument('--params', required=False,
help="YAML file containing parameters for block-device" help="YAML file containing parameters for "
"handling. Default is DIB_BLOCK_DEVICE_PARAMS_YAML") "block-device handling. Default is "
"DIB_BLOCK_DEVICE_PARAMS_YAML")
parser.add_argument('--symbol', required=False, parser.add_argument('--symbol', required=False,
help="symbol to query for getval") help="symbol to query for getval")
args = parser.parse_args() args = parser.parse_args()
@ -57,7 +60,8 @@ def main():
if 'DIB_BLOCK_DEVICE_PARAMS_YAML' in os.environ: if 'DIB_BLOCK_DEVICE_PARAMS_YAML' in os.environ:
param_file = os.environ['DIB_BLOCK_DEVICE_PARAMS_YAML'] param_file = os.environ['DIB_BLOCK_DEVICE_PARAMS_YAML']
else: else:
parser.error("DIB_BLOCK_DEVICE_PARAMS_YAML or --params not set") parser.error(
"DIB_BLOCK_DEVICE_PARAMS_YAML or --params not set")
else: else:
param_file = args.params param_file = args.params
logger.info("params [%s]" % param_file) logger.info("params [%s]" % param_file)
@ -84,7 +88,10 @@ def main():
logger.error("phase [%s] does not exists" % args.phase) logger.error("phase [%s] does not exists" % args.phase)
return 1 return 1
return 0
def main():
bdc = BlockDeviceCmd()
return bdc.main()
if __name__ == "__main__": if __name__ == "__main__":
main() sys.exit(main())