From d44f732d3b644ddc9f2f3d2d47c177561403f0af Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 11 May 2023 20:34:46 -0700 Subject: [PATCH] main should use argv if args not provided --- aider/main.py | 14 ++++++++++---- tests/test_main.py | 1 + 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/aider/main.py b/aider/main.py index 629db16d1..11d4ff9f0 100644 --- a/aider/main.py +++ b/aider/main.py @@ -5,7 +5,10 @@ from dotenv import load_dotenv from aider.coder import Coder -def main(argv=[]): +def main(args=None): + if not args: + args = sys.argv[1:] + load_dotenv() env_prefix = "AIDER_" parser = argparse.ArgumentParser(description="aider - chat with GPT about your code") @@ -66,12 +69,15 @@ def main(argv=[]): "--yes", action="store_true", help="Always say yes to every confirmation", + default=False, ) - args = parser.parse_args(argv) + args = parser.parse_args(args) fnames = args.files pretty = args.pretty - coder = Coder(args.model, fnames, pretty, args.history_file, args.show_diffs, args.auto_commits, args.yes) + coder = Coder( + args.model, fnames, pretty, args.history_file, args.show_diffs, args.auto_commits, args.yes + ) coder.commit(ask=True, prefix="wip: ", which="repo_files") if args.apply: @@ -84,5 +90,5 @@ def main(argv=[]): if __name__ == "__main__": - status = main(sys.argv[1:]) + status = main() sys.exit(status) diff --git a/tests/test_main.py b/tests/test_main.py index fb5eeb8f5..646e9111c 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -35,3 +35,4 @@ class TestMain(TestCase): sys.stdin = dev_null main(["--yes", "foo.txt"]) sys.stdin = save_stdin + self.assertTrue(os.path.exists("foo.txt"))