| from __future__ import annotations |
|
|
| import argparse |
| import os |
| from pathlib import Path |
| from zoneinfo import ZoneInfo, ZoneInfoNotFoundError |
|
|
| from .core import build_report |
| from .render import write_reports |
|
|
|
|
| def main(argv: list[str] | None = None) -> int: |
| parser = _build_parser() |
| args = parser.parse_args(argv) |
|
|
| try: |
| ZoneInfo(args.tz) |
| except ZoneInfoNotFoundError: |
| parser.error(f"unknown timezone: {args.tz}") |
|
|
| report = build_report( |
| state_path=Path(args.state).expanduser(), |
| sessions_dir=Path(args.sessions_dir).expanduser(), |
| tz_name=args.tz, |
| days=args.days, |
| top=args.top, |
| ) |
| written = write_reports(report, Path(args.out).expanduser(), args.format) |
|
|
| for path in written: |
| print(f"Wrote {path}") |
| print(f"Sessions: {len(report.sessions):,}") |
| print(f"Total tokens: {report.total_tokens:,}") |
| if report.warnings: |
| print(f"Warnings: {len(report.warnings):,}") |
| return 0 |
|
|
|
|
| def _build_parser() -> argparse.ArgumentParser: |
| parser = argparse.ArgumentParser( |
| prog="codex-usage-profiler", |
| description="Summarize local Codex usage from state_5.sqlite and rollout JSONL files.", |
| ) |
| parser.add_argument( |
| "--state", |
| default="~/.codex/state_5.sqlite", |
| help="Path to Codex state_5.sqlite. Default: %(default)s", |
| ) |
| parser.add_argument( |
| "--sessions-dir", |
| default="~/.codex/sessions", |
| help="Root directory containing rollout JSONL files. Default: %(default)s", |
| ) |
| parser.add_argument( |
| "--out", |
| default="reports", |
| help="Output directory. Default: %(default)s", |
| ) |
| parser.add_argument( |
| "--format", |
| choices=["md", "html", "both"], |
| default="both", |
| help="Report format to write. Default: %(default)s", |
| ) |
| parser.add_argument( |
| "--tz", |
| default=os.environ.get("TZ", "Europe/Berlin"), |
| help="Timezone for day/week grouping. Default: %(default)s", |
| ) |
| parser.add_argument( |
| "--days", |
| type=int, |
| default=None, |
| help="Include the last N local calendar days including today.", |
| ) |
| parser.add_argument( |
| "--top", |
| type=int, |
| default=20, |
| help="Number of top sessions to include. Default: %(default)s", |
| ) |
| return parser |
|
|
