项目基础模块代码

app/service/project_service.py

@@ -1,10 +1,11 @@
 from app.model.bussiness_model import ProjectImage
-from app.model.schemas.project_info_schemas import ProjectInfoIn, ProjectInfoOut
 from app.model.bussiness_model import ProjectInfo
+from app.model.schemas.project_info_schemas import ProjectInfoIn, ProjectInfoOut
+from app.model.crud import project_info_crud as pic
+from app.model.crud import project_image_crud as pimc
+from app.util import os_utils as os
 from app.util import random_utils as ru
 from app.config.config_reader import datasets_url, runs_url, images_url
-from app.model.crud import project_info_crud as pic
-from app.util import os_utils as os
 
 from sqlalchemy.orm import Session
 from typing import List
@@ -30,21 +31,27 @@ def add_project(info: ProjectInfoIn, session: Session, user_id: int):
     return project_info
 
 
-def upload_project_image(session: Session, project_info: ProjectInfoOut, files: List[UploadFile]):
+def upload_project_image(project_info: ProjectInfoOut, files: List[UploadFile], session: Session):
     """
     上传项目的图片
     :param files: 上传的图片
     :param project_info: 项目信息
-    :param image:
     :param session:
     :return:
     """
+    images = []
     for file in files:
         image = ProjectImage()
         image.project_id = project_info.id
         # 保存原图
         path = os.save_images(images_url, project_info.project_no, file=file)
         image.image_url = path
+        # 生成缩略图
+        thumb_image_url = images_url + "\\thumb\\" + project_info.project_no + "\\" + ru.random_str(10) + ".jpg"
+        os.create_thumbnail(path, thumb_image_url)
+        image.thumb_image_url = thumb_image_url
+        images.append(image)
+    pimc.add_image_batch(images, session)