diff --git a/utilities/database_upgrader.py b/utilities/database_upgrader.py index 6d052c2..c2a8c81 100644 --- a/utilities/database_upgrader.py +++ b/utilities/database_upgrader.py @@ -113,7 +113,7 @@ def main(argv): parser.set_defaults(func=upgrade_all_argparse) args = parser.parse_args(argv) - args.func(args) + return args.func(args) if __name__ == '__main__': raise SystemExit(main(sys.argv[1:])) diff --git a/utilities/ytqueue.py b/utilities/ytqueue.py index 7ae3e1b..698f034 100644 --- a/utilities/ytqueue.py +++ b/utilities/ytqueue.py @@ -39,7 +39,7 @@ def main(argv): parser.set_defaults(func=ytqueue_argparse) args = parser.parse_args(argv) - args.func(args) + return args.func(args) if __name__ == '__main__': - main(sys.argv[1:]) + raise SystemExit(main(sys.argv[1:]))