添加选择分片后端

DAPlatform/Dao/Relation.py

@@ -35,10 +35,10 @@ def create_relation(slice_id, user_id, data_id):
         Slice.add_annotation(slice_id)
         relation = Relation(slice_id=slice_id, user_id=user_id, data_id=data_id)
         relation.save()
-    except:
-        return 0
+    except Exception:
+        return Result(0, "create failed", {})
     else:
-        return 1
+        return Result(1, "create success", {})
 
 
 # 根据用户id找寻标注关系
@@ -46,9 +46,9 @@ def find_relationbyuserid(user_id):
     try:
         p = Relation.get(Relation.user_id == user_id)
     except Exception:
-        print(0)
+        return Result(0, "find failed", {})
     else:
-        print(p.email)
+        return Result(1, "find success", p.email)
 
 
 # 根据数据id找标注关系
@@ -58,9 +58,9 @@ def find_relationbydataid(data_id):
     try:
         p = Relation.get(Relation.data_id == data_id)
     except Exception:
-        print(0)
+        return Result(0, "find failed", {})
     else:
-        print(p.email)
+        return Result(1, "find success", p.email)
 
 
 if __name__ == '__main__':

DAPlatform/views/slice.py

 import json
 from flask import Blueprint
 from flask import request
-from Dao import Slice, Relation
+from Dao import Data, Slice, Relation
 
 slice = Blueprint("slice", __name__, url_prefix="/api/slice")
 
@@ -19,7 +19,27 @@ def get_pieces():
     return json.dumps(res)
 
 
-# # 用户选择想要标注的分片后，后端接收信息的接口
-# @slice.route("/selectPieces", methods=["Post"])
-# def select_pieces():
-#     # 接收前端传回的用户id与分片列表，将每个分片下的每一个文件创建一个relation
+# 用户选择想要标注的分片后，后端接收信息的接口
+@slice.route("/selectPieces", methods=["Post"])
+def select_pieces():
+    # 接收前端传回的用户id与分片列表，将每个分片下的每一个文件创建一个relation
+    userId = request.args['userId']
+    selected = request.args['selected']
+    print("select_pieces")
+    for sliceItem in selected:
+        dataList = Data.find_data_by_slice(sliceItem)
+        for dataItem in dataList:
+            ret = Relation.create_relation(sliceItem, userId, dataItem)
+            if ret.code == 0:
+                res = {
+                    'code': ret.code,
+                    'message': ret.message,
+                    'data': ret.data
+                }
+                return json.dumps(res)
+
+    res = {
+        'code': 1,
+        'message': "ok",
+    }
+    return json.dumps(res)
\ No newline at end of file

DAPlatform/views/task.py

@@ -51,7 +51,7 @@ def create_task():
     i = 0
     print(data)
     while i < len(data['task']['domains']):
-        s = Slice.create_text_slice(tid, data['task']['domains'][i]['piecesOptions'])
+        s = Slice.create_text_slice(tid, data['task']['domains'][i]['textModel'])
         j = 0
         while j < len(data['task']['domains'][i]['fileList']):
             Data.create_data(s.data, data['task']['domains'][i]['fileList'][j]['name'],