Update cli.py

Author: KRRT7

codeflash/cli_cmds/cli.py

@@ -8,6 +8,7 @@
 from codeflash.cli_cmds.cmd_init import init_codeflash, install_github_actions
 from codeflash.cli_cmds.console import logger
 from codeflash.code_utils import env_utils
+from codeflash.code_utils.code_utils import exit_with_message
 from codeflash.code_utils.config_parser import parse_config_file
 from codeflash.version import __version__ as version
 
@@ -83,25 +84,21 @@ def process_and_validate_cmd_args(args: Namespace) -> Namespace:
         sys.exit()
     if not check_running_in_git_repo(module_root=args.module_root):
         if not confirm_proceeding_with_no_git_repo():
-            logger.critical("No git repository detected and user aborted run. Exiting...")
-            sys.exit(1)
+            exit_with_message("No git repository detected and user aborted run. Exiting...", error_on_exit=True)
         args.no_pr = True
     if args.function and not args.file:
-        logger.error("If you specify a --function, you must specify the --file it is in")
-        sys.exit(1)
+        exit_with_message("If you specify a --function, you must specify the --file it is in", error_on_exit=True)
     if args.file:
         if not Path(args.file).exists():
-            logger.error(f"File {args.file} does not exist")
-            sys.exit(1)
+            exit_with_message(f"File {args.file} does not exist", error_on_exit=True)
         args.file = Path(args.file).resolve()
         if not args.no_pr:
             owner, repo = get_repo_owner_and_name()
             require_github_app_or_exit(owner, repo)
     if args.replay_test:
         for test_path in args.replay_test:
             if not Path(test_path).is_file():
-                logger.error(f"Replay test file {test_path} does not exist")
-                sys.exit(1)
+                exit_with_message(f"Replay test file {test_path} does not exist", error_on_exit=True)
         args.replay_test = [Path(replay_test).resolve() for replay_test in args.replay_test]
 
     return args
@@ -111,8 +108,7 @@ def process_pyproject_config(args: Namespace) -> Namespace:
     try:
         pyproject_config, pyproject_file_path = parse_config_file(args.config_file)
     except ValueError as e:
-        logger.error(e)
-        sys.exit(1)
+        exit_with_message(f"Error parsing config file: {e}", error_on_exit=True)
     supported_keys = [
         "module_root",
         "tests_root",
@@ -207,8 +203,7 @@ def handle_optimize_all_arg_parsing(args: Namespace) -> Namespace:
             )
             apologize_and_exit()
         if not args.no_pr and not check_and_push_branch(git_repo):
-            logger.critical("❌ Branch is not pushed. Exiting...")
-            sys.exit(1)
+            exit_with_message("Branch is not pushed...", error_on_exit=True)
         owner, repo = get_repo_owner_and_name(git_repo)
         if not args.no_pr:
             require_github_app_or_exit(owner, repo)