114 lines
3.5 KiB
Python
114 lines
3.5 KiB
Python
import argparse
|
|
import sys
|
|
|
|
from app.config import load_config
|
|
|
|
|
|
def _store(cfg):
|
|
import os
|
|
from photoflow.core import Store
|
|
os.makedirs(cfg.data_dir, exist_ok=True)
|
|
return Store(cfg.db_path).connect()
|
|
|
|
|
|
def _immich(deps, cfg):
|
|
if "immich" in deps:
|
|
return deps["immich"]
|
|
from photoflow.immich import ImmichClient
|
|
return ImmichClient(cfg.immich_url, cfg.immich_api_key)
|
|
|
|
|
|
def cmd_ingest(deps, *, date_from, date_to, tag, subset, full) -> int:
|
|
cfg = deps["config"]
|
|
store = _store(cfg)
|
|
client = _immich(deps, cfg)
|
|
from app.ingest import run_ingest
|
|
res = run_ingest(client, store, cfg.thumbs_dir, date_from=date_from,
|
|
date_to=date_to, tag=tag, subset=subset, full=full)
|
|
store.close()
|
|
print(f"Ingested {res['fetched']} asset(s); "
|
|
f"{res['processed_marked']} already-processed.")
|
|
return 0
|
|
|
|
|
|
def cmd_cluster(deps, *, gap_factor) -> int:
|
|
cfg = deps["config"]
|
|
store = _store(cfg)
|
|
from app.cluster_run import run_cluster
|
|
res = run_cluster(store, gap_factor=gap_factor)
|
|
store.close()
|
|
print(f"Built {res['clusters']} candidate cluster(s). Run 'categorize serve' to review.")
|
|
return 0
|
|
|
|
|
|
def cmd_apply(deps, *, yes) -> int:
|
|
cfg = deps["config"]
|
|
store = _store(cfg)
|
|
from app.writeback import apply_all, APPLYABLE
|
|
pending = [c for c in store.all_clusters() if c.status in APPLYABLE]
|
|
if not pending:
|
|
print("Nothing to apply.")
|
|
store.close()
|
|
return 0
|
|
if not yes:
|
|
ans = input(f"Apply {len(pending)} cluster decision(s) to Immich? [y/N] ")
|
|
if ans.strip().lower() not in ("y", "yes"):
|
|
print("Aborted.")
|
|
store.close()
|
|
return 1
|
|
client = _immich(deps, cfg)
|
|
results = apply_all(client, store)
|
|
store.close()
|
|
ok = sum(len(r["succeeded"]) for r in results)
|
|
bad = sum(len(r["failed"]) for r in results)
|
|
print(f"Applied {ok} tag write(s); {bad} failure(s) across {len(results)} cluster(s).")
|
|
return 0 if bad == 0 else 1
|
|
|
|
|
|
def cmd_serve(deps) -> int:
|
|
from app import create_app
|
|
create_app(deps["config"]).run(host="0.0.0.0", port=8084)
|
|
return 0
|
|
|
|
|
|
def build_parser() -> argparse.ArgumentParser:
|
|
p = argparse.ArgumentParser(prog="categorize")
|
|
sub = p.add_subparsers(dest="command", required=True)
|
|
|
|
ing = sub.add_parser("ingest")
|
|
ing.add_argument("--from", dest="date_from")
|
|
ing.add_argument("--to", dest="date_to")
|
|
ing.add_argument("--tag")
|
|
ing.add_argument("--subset", type=int)
|
|
ing.add_argument("--full", action="store_true", help="ignore incremental updatedAfter")
|
|
|
|
cl = sub.add_parser("cluster")
|
|
cl.add_argument("--gap-factor", type=float, default=6.0)
|
|
|
|
sub.add_parser("serve")
|
|
|
|
ap = sub.add_parser("apply")
|
|
ap.add_argument("--yes", action="store_true", help="skip the confirmation prompt")
|
|
|
|
return p
|
|
|
|
|
|
def main(argv=None) -> int:
|
|
args = build_parser().parse_args(argv)
|
|
deps = {"config": load_config()}
|
|
if args.command == "serve":
|
|
return cmd_serve(deps)
|
|
if args.command == "ingest":
|
|
return cmd_ingest(deps, date_from=args.date_from, date_to=args.date_to,
|
|
tag=args.tag, subset=args.subset, full=args.full)
|
|
if args.command == "cluster":
|
|
return cmd_cluster(deps, gap_factor=args.gap_factor)
|
|
if args.command == "apply":
|
|
return cmd_apply(deps, yes=args.yes)
|
|
print(f"Command '{args.command}' is not implemented yet.")
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|