前后端交互完成

app/service/project_service.py

@@ -1,8 +1,9 @@
-from app.model.bussiness_model import ProjectImage
-from app.model.bussiness_model import ProjectInfo
+from app.model.bussiness_model import ProjectImage, ProjectInfo, ProjectImgLeafer, ProjectImgLabel
 from app.model.schemas.project_info_schemas import ProjectInfoIn, ProjectInfoOut
+from app.model.schemas.project_image_schemas import ProjectImgLeaferLabel, ProjectImgLeaferOut
 from app.model.crud import project_info_crud as pic
 from app.model.crud import project_image_crud as pimc
+from app.model.crud import project_img_leafer_label_crud as pillc
 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
@@ -54,5 +55,31 @@ def upload_project_image(project_info: ProjectInfoOut, files: List[UploadFile],
     pimc.add_image_batch(images, session)
 
 
+def save_img_label(img_leafer_label: ProjectImgLeaferLabel, session: Session):
+    """
+    保存图片的标签框选信息，每次保存都会针对图片的信息全部删除，然后重新保存
+    :param img_leafer_label:
+    :param session:
+    :return:
+    """
+    img_leafer = ProjectImgLeafer(**img_leafer_label)
+    pillc.save_img_leafer(img_leafer, session)
+    label_infos = img_leafer_label.label_infos
+    img_labels = []
+    for label_info in label_infos:
+        img_label = ProjectImgLabel(**label_info.dict())
+        img_label.image_id = img_leafer_label.image_id
+        img_labels.append(img_label)
+    pillc.save_img_label_batch(img_leafer_label.image_id, img_labels, session)
 
 
+def get_img_leafer(image_id: int, session: Session):
+    """
+    根据图片id查询图片的leafer信息
+    :param image_id:
+    :param session:
+    :return:
+    """
+    img_leafer = pillc.get_img_leafer(image_id, session)
+    img_leafer_out = ProjectImgLeaferOut.from_orm(img_leafer).dict()
+    return img_leafer_out