#!/usr/bin/env python3
"""P2Z-I: Baseline erfassen vor den 3 Test-Writes."""
import sqlite3, json, urllib.request, datetime

def credits():
    key = ""
    with open("/etc/graphiti/graphiti.env") as f:
        for line in f:
            if line.startswith("OPENROUTER_API_KEY="):
                key = line.split("=", 1)[1].strip()
                break
    req = urllib.request.Request("https://openrouter.ai/api/v1/credits",
                                 headers={"Authorization": f"Bearer {key}"})
    return json.loads(urllib.request.urlopen(req, timeout=30).read())["data"]

k = sqlite3.connect('/opt/struktur/youtube-research/knowledge.db')
for st, c in k.execute("select graphiti_status, count(*) from graphiti_import_queue where reconciliation_record_id like 'aa043:%' group by 1"):
    print(f"queue {st}: {c}")

s = sqlite3.connect('/opt/struktur/obsidian-graphiti-import/obsidian_graphiti_import.db')
n_done_files = s.execute("select count(*) from files where status='done'").fetchone()[0]
print(f"files done: {n_done_files}")

episodes = json.loads(urllib.request.urlopen('http://127.0.0.1:8644/episodes/inventory', timeout=60).read())['episodes']
aa = [e for e in episodes if (e.get('name') or '').startswith('aa043_')]
print(f"Graphiti Episoden total: {len(episodes)} | aa043: {len(aa)}")

u = credits()['total_usage']
print(f"OpenRouter usage BEFORE: {u}")
print(f"MEASUREMENT_START={datetime.datetime.now(datetime.timezone.utc).isoformat()}")