refactor: split cli.py into package (ref.md 3.7)

2026-07-04 19:17:59 +00:00
parent 574b076f5d
commit d556b02bfb
15 changed files with 1251 additions and 1080 deletions
@@ -0,0 +1,66 @@
+# retoor <retoor@molodetz.nl>
+
+import sys
+from devplacepy.database import get_table
+from devplacepy.cli._shared import _audit_cli
+
+
+def cmd_devii_reset_quota(args):
+    from devplacepy.database import db
+
+    table_name = "devii_usage_ledger"
+    if table_name not in db.tables:
+        print(f"Table '{table_name}' does not exist, nothing to reset")
+        return
+    table = db[table_name]
+    if args.all:
+        count = table.count()
+        table.delete()
+        _audit_cli("cli.devii.quota.reset", "CLI reset AI quota (all)", metadata={"scope": "all", "rows_removed": count})
+        print(f"Reset all AI quotas ({count} ledger rows deleted)")
+        return
+    if args.guests:
+        count = table.count(owner_kind="guest")
+        table.delete(owner_kind="guest")
+        _audit_cli("cli.devii.quota.reset", "CLI reset AI quota (guests)", metadata={"scope": "guests", "rows_removed": count})
+        print(f"Reset all guest AI quotas ({count} ledger rows deleted)")
+        return
+    if not args.username:
+        print("Provide a username, or --guests, or --all")
+        sys.exit(1)
+    user = get_table("users").find_one(username=args.username)
+    if not user:
+        print(f"User '{args.username}' not found")
+        sys.exit(1)
+    count = table.count(owner_kind="user", owner_id=user["uid"])
+    table.delete(owner_kind="user", owner_id=user["uid"])
+    from devplacepy.services.audit import record as audit
+
+    _audit_cli(
+        "cli.devii.quota.reset",
+        f"CLI reset AI quota for {args.username}",
+        metadata={"scope": "user", "rows_removed": count},
+        target_type="user",
+        target_uid=user["uid"],
+        target_label=args.username,
+        links=[audit.target("user", user["uid"], args.username)],
+    )
+    print(f"Reset AI quota for '{args.username}' ({count} ledger rows deleted)")
+
+
+def register_devii(subparsers):
+    devii = subparsers.add_parser("devii", help="Devii assistant management")
+    devii_sub = devii.add_subparsers(title="action", dest="action")
+    devii_reset = devii_sub.add_parser(
+        "reset-quota", help="Reset the rolling 24h AI spend quota"
+    )
+    devii_reset.add_argument(
+        "username", nargs="?", help="Reset the quota for a single user"
+    )
+    devii_reset.add_argument(
+        "--guests", action="store_true", help="Reset every guest quota"
+    )
+    devii_reset.add_argument(
+        "--all", action="store_true", help="Reset every quota (users and guests)"
+    )
+    devii_reset.set_defaults(func=cmd_devii_reset_quota)