init_db.py

#!/usr/bin/env python3
"""Initialize the database and create tables."""
import sys
from pathlib import Path

# Add src to path
sys.path.insert(0, str(Path(__file__).parent / 'src'))

from storage.psql import db
from core.logger import logger

def main():
    """Initialize database tables."""
    try:
        logger.logger.info("Initializing database...")
        db.init_db()
        logger.logger.info("✓ Database initialized successfully!")
        logger.logger.info("Tables created:")
        logger.logger.info("  - classes")
        logger.logger.info("  - quizzes")
        logger.logger.info("  - quiz_class_association")
        return 0
    except Exception as e:
        logger.logger.error(f"✗ Failed to initialize database: {str(e)}")
        return 1

if __name__ == "__main__":
    sys.exit(main())