From 47e6ee9ca8bf07483bb6f0c51386340980db089a Mon Sep 17 00:00:00 2001 From: Nicolo P Date: Wed, 23 Feb 2022 23:15:57 +0100 Subject: [PATCH] Refactor main code into function --- crapgrep/crapgrep.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/crapgrep/crapgrep.py b/crapgrep/crapgrep.py index 8067e09..1354284 100644 --- a/crapgrep/crapgrep.py +++ b/crapgrep/crapgrep.py @@ -158,8 +158,7 @@ def process_grep(args_tree: dict) -> list: return found_lines -# When the script is executed directly... -if __name__ == "__main__": +def main(): args = sys.argv[1:] if not check_args(args): @@ -179,3 +178,8 @@ if __name__ == "__main__": except exceptions.InvalidPatternError as e: print(e.get_message()) sys.exit(1) + + +# When the script is executed directly... +if __name__ == "__main__": + main()