dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-06 12:45:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

feat: enhance git repository selection in setup_git_home

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-28 10:39:59 -07:00
parent 36aae9cef3
commit cdbaa58f08
1 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aider/main.py
View file

@ -24,17 +24,22 @@ from .dump import dump # noqa: F401
def setup_git_home(io): def setup_git_home(io):
home = Path.home() home = Path.home()
git_repos = 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 = {}
for i, repo in enumerate(git_repos, 1): for i, repo in enumerate(git_repos, 1):
repo_name = repo.parent.name repo_name = "Home directory" if repo == home_git else repo.parent.name
io.tool_output(f"{i}. {repo_name}") io.tool_output(f"{i}. {repo_name}")
repo_dict[repo_name.lower()] = repo
while True: while True:
choice = io.prompt_ask( choice = io.prompt_ask(
"Enter the number of the repository you want to work on, or ENTER for a new" "Enter the number or name of the repository you want to work on, or ENTER for a new"
" project:" " project:"
) )
@ -49,7 +54,12 @@ def setup_git_home(io):
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:
io.tool_error("Please enter a valid number") choice_lower = choice.lower()
if choice_lower in repo_dict:
chosen_repo = repo_dict[choice_lower]
return str(home if chosen_repo == home_git else chosen_repo.parent)
else:
io.tool_error("Please enter a valid number or repository name")
project_name = io.user_input("Enter a name for your new project directory:") project_name = io.user_input("Enter a name for your new project directory:")
new_dir = home / project_name new_dir = home / project_name