daily/main.py

49 lines
1.2 KiB
Python

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)