mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 04:40:45 +08:00
77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
"""Prune aged tool/trace noise from the assistant SQLite (or Postgres) DB.
|
|
|
|
Dry-run (default)::
|
|
|
|
python runtime/operations/scripts/prune_sqlite_retention.py --keep-days 30
|
|
|
|
Apply delete (keeps last N days of tool messages / tool_log / traces / sent outbound)::
|
|
|
|
python runtime/operations/scripts/prune_sqlite_retention.py --keep-days 30 --apply --yes
|
|
|
|
Also reclaim file space on SQLite (needs ~DB-size free disk; stop writers if possible)::
|
|
|
|
python runtime/operations/scripts/prune_sqlite_retention.py --keep-days 30 --apply --yes --vacuum
|
|
|
|
Point at a specific file::
|
|
|
|
set OPS_ASSISTANT_DB_PATH=D:\\path\\ai_ops.sqlite
|
|
python runtime/operations/scripts/prune_sqlite_retention.py --keep-days 14 --dry-run
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
def main() -> int:
|
|
# Allow `python runtime/operations/scripts/...` from repo root without PYTHONPATH.
|
|
root = Path(__file__).resolve().parents[3]
|
|
if str(root) not in sys.path:
|
|
sys.path.insert(0, str(root))
|
|
|
|
p = argparse.ArgumentParser(description=__doc__)
|
|
p.add_argument("--keep-days", type=int, default=30, help="Retain rows newer than this many days.")
|
|
p.add_argument("--dry-run", action="store_true", default=False, help="Force dry-run (default).")
|
|
p.add_argument("--apply", action="store_true", help="Actually delete (requires --yes).")
|
|
p.add_argument("--yes", action="store_true", help="Confirm destructive delete.")
|
|
p.add_argument("--vacuum", action="store_true", help="SQLite VACUUM after delete.")
|
|
p.add_argument("--include-scheduled-runs", action="store_true", help="Also prune old scheduled_job_run.")
|
|
p.add_argument("--no-outbound", action="store_true", help="Skip channel_outbound_message prune.")
|
|
p.add_argument("--db", default="", help="Override OPS_ASSISTANT_DB_PATH for this run.")
|
|
args = p.parse_args()
|
|
|
|
if str(args.db or "").strip():
|
|
os.environ["OPS_ASSISTANT_DB_PATH"] = str(args.db).strip()
|
|
os.environ["AIA_ASSISTANT_DB_BACKEND"] = "sqlite"
|
|
|
|
from svc.persistence.assistant_store import get_assistant_store, reset_assistant_store_singleton
|
|
from svc.persistence.sqlite_retention import prune_sqlite_retention
|
|
|
|
reset_assistant_store_singleton()
|
|
store = get_assistant_store()
|
|
|
|
dry_run = not bool(args.apply)
|
|
if args.dry_run:
|
|
dry_run = True
|
|
if args.apply and not args.yes:
|
|
print("Refusing --apply without --yes", file=sys.stderr)
|
|
return 2
|
|
|
|
out = prune_sqlite_retention(
|
|
store,
|
|
keep_days=int(args.keep_days),
|
|
dry_run=dry_run,
|
|
vacuum=bool(args.vacuum) and not dry_run,
|
|
include_outbound_sent=not bool(args.no_outbound),
|
|
include_scheduled_runs=bool(args.include_scheduled_runs),
|
|
)
|
|
print(json.dumps(out, ensure_ascii=False, indent=2))
|
|
return 0 if out.get("ok") else 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|