File size: 711 Bytes
021e065
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from backend.database.postgres.db import SessionLocal
from backend.database.postgres.models import ApiKey
import os

def seed_api_key():
    db = SessionLocal()
    key_val = os.getenv("API_KEY", "dev_key_123")
    
    existing = db.query(ApiKey).filter(ApiKey.key == key_val).first()
    if not existing:
        new_key = ApiKey(
            key=key_val,
            owner="Default Developer",
            tier="premium" # Unlimited for dev
        )
        db.add(new_key)
        db.commit()
        print(f"--- API SECURITY: Seeded developer key: {key_val} ---")
    else:
        print(f"--- API SECURITY: Developer key already exists ---")
    db.close()

if __name__ == "__main__":
    seed_api_key()