37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from pathlib import Path
|
|
|
|
from fastapi import FastAPI, Request, Response
|
|
from fastapi.staticfiles import StaticFiles
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
|
|
from app.config import UPLOAD_DIR
|
|
from app.database import init_db
|
|
from app.routers import web, api
|
|
|
|
app = FastAPI(title="Mailova 星笺")
|
|
|
|
UPLOAD_DIR.mkdir(exist_ok=True)
|
|
app.mount("/static", StaticFiles(directory=str(Path(__file__).resolve().parent / "static")), name="static")
|
|
app.mount("/uploads", StaticFiles(directory=str(UPLOAD_DIR)), name="uploads")
|
|
|
|
|
|
# Browser cache: 30 days for uploaded images (they never change)
|
|
class CacheControlMiddleware(BaseHTTPMiddleware):
|
|
async def dispatch(self, request: Request, call_next):
|
|
response: Response = await call_next(request)
|
|
if request.url.path.startswith("/uploads/"):
|
|
response.headers["Cache-Control"] = "public, max-age=2592000, immutable"
|
|
return response
|
|
|
|
|
|
app.add_middleware(CacheControlMiddleware)
|
|
|
|
app.include_router(web.router)
|
|
app.include_router(api.router)
|
|
|
|
init_db()
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run("app.main:app", host="0.0.0.0", port=8000, reload=True)
|