Compare commits

...

2 Commits

Author SHA1 Message Date
wsy182 66078799ff Update website.py 2024-07-01 14:08:44 +08:00
wsy182 152bc9fd83 Update requirements.txt 2024-07-01 14:08:36 +08:00
2 changed files with 9 additions and 7 deletions

View File

@ -14,6 +14,5 @@ sniffio==1.3.0
starlette==0.32.0.post1
typing_extensions==4.9.0
uvicorn==0.25.0
uvloop==0.19.0
watchfiles==0.21.0
websockets==12.0

View File

@ -1,14 +1,16 @@
from fastapi import FastAPI, HTTPException, Request
import os
from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import FileResponse
from fastapi.staticfiles import StaticFiles
from utils.pictures_handle import get_pictures_info
from utils.pictures_handle import get_total_pages
import os
from fastapi.staticfiles import StaticFiles
app = FastAPI()
app.mount("/static", StaticFiles(directory="pictures"), name="static")
app.mount("/static", StaticFiles(directory="assets/images/pictures/"), name="static")
app.add_middleware(
CORSMiddleware,
allow_origins=["http://localhost:5173"], # 允许的源列表
@ -40,7 +42,7 @@ async def get_menu():
@app.get("/api/get-photo-list")
async def get_photo_list(page: int = 1, page_size: int = 10):
dir_path = "pictures"
dir_path = "assets/images/pictures"
total_pages = get_total_pages(dir_path, page_size)
photo_list = get_pictures_info(dir_path, page, page_size)
for photo in photo_list:
@ -56,8 +58,9 @@ async def get_photo_list(page: int = 1, page_size: int = 10):
@app.get("/static/{photo_name}")
async def get_photo(photo_name: str,request: Request):
request_path = str(request.url)
dir = "pictures"
dir = "/assets/images/pictures"
file_path = os.path.join(dir, photo_name)
print(file_path)
if os.path.exists(file_path):
return FileResponse(file_path)
else: