限制用户多次选择同一文件

DAPlatform/dao/Relation.py

@@ -31,9 +31,11 @@ class Relation(Model):
 # 创建标注关系
 def create_relation(slice_id, user_id, data_id):
     try:
-        Slice.add_annotation(slice_id)
-        relation = Relation(slice_id=slice_id, user_id=user_id, data_id=data_id)
-        relation.save()
+        r = Relation.get(slice_id=slice_id, user_id=user_id, data_id=data_id)
+        if not r:
+            Slice.add_annotation(slice_id)
+            relation = Relation(slice_id=slice_id, user_id=user_id, data_id=data_id)
+            relation.save()
     except:
         return Result(0, "create failed", {})
     else:
@@ -215,7 +217,8 @@ def get_relation_id(uid, did):
 
 
 if __name__ == '__main__':
-    Relation.drop_table()
-    Relation.create_table()
+    # Relation.drop_table()
+    # Relation.create_table()
+    Relation.delete().where(Relation.relation_id == 67).execute()
     # print(get_state(1, 5))
     # create_relation(3, 1, 2)