更新 website.py

master
wangsiyuan 2024-01-24 16:00:25 +08:00
parent fc8976efba
commit 6f93326ba0
1 changed files with 7 additions and 6 deletions

View File

@ -2,6 +2,7 @@ from fastapi import FastAPI, HTTPException, Request
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import FileResponse from fastapi.responses import FileResponse
from utils.pictures_handle import get_pictures_info from utils.pictures_handle import get_pictures_info
from utils.pictures_handle import get_total_pages
import os import os
from fastapi.staticfiles import StaticFiles from fastapi.staticfiles import StaticFiles
@ -38,17 +39,17 @@ async def get_menu():
@app.get("/api/get-photo-list") @app.get("/api/get-photo-list")
async def get_photo_list(): async def get_photo_list(page: int = 1, page_size: int = 10):
base_url = "http://127.0.0.1:5173/" dir_path = "pictures"
dir = "pictures" total_pages = get_total_pages(dir_path, page_size)
photo_list = get_pictures_info(dir) photo_list = get_pictures_info(dir_path, page, page_size)
for photo in photo_list: for photo in photo_list:
photo["photo_url"] = f"http://127.0.0.1:8000/static/{photo['photo_name']}" photo["photo_url"] = f"http://127.0.0.1:8000/static/{photo['photo_name']}"
# print(photo_list)
return { return {
"code": 0, "code": 0,
"msg": "ok", "msg": "ok",
"data": photo_list "data": photo_list,
"total_pages": total_pages
} }