Support writing to a different file

This commit is contained in:
Gilles Peskine 2019-07-27 23:44:01 +02:00
parent 5639aef7d7
commit 40f103cea2

View file

@ -330,6 +330,8 @@ if __name__ == '__main__':
parser.add_argument('--force', '-o', parser.add_argument('--force', '-o',
help="""For the set command, if SYMBOL is not help="""For the set command, if SYMBOL is not
present, add a definition for it.""") present, add a definition for it.""")
parser.add_argument('--write', '-w',
help="""File to write to instead of the input file.""")
subparsers = parser.add_subparsers(dest='command', subparsers = parser.add_subparsers(dest='command',
title='Commands') title='Commands')
parser_get = subparsers.add_parser('get', parser_get = subparsers.add_parser('get',
@ -389,7 +391,7 @@ if __name__ == '__main__':
config.unset(args.symbol) config.unset(args.symbol)
else: else:
config.adapt(args.adapter) config.adapt(args.adapter)
config.write() config.write(args.write)
# Import modules only used by main only if main is defined and called. # Import modules only used by main only if main is defined and called.
# pylint: disable=wrong-import-position # pylint: disable=wrong-import-position