项目基础模块代码

app/model/crud/project_image_crud.py

@@ -1,5 +1,6 @@
 from sqlalchemy.orm import Session
 from sqlalchemy import asc
+from typing import List
 
 from app.model.bussiness_model import ProjectImage as piModel
 from app.model.schemas.project_image_schemas import ProjectImage, ProjectImagePager
@@ -13,9 +14,9 @@ def get_image_pager(image: ProjectImagePager, session: Session):
     return pager
 
 
-def get_image_list(image: ProjectImage, session: Session):
-    query = session.query(piModel).filter(piModel.project_id == image.project_id).order_by(asc(piModel.id))
-    image_list =  [ProjectImage.from_orm(image) for image in query.all()]
+def get_image_list(project_id: int, session: Session):
+    query = session.query(piModel).filter(piModel.project_id == project_id).order_by(asc(piModel.id))
+    image_list = [ProjectImage.from_orm(image).dict() for image in query.all()]
     return image_list
 
 
@@ -25,8 +26,14 @@ def add_image(image: ProjectImage, session: Session):
     return image
 
 
-def del_image(id: str, session: Session):
-    row_del = session.query(piModel).filter_by(id=id).delete()
+def add_image_batch(images: List[ProjectImage], session: Session):
+    for image in images:
+        session.add(image)
+    session.commit()
+
+
+def del_image(image_id: str, session: Session):
+    row_del = session.query(piModel).filter_by(id=image_id).delete()
     session.commit()
     return row_del