flanandClaude Sonnet 4.6
eedbd3180a
Linting (ruff): - cli.py: sort relative imports (I001) - jobs.py: remove unused get_people import (F401), wrap long line (E501) - executor.py: remove unused success variable (F841), wrap 4 long lines (E501) Tests (24 passing): - tests/test_config.py: config singleton defaults + env var overrides - tests/test_upload_tracker.py: mark/filter/reset/summary logic - tests/test_immich_api.py: filter_recent_assets date boundary cases - tests/test_jobs.py: _resolve_strategy with LIMIT env var and fallbacks - pyproject.toml: add [tool.pytest.ini_options] testpaths=["tests"] so pytest doesn't scan .venv in CI Security: - Dockerfile: bump CUDA base from 12.6.3 to 12.9.2 to pick up patched Ubuntu packages (fixes Dependabot low-severity alert) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
88 lines
3.0 KiB
Python
88 lines
3.0 KiB
Python
"""Interactive CLI for if-curator."""
|
|
|
|
import logging
|
|
import os
|
|
|
|
from rich import print as rprint
|
|
from rich.prompt import Confirm
|
|
|
|
from .config import Config, ConfigManager
|
|
from .executor import execute_jobs, upload_to_frigate
|
|
from .immich_api import get_people
|
|
from .jobs import _show_preview, auto_configure, interactive_configure
|
|
from .logging import console, setup_logging
|
|
from .upload_tracker import get_person_summary, reset_person
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def main() -> None:
|
|
"""Entry point for if-curator CLI."""
|
|
try:
|
|
setup_logging(verbose=False)
|
|
|
|
console.print(r"""
|
|
[bold blue]if-curator[/bold blue]
|
|
[dim]Immich -> Frigate Training Data Curator[/dim]
|
|
""")
|
|
|
|
ConfigManager.get().interactive_setup()
|
|
|
|
try:
|
|
Config.validate()
|
|
except ValueError as e:
|
|
rprint(f"[bold red]Configuration Error:[/bold red] {e}")
|
|
return
|
|
|
|
rprint(f"Server: [dim]{Config.IMMICH_URL}[/dim]")
|
|
rprint(f"Output: [dim]{Config.OUTPUT_DIR}[/dim]")
|
|
|
|
# Handle RESET_PERSON before anything else
|
|
reset_person_name = os.environ.get("RESET_PERSON", "").strip()
|
|
if reset_person_name:
|
|
reset_person(reset_person_name)
|
|
rprint(f"[bold yellow]Reset tracking data for: {reset_person_name}[/bold yellow]")
|
|
|
|
# Show per-person tracker summary if data exists
|
|
summary = get_person_summary()
|
|
if summary:
|
|
rprint("\n[dim]Tracker summary:[/dim]")
|
|
for person_name, counts in summary.items():
|
|
rprint(f" [dim]{person_name}: {counts['uploaded']} uploaded, {counts['rejected']} rejected[/dim]")
|
|
|
|
people = get_people()
|
|
if not people:
|
|
rprint("[bold red]Could not fetch people from Immich. Check URL/Key.[/bold red]")
|
|
return
|
|
|
|
# Check for non-interactive mode
|
|
auto_mode = os.environ.get("AUTO_MODE", "false").lower() == "true"
|
|
dry_run = os.environ.get("DRY_RUN", "false").lower() in ("true", "1", "yes")
|
|
|
|
if dry_run:
|
|
rprint("[bold yellow]DRY RUN — no images will be downloaded or uploaded[/bold yellow]")
|
|
|
|
if auto_mode:
|
|
rprint("[bold cyan]Running in AUTO mode (non-interactive)[/bold cyan]")
|
|
jobs = auto_configure(people)
|
|
else:
|
|
jobs = interactive_configure(people)
|
|
|
|
if jobs:
|
|
_show_preview(jobs)
|
|
if dry_run:
|
|
rprint("\n[bold yellow]Dry run complete — skipping execute and upload.[/bold yellow]")
|
|
elif auto_mode or Confirm.ask(f"Ready to process {sum(j['limit'] for j in jobs)} images?"):
|
|
execute_jobs(jobs)
|
|
upload_to_frigate(jobs)
|
|
rprint("\n[bold green]Done! Happy Training.[/bold green]")
|
|
else:
|
|
rprint("[yellow]No jobs configured.[/yellow]")
|
|
|
|
except KeyboardInterrupt:
|
|
rprint("\n[bold red]Aborted by user.[/bold red]")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|