import os

import uvicorn
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from sqlalchemy.orm import Session

from Crud.UserCrud import load_company_config_to_db
from Models.DailyModel import Daily
from Router import UserRouter, DailyRouter,AuthRouter
from Utils.DataToDbUtils import user_table_to_db
from Utils.SqlAlchemyUtils import init_database, get_db_i
from fastapi.staticfiles import StaticFiles

init_database()
# load_company_config_to_db()
# user_table_to_db()
app = FastAPI(
    title="日报小程序",
    description="",
    version="v1.0.0"
)
if not os.path.exists('static_data'):
    os.mkdir('static_data')
app.mount("/static_data", StaticFiles(directory="static_data"), name="static_data")
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.include_router(UserRouter.router)
app.include_router(DailyRouter.router)
app.include_router(AuthRouter.router)


def test():
    from Utils.SqlAlchemyUtils import get_db
    db: Session = get_db_i()
    c = db.query(Daily).all()
    e = 2
    db.close()


# test()
# uvicorn.run(app=app, host="0.0.0.0", port=8006)