77 lines
2.1 KiB
Python
77 lines
2.1 KiB
Python
from pathlib import Path
|
|
|
|
import questionary
|
|
import typer
|
|
import yaml
|
|
|
|
app = typer.Typer(
|
|
name="setup",
|
|
help="⚙️ Setup and configuration management.",
|
|
no_args_is_help=True,
|
|
rich_markup_mode="markdown",
|
|
)
|
|
|
|
|
|
@app.command("config")
|
|
def setup_config() -> None:
|
|
typer.secho("🛠️ Interactive Configuration Setup", fg=typer.colors.CYAN, bold=True)
|
|
|
|
config_path = Path.home() / ".devgen.yaml"
|
|
current_config = {}
|
|
if config_path.exists():
|
|
try:
|
|
with config_path.open("r", encoding="utf-8") as f:
|
|
current_config = yaml.safe_load(f) or {}
|
|
except Exception:
|
|
pass
|
|
|
|
from devgen.utils import get_questionary_style
|
|
|
|
style = get_questionary_style()
|
|
|
|
# Questions
|
|
provider = questionary.select(
|
|
"Select AI Provider:",
|
|
choices=["gemini", "openai", "huggingface", "openrouter", "anthropic"],
|
|
default=current_config.get("provider", "gemini"),
|
|
style=style,
|
|
).ask()
|
|
|
|
model_default = current_config.get("model", "gemini-2.5-flash")
|
|
|
|
model = questionary.text(
|
|
"Enter Model Name:", default=model_default, style=style
|
|
).ask()
|
|
|
|
api_key = questionary.password(
|
|
"Enter API Key (leave empty to keep existing or none):", style=style
|
|
).ask()
|
|
|
|
if not api_key:
|
|
api_key = current_config.get("api_key", "")
|
|
|
|
emoji_choice = questionary.select(
|
|
"Use Emojis in Commit Messages?",
|
|
choices=["Yes", "No"],
|
|
default="Yes" if current_config.get("emoji", True) else "No",
|
|
style=style,
|
|
).ask()
|
|
emoji = emoji_choice == "Yes"
|
|
|
|
# Save Config
|
|
new_config = {
|
|
"provider": provider,
|
|
"model": model,
|
|
"api_key": api_key,
|
|
"emoji": emoji,
|
|
}
|
|
|
|
try:
|
|
with config_path.open("w", encoding="utf-8") as f:
|
|
yaml.dump(new_config, f, default_flow_style=False)
|
|
typer.secho(f"\n Configuration saved to {config_path}", fg=typer.colors.GREEN)
|
|
typer.echo(yaml.dump(new_config, default_flow_style=False))
|
|
except Exception as e:
|
|
typer.secho(f"\n Failed to save configuration: {e}", fg=typer.colors.RED)
|
|
raise typer.Exit(code=1)
|