更新 website.py

website.py

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