87 lines
2.2 KiB
Python
87 lines
2.2 KiB
Python
# app/core/lifespan.py
|
|
"""Summary description ....
|
|
|
|
This module provides classes and functions for:
|
|
- ....
|
|
- ....
|
|
- ....
|
|
"""
|
|
|
|
import logging
|
|
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
from sqlalchemy import text
|
|
|
|
from middleware.auth import AuthManager
|
|
from models.database_models import Base
|
|
|
|
from .database import SessionLocal, engine
|
|
from .logging import setup_logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
auth_manager = AuthManager()
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""Application lifespan events."""
|
|
# Startup
|
|
app_logger = setup_logging() # Configure logging first
|
|
app_logger.info("Starting up ecommerce API")
|
|
|
|
# Create tables
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
# Create indexes
|
|
create_indexes()
|
|
|
|
# Create default admin user
|
|
db = SessionLocal()
|
|
try:
|
|
auth_manager.create_default_admin_user(db)
|
|
except Exception as e:
|
|
logger.error(f"Failed to create default admin user: {e}")
|
|
finally:
|
|
db.close()
|
|
|
|
yield
|
|
|
|
# Shutdown
|
|
app_logger.info("Shutting down ecommerce API")
|
|
|
|
|
|
def create_indexes():
|
|
"""Create database indexes."""
|
|
with engine.connect() as conn:
|
|
try:
|
|
# User indexes
|
|
conn.execute(
|
|
text("CREATE INDEX IF NOT EXISTS idx_user_email ON users(email)")
|
|
)
|
|
conn.execute(
|
|
text("CREATE INDEX IF NOT EXISTS idx_user_username ON users(username)")
|
|
)
|
|
|
|
# Product indexes
|
|
conn.execute(
|
|
text("CREATE INDEX IF NOT EXISTS idx_product_gtin ON products(gtin)")
|
|
)
|
|
conn.execute(
|
|
text(
|
|
"CREATE INDEX IF NOT EXISTS idx_product_marketplace ON products(marketplace)"
|
|
)
|
|
)
|
|
|
|
# Stock indexes
|
|
conn.execute(
|
|
text(
|
|
"CREATE INDEX IF NOT EXISTS idx_stock_gtin_location ON stock(gtin, location)"
|
|
)
|
|
)
|
|
|
|
conn.commit()
|
|
logger.info("Database indexes created successfully")
|
|
except Exception as e:
|
|
logger.warning(f"Index creation warning: {e}")
|