refactor: remove home_git logic from setup_git_home function

This commit is contained in:
Paul Gauthier (aider) 2024-08-28 10:42:37 -07:00
parent cdbaa58f08
commit abb94fc582

View file

@ -25,15 +25,12 @@ from .dump import dump # noqa: F401
def setup_git_home(io): def setup_git_home(io):
home = Path.home() home = Path.home()
git_repos = list(home.glob("*/.git")) git_repos = list(home.glob("*/.git"))
home_git = home / ".git"
if home_git.exists():
git_repos.insert(0, home_git)
if git_repos: if git_repos:
io.tool_output("Found existing Git repositories in your home directory:") io.tool_output("Found existing Git repositories in your home directory:")
repo_dict = {} repo_dict = {}
for i, repo in enumerate(git_repos, 1): for i, repo in enumerate(git_repos, 1):
repo_name = "Home directory" if repo == home_git else repo.parent.name repo_name = repo.parent.name
io.tool_output(f"{i}. {repo_name}") io.tool_output(f"{i}. {repo_name}")
repo_dict[repo_name.lower()] = repo repo_dict[repo_name.lower()] = repo
@ -50,14 +47,14 @@ def setup_git_home(io):
choice_num = int(choice) choice_num = int(choice)
if 1 <= choice_num <= len(git_repos): if 1 <= choice_num <= len(git_repos):
chosen_repo = git_repos[choice_num - 1] chosen_repo = git_repos[choice_num - 1]
return str(home if chosen_repo == home_git else chosen_repo.parent) return str(chosen_repo.parent)
else: else:
io.tool_error(f"Please enter a number between 1 and {len(git_repos)}") io.tool_error(f"Please enter a number between 1 and {len(git_repos)}")
except ValueError: except ValueError:
choice_lower = choice.lower() choice_lower = choice.lower()
if choice_lower in repo_dict: if choice_lower in repo_dict:
chosen_repo = repo_dict[choice_lower] chosen_repo = repo_dict[choice_lower]
return str(home if chosen_repo == home_git else chosen_repo.parent) return str(chosen_repo.parent)
else: else:
io.tool_error("Please enter a valid number or repository name") io.tool_error("Please enter a valid number or repository name")