update
This commit is contained in:
2 files changed
+171
No files matched your search
@@ -0,0 +1,169 @@
|
||||
from pathlib import Path
|
||||
from typing import Any, Optional
|
||||
|
||||
import questionary
|
||||
import typer
|
||||
import yaml
|
||||
from typing_extensions import Annotated
|
||||
|
||||
from devgen.utils import get_questionary_style, load_config
|
||||
|
||||
app = typer.Typer(
|
||||
name="config",
|
||||
help="🔧 Manage configuration values.",
|
||||
no_args_is_help=True,
|
||||
rich_markup_mode="markdown",
|
||||
)
|
||||
|
||||
|
||||
def _save_config(config: dict[str, Any]) -> None:
|
||||
config_path = Path.home() / ".devgen.yaml"
|
||||
try:
|
||||
with config_path.open("w", encoding="utf-8") as f:
|
||||
yaml.dump(config, f, default_flow_style=False)
|
||||
except Exception as e:
|
||||
typer.secho(f"❌ Failed to save config: {e}", fg=typer.colors.RED)
|
||||
raise typer.Exit(code=1)
|
||||
|
||||
|
||||
@app.command("list")
|
||||
def list_config() -> None:
|
||||
"""Show all configuration values."""
|
||||
config = load_config()
|
||||
if not config:
|
||||
typer.echo("Configuration is empty.")
|
||||
return
|
||||
|
||||
typer.echo(yaml.dump(config, default_flow_style=False))
|
||||
|
||||
|
||||
@app.command("edit")
|
||||
def edit_config(
|
||||
key: Annotated[Optional[str], typer.Argument(help="Configuration key.")] = None,
|
||||
value: Annotated[Optional[str], typer.Argument(help="New value.")] = None,
|
||||
) -> None:
|
||||
"""Edit a specific configuration value."""
|
||||
config = load_config()
|
||||
style = get_questionary_style()
|
||||
|
||||
if not key:
|
||||
choices = list(config.keys())
|
||||
key = questionary.select(
|
||||
"Select key to update:", choices=choices, style=style
|
||||
).ask()
|
||||
|
||||
if not key:
|
||||
raise typer.Exit()
|
||||
|
||||
if value is None:
|
||||
# Check if existing value is boolean to offer select
|
||||
current_val = config.get(key)
|
||||
|
||||
if key == "provider":
|
||||
value = questionary.select(
|
||||
"Select AI Provider:",
|
||||
choices=["gemini", "openai", "huggingface", "openrouter", "anthropic"],
|
||||
default=str(current_val) if current_val else "gemini",
|
||||
style=style,
|
||||
).ask()
|
||||
if value is None:
|
||||
raise typer.Exit(code=130)
|
||||
elif isinstance(current_val, bool):
|
||||
val_choice = questionary.select(
|
||||
f"Select value for '{key}':",
|
||||
choices=["True", "False"],
|
||||
default=str(current_val),
|
||||
style=style,
|
||||
).ask()
|
||||
if val_choice is None:
|
||||
raise typer.Exit(code=130)
|
||||
value = val_choice
|
||||
else:
|
||||
value = questionary.text(
|
||||
f"Enter value for '{key}':",
|
||||
default=str(current_val) if current_val is not None else "",
|
||||
style=style,
|
||||
).ask()
|
||||
|
||||
if value is None:
|
||||
raise typer.Exit(code=130)
|
||||
|
||||
# Basic type inference
|
||||
if isinstance(value, str):
|
||||
if value.lower() == "true":
|
||||
val = True
|
||||
elif value.lower() == "false":
|
||||
val = False
|
||||
elif value.isdigit():
|
||||
val = int(value)
|
||||
else:
|
||||
val = value
|
||||
else:
|
||||
val = value
|
||||
|
||||
config[key] = val
|
||||
_save_config(config)
|
||||
typer.secho(f" Set '{key}' to '{val}'", fg=typer.colors.GREEN)
|
||||
|
||||
|
||||
@app.command("set")
|
||||
def set_config() -> None:
|
||||
"""Run the interactive configuration wizard."""
|
||||
typer.secho("🛠️ Interactive Configuration Setup", fg=typer.colors.CYAN, bold=True)
|
||||
|
||||
config = load_config()
|
||||
style = get_questionary_style()
|
||||
|
||||
# Questions
|
||||
provider = questionary.select(
|
||||
"Select AI Provider:",
|
||||
choices=["gemini", "openai", "huggingface", "openrouter", "anthropic"],
|
||||
default=config.get("provider", "gemini"),
|
||||
style=style,
|
||||
).ask()
|
||||
if provider is None:
|
||||
raise typer.Exit(code=130)
|
||||
|
||||
model_default = config.get("model", "gemini-2.5-flash")
|
||||
|
||||
model = questionary.text(
|
||||
"Enter Model Name:", default=model_default, style=style
|
||||
).ask()
|
||||
if model is None:
|
||||
raise typer.Exit(code=130)
|
||||
|
||||
api_key = questionary.password(
|
||||
"Enter API Key (leave empty to keep existing or none):", style=style
|
||||
).ask()
|
||||
if api_key is None:
|
||||
raise typer.Exit(code=130)
|
||||
|
||||
if not api_key:
|
||||
api_key = config.get("api_key", "")
|
||||
|
||||
emoji_choice = questionary.select(
|
||||
"Use Emojis in Commit Messages?",
|
||||
choices=["Yes", "No"],
|
||||
default="Yes" if config.get("emoji", True) else "No",
|
||||
style=style,
|
||||
).ask()
|
||||
if emoji_choice is None:
|
||||
raise typer.Exit(code=130)
|
||||
emoji = emoji_choice == "Yes"
|
||||
|
||||
# Save Config
|
||||
new_config = {
|
||||
"provider": provider,
|
||||
"model": model,
|
||||
"api_key": api_key,
|
||||
"emoji": emoji,
|
||||
}
|
||||
|
||||
# Merge with existing config to preserve other keys?
|
||||
# The setup wizard usually overwrites or defines the core keys.
|
||||
# Let's preserve other keys.
|
||||
config.update(new_config)
|
||||
|
||||
_save_config(config)
|
||||
typer.secho("\nConfiguration saved.", fg=typer.colors.GREEN)
|
||||
typer.echo(yaml.dump(new_config, default_flow_style=False))
|
||||
Reference in new issue
Block a user