Spaces:
Running
Running
| 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() | |