From b6e80b7288ef24d942fc7422cd2e0d6f55205fdb Mon Sep 17 00:00:00 2001 From: Sebastian Lohff Date: Wed, 27 Mar 2019 00:42:56 +0100 Subject: [PATCH] Add format argument to argparser and actually use it --- genconfdrv/genconfdrv.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/genconfdrv/genconfdrv.py b/genconfdrv/genconfdrv.py index 392a992..43c6e5a 100755 --- a/genconfdrv/genconfdrv.py +++ b/genconfdrv/genconfdrv.py @@ -273,6 +273,8 @@ def main(): parser.add_argument("-a", "--add-user", default=[], nargs="+", help="Add users, format is username:key?:sudo?:gecos?:password?, " "sudo is a bool, key is either an ssh key or a path to an ssh key") + parser.add_argument("-f", "--format", default=None, choices=('tgz', 'iso'), + help="Specify output format, default is to infer from output file extension") args = parser.parse_args() @@ -339,7 +341,7 @@ def main(): cfgdrv.add_user(user[0], keys, sudo=sudo, gecos=gecos, password=password) if args.output: - cfgdrv.write_iso(args.output) + cfgdrv.write_drive(args.output, args.format) finally: if cfgdrv: cfgdrv.close()