main.py 948 B

1234567891011121314151617181920212223242526272829303132
  1. from fastapi import FastAPI
  2. from fastapi.staticfiles import StaticFiles
  3. from fastapi.middleware.cors import CORSMiddleware
  4. from app.api import storage
  5. from app.core.config import settings
  6. import os
  7. # 确保启动时创建目录
  8. if not os.path.exists(settings.UPLOAD_PATH):
  9. os.makedirs(settings.UPLOAD_PATH)
  10. app = FastAPI(title="Simple Image Server", version="1.0.0")
  11. # 1. 挂载静态目录
  12. # 这样访问 http://192.168.77.249:7733/static/xxx.jpg 就能直接看到图片
  13. app.mount("/static", StaticFiles(directory=settings.UPLOAD_PATH), name="static")
  14. # 2. 设置 CORS (允许所有来源,方便调试)
  15. app.add_middleware(
  16. CORSMiddleware,
  17. allow_origins=["*"],
  18. allow_credentials=True,
  19. allow_methods=["*"],
  20. allow_headers=["*"],
  21. )
  22. # 3. 注册路由
  23. app.include_router(storage.router, prefix="/api")
  24. @app.get("/")
  25. async def root():
  26. return {"message": "Image Server is running", "docs": f"{settings.BASE_URL}/docs"}