main.py

from loguru import logger
import casbin
from fastapi import FastAPI
from starlette.middleware.cors import CORSMiddleware

from core.config import settings
from api.api_v1.api import api_router
from middleware.auto_db_session import DBSessionMiddleware
from middleware.authentication import BearerAuthenticationMiddleware
from middleware.casbinMiddleware import CasbinMiddleware
from middleware.secureHttpsMiddleware import MessageSecureHTTPMiddleware
from starlette.middleware.sessions import SessionMiddleware
app = FastAPI(
    title=settings.PROJECT_NAME,
    openapi_url=f'{settings.API_V1_STR}/openapi.json',
    docs_url='/docs',
    redoc_url='/redoc',
)

@app.on_event("startup")
async def init_scheduler():
    from libs.rsalib import generateRsaKeySave
    await generateRsaKeySave()

app.add_middleware(MessageSecureHTTPMiddleware)
app.add_middleware(SessionMiddleware,secret_key="example",max_age=1 * 24 * 60 * 60)
enforcer = casbin.Enforcer('core/rbac_model.conf', 'core/rbac_policy.csv')
app.add_middleware(CasbinMiddleware, enforcer=enforcer)
app.add_middleware(BearerAuthenticationMiddleware)  # auth middleware
app.add_middleware(DBSessionMiddleware)  # auto db session manage middleware
if settings.BACKEND_CORS_ORIGINS:
    app.add_middleware(
        CORSMiddleware,
        allow_origins=settings.BACKEND_CORS_ORIGINS,
        allow_credentials=True,
        allow_methods=["*"],
        allow_headers=["*"],
    )

# log config
# logger.remove(handler_id=None)
# logger.add(sink=f'logs/{settings.PROJECT_NAME}-{{time:YYYY-MM-DD}}.log',
#            format="{time:YYYY-MM-DD HH:mm:ss}-{level}-{name}:{function}:{line}-{level}-{message}",
#            level=settings.LOG_LEVEL,
#            enqueue=True,
#            diagnose=True,
#            retention="10 days",
#            rotation="24h",
#            encoding='utf-8',
#            # compression='zip'
#            )


# app.include_router(api_router)

# V1
app.include_router(api_router, prefix=settings.API_V1_STR)


if __name__ == '__main__':
    import uvicorn
    uvicorn.run(app, host='127.0.0.1', port=8000)