From 6f93326ba00843e98aba61d510c02bfd4a11630b Mon Sep 17 00:00:00 2001
From: wangsiyuan <2392948297@qq.com>
Date: Wed, 24 Jan 2024 16:00:25 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20website.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 website.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/website.py b/website.py
index 0a97edf..fcb75ad 100644
--- a/website.py
+++ b/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
     }