Initial commit

2026-05-16 12:05:36 -03:00
parent 0ce972a361
commit e82cee97a7
65 changed files with 9051 additions and 5 deletions
@@ -0,0 +1,79 @@
+from __future__ import annotations
+
+import argparse
+from datetime import datetime
+
+from app.config import configure_logging, get_settings
+from app.db import init_db, session_scope
+from app.inbox_locks import inbox_run_locks
+from app.message_processor import process_inbox
+
+
+def _date(value: str | None):
+    return datetime.strptime(value, "%Y-%m-%d").date() if value else None
+
+
+def backlog(args: argparse.Namespace) -> int:
+    settings = get_settings()
+    configure_logging(settings)
+    init_db()
+    inbox = settings.get_inbox(args.inbox)
+    lease = inbox_run_locks.acquire(inbox.id, blocking=False)
+    if not lease:
+        print(f"Inbox {inbox.id} is already processing.")
+        return 1
+    with session_scope() as session:
+        with lease:
+            summary = process_inbox(
+                session,
+                settings,
+                inbox,
+                folder=args.folder or inbox.folder,
+                mode="backlog",
+                since=_date(args.since),
+                before=_date(args.before),
+                limit=args.limit,
+                dry_run=args.dry_run,
+                reprocess=args.reprocess,
+                mark_seen=args.mark_seen,
+            )
+    print("Backlog run complete")
+    print(f"Inbox: {summary.inbox_id}")
+    print(f"Folder: {summary.folder}")
+    print(f"Scanned messages: {summary.scanned_messages}")
+    print(f"Candidate messages: {summary.candidate_messages}")
+    print(f"Valid reports imported: {summary.valid_reports_imported}")
+    print(f"Duplicate messages skipped: {summary.duplicate_messages_skipped}")
+    print(f"Duplicate report payloads skipped: {summary.duplicate_reports_skipped}")
+    print(f"Rejected messages: {summary.rejected_messages}")
+    print(f"Failed messages: {summary.failed_messages}")
+    print(f"Records imported: {summary.records_imported}")
+    print(f"Alerts created: {summary.alerts_created}")
+    print(f"LLM explanations generated: {summary.llm_explanations_generated}")
+    return 0
+
+
+def build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(prog="python -m app.cli")
+    sub = parser.add_subparsers(dest="command", required=True)
+    backlog_parser = sub.add_parser("backlog")
+    backlog_parser.add_argument("--inbox", required=True)
+    backlog_parser.add_argument("--folder")
+    backlog_parser.add_argument("--since")
+    backlog_parser.add_argument("--before")
+    backlog_parser.add_argument("--limit", type=int, default=500)
+    backlog_parser.add_argument("--dry-run", action="store_true")
+    backlog_parser.add_argument("--reprocess", action="store_true")
+    backlog_parser.add_argument("--mark-seen", action="store_true")
+    backlog_parser.set_defaults(func=backlog)
+    return parser
+
+
+def main() -> int:
+    parser = build_parser()
+    args = parser.parse_args()
+    return args.func(args)
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())