From aeda2e0ab9d78bc380af3840fc7fbca7c9a24521 Mon Sep 17 00:00:00 2001 From: Bjarne von Horn Date: Wed, 9 Aug 2023 22:26:44 +0200 Subject: [PATCH] improve cli interface --- devices/create_driver_table.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/devices/create_driver_table.py b/devices/create_driver_table.py index 6780f643..d492474b 100644 --- a/devices/create_driver_table.py +++ b/devices/create_driver_table.py @@ -144,10 +144,20 @@ if __name__ == "__main__": from sys import argv import argparse parser = argparse.ArgumentParser() - parser.add_argument("--markdown", nargs=1) - parser.add_argument("devices_dir", nargs=1) + parser.add_argument( + "--markdown", + nargs=1, + help="Markdown output file", + type=argparse.FileType("w") + ) + parser.add_argument( + "devices_dir", + nargs=1, + help="Devices driver source dir" + ) args = parser.parse_args(argv[1:]) table = compute_table(get_all_drivers(args.devices_dir[0])) if args.markdown is not None: - with open(args.markdown[0], "w") as f: + with args.markdown[0] as f: f.write(dump_markdown(table)) + f.write('\n')