268214d971
- Migrated from HIVE branding to WHOOSH across all components - Enhanced backend API with new services: AI models, BZZZ integration, templates, members - Added comprehensive testing suite with security, performance, and integration tests - Improved frontend with new components for project setup, AI models, and team management - Updated MCP server implementation with WHOOSH-specific tools and resources - Enhanced deployment configurations with production-ready Docker setups - Added comprehensive documentation and setup guides - Implemented age encryption service and UCXL integration 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
126 lines
3.8 KiB
Python
126 lines
3.8 KiB
Python
"""
|
|
Database initialization script for WHOOSH platform.
|
|
Creates all tables and sets up initial data.
|
|
"""
|
|
|
|
import logging
|
|
from sqlalchemy.orm import Session
|
|
from app.core.database import engine, SessionLocal, Base
|
|
from app.models.user import User
|
|
from app.models.auth import API_SCOPES, APIKey
|
|
|
|
# Import all models to ensure they're registered with Base
|
|
from app.models import user, auth, agent, project
|
|
|
|
def create_tables():
|
|
"""Create all database tables."""
|
|
try:
|
|
# Create all tables using the unified Base
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
# Add other model bases here
|
|
# WorkflowsBase.metadata.create_all(bind=engine)
|
|
# AgentsBase.metadata.create_all(bind=engine)
|
|
|
|
logging.info("Database tables created successfully")
|
|
return True
|
|
except Exception as e:
|
|
logging.error(f"Failed to create database tables: {e}")
|
|
return False
|
|
|
|
|
|
def create_initial_user(db: Session):
|
|
"""Create initial admin user if none exists."""
|
|
try:
|
|
# Check if any users exist
|
|
user_count = db.query(User).count()
|
|
if user_count > 0:
|
|
logging.info("Users already exist, skipping initial user creation")
|
|
return True
|
|
|
|
# Create initial admin user
|
|
admin_user = User(
|
|
username="admin",
|
|
email="admin@whoosh.local",
|
|
full_name="WHOOSH Administrator",
|
|
hashed_password=User.hash_password("admin123"), # Change this!
|
|
is_active=True,
|
|
is_superuser=True,
|
|
is_verified=True
|
|
)
|
|
|
|
db.add(admin_user)
|
|
db.commit()
|
|
db.refresh(admin_user)
|
|
|
|
logging.info("Initial admin user created: admin/admin123")
|
|
logging.warning("SECURITY: Please change the default admin password!")
|
|
|
|
# Create initial API key for the admin user
|
|
from app.core.security import APIKeyManager
|
|
plain_key, hashed_key, prefix = APIKeyManager.generate_api_key()
|
|
|
|
admin_api_key = APIKey(
|
|
user_id=admin_user.id,
|
|
name="Default Admin API Key",
|
|
key_hash=hashed_key,
|
|
key_prefix=prefix,
|
|
is_active=True
|
|
)
|
|
admin_api_key.set_scopes(["admin"])
|
|
|
|
db.add(admin_api_key)
|
|
db.commit()
|
|
|
|
logging.info(f"Initial admin API key created: {plain_key}")
|
|
logging.warning("SECURITY: Save this API key securely, it won't be shown again!")
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
logging.error(f"Failed to create initial user: {e}")
|
|
db.rollback()
|
|
return False
|
|
|
|
|
|
def initialize_database():
|
|
"""Initialize the complete database."""
|
|
logging.info("Starting database initialization...")
|
|
|
|
# Create tables
|
|
if not create_tables():
|
|
return False
|
|
|
|
# Create initial data
|
|
db = SessionLocal()
|
|
try:
|
|
# Create initial admin user
|
|
if not create_initial_user(db):
|
|
return False
|
|
|
|
logging.info("Database initialization completed successfully")
|
|
return True
|
|
|
|
except Exception as e:
|
|
logging.error(f"Database initialization failed: {e}")
|
|
return False
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Configure logging
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
|
)
|
|
|
|
# Initialize database
|
|
success = initialize_database()
|
|
if success:
|
|
print("✅ Database initialization completed successfully")
|
|
print("🔑 Default admin credentials: admin/admin123")
|
|
print("⚠️ SECURITY: Please change the default password immediately!")
|
|
else:
|
|
print("❌ Database initialization failed")
|
|
exit(1) |