chore: initialize tuhui repository
This commit is contained in:
65
backend/app/main.py
Normal file
65
backend/app/main.py
Normal file
@@ -0,0 +1,65 @@
|
||||
from fastapi import FastAPI, Request, Response
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from fastapi.staticfiles import StaticFiles
|
||||
from app.core.config import settings
|
||||
from app.core.database import Base, engine
|
||||
from app.api import auth, works, orders, payment, upload
|
||||
|
||||
# 创建数据库表
|
||||
Base.metadata.create_all(bind=engine)
|
||||
|
||||
# 创建 FastAPI 应用
|
||||
app = FastAPI(
|
||||
title=settings.PROJECT_NAME,
|
||||
version=settings.VERSION,
|
||||
description="图绘 - 设计作品交易平台 API"
|
||||
)
|
||||
|
||||
# 配置 CORS
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
allow_origins=settings.ALLOWED_ORIGINS,
|
||||
allow_credentials=True,
|
||||
allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
|
||||
allow_headers=["*"],
|
||||
expose_headers=["*"],
|
||||
max_age=600,
|
||||
)
|
||||
|
||||
# 添加全局 CORS 中间件
|
||||
@app.middleware("http")
|
||||
async def add_cors_headers(request: Request, call_next):
|
||||
response = await call_next(request)
|
||||
origin = request.headers.get("origin", "")
|
||||
if origin in settings.ALLOWED_ORIGINS:
|
||||
response.headers["Access-Control-Allow-Origin"] = origin
|
||||
response.headers["Access-Control-Allow-Credentials"] = "true"
|
||||
response.headers["Access-Control-Allow-Methods"] = "GET, POST, PUT, DELETE, OPTIONS"
|
||||
response.headers["Access-Control-Allow-Headers"] = "*"
|
||||
|
||||
if request.method == "OPTIONS":
|
||||
response.headers["Access-Control-Max-Age"] = "600"
|
||||
|
||||
return response
|
||||
|
||||
# 挂载静态文件目录
|
||||
app.mount("/uploads", StaticFiles(directory="uploads"), name="uploads")
|
||||
|
||||
# 注册路由
|
||||
app.include_router(auth.router, prefix=settings.API_PREFIX)
|
||||
app.include_router(works.router, prefix=settings.API_PREFIX)
|
||||
app.include_router(orders.router, prefix=settings.API_PREFIX)
|
||||
app.include_router(payment.router, prefix=settings.API_PREFIX)
|
||||
app.include_router(upload.router, prefix=settings.API_PREFIX)
|
||||
|
||||
@app.get("/")
|
||||
def root():
|
||||
return {
|
||||
"message": "欢迎使用图绘 API",
|
||||
"version": settings.VERSION,
|
||||
"docs": "/docs"
|
||||
}
|
||||
|
||||
@app.get("/health")
|
||||
def health_check():
|
||||
return {"status": "healthy"}
|
||||
Reference in New Issue
Block a user