|
# 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)
|