45 lines
984 B
Python
45 lines
984 B
Python
|
import os
|
||
|
|
|
||
|
|
import uvicorn
|
||
|
|
from fastapi import FastAPI
|
||
|
|
from fastapi.middleware.cors import CORSMiddleware
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
|
||
|
|
from Models.DailyModel import Daily
|
||
|
|
from Router import UserRouter, DailyRouter
|
||
|
|
from Utils.SqlAlchemyUtils import init_database, get_db_i
|
||
|
|
from fastapi.staticfiles import StaticFiles
|
||
|
|
|
||
|
|
init_database()
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
|
||
|
|
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, port=8006)
|