From a0ce83fb7266d38c4923de031e3708f4ad8a9ae1 Mon Sep 17 00:00:00 2001 From: "James Smith (aider)" Date: Tue, 11 Mar 2025 21:41:29 +0200 Subject: [PATCH] refactor: Extract main block to main function --- git-tidy | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/git-tidy b/git-tidy index 86762eb..951dd08 100755 --- a/git-tidy +++ b/git-tidy @@ -125,9 +125,8 @@ def get_default_branch() -> str: return branch_list[selection] - -if __name__ == "__main__": - +def main(): + """Main function.""" default_branch = get_default_branch() print(f"Switching to {default_branch}") run(["git", "checkout", default_branch]) @@ -156,3 +155,6 @@ if __name__ == "__main__": for branch in merged_branches: if branch[0] != "*": # This time we want to explicitly exclude the checked-out branch. subprocess.run(["git", "branch", "-d", branch.strip()], check=True) + +if __name__ == "__main__": + main()