#!/usr/bin/env python3
from database import Base, engine, SessionLocal
from models import APIKey
from crud import create_api_key

def setup_admin():
    Base.metadata.create_all(bind=engine)
    db = SessionLocal()
    existing = db.query(APIKey).filter(APIKey.name == "default").first()
    if not existing:
        _, raw_key = create_api_key(
            db,
            name="default",
            daily_tokens=1_000_000,
            monthly_tokens=10_000_000,
            daily_requests=1000,
            monthly_requests=10000,
        )
        print(f"API Key created: {raw_key}")
    else:
        print("Default API key already exists")
    db.close()

if __name__ == "__main__":
    setup_admin()