main.py

# from chat.api import chat_router

# from db import database
# from game.api import game_router

import uvicorn
from goods.api import goods_router
from user.api import user_router
from order.api import order_router
from order_list.api import orderlist_router
from notify.api import notify_router

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from db import database, engine, metadata

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=['*'],
    allow_credentials=['*'],
    allow_methods=["*"],
    allow_headers=["*"],
)

metadata.create_all(engine)
app.state.database = database


@app.on_event("startup")
async def startup() -> None:
    database_ = app.state.database
    if not database_.is_connected:
        await database_.connect()


@app.on_event("shutdown")
async def shutdown() -> None:
    database_ = app.state.database
    if database_.is_connected:
        await database_.disconnect()


app.include_router(goods_router)
app.include_router(user_router)
app.include_router(order_router)
app.include_router(orderlist_router)
app.include_router(notify_router)


if __name__ == "__main__":
    uvicorn.run(app, use_colors=True, host="0.0.0.0")