修改slice

DAPlatform/Dao/Slice.py

+from Dao import Task
+
 from peewee import *
 
 mysql = MySQLDatabase(
@@ -37,6 +39,7 @@ class Slice(Model):
 # 创建文本分片 文本分片类型值为0
 def create_text_slice(task_id, model_id):
     try:
+        Task.add_slice(task_id)
         slice = Slice(task_id=task_id, model_id=model_id, type=0)
         slice.save()
     except:
@@ -48,6 +51,7 @@ def create_text_slice(task_id, model_id):
 # 创建图片分片 图片分片类型值为1
 def create_pic_slice(task_id, description):
     try:
+        Task.add_slice(task_id)
         slice = Slice(task_id=task_id, illustration=description, type=1)
         slice.save()
     except:
@@ -59,6 +63,7 @@ def create_pic_slice(task_id, description):
 # 创建图层分片 图层分片类型值为2
 def create_layer_slice(task_id, description):
     try:
+        Task.add_slice(task_id)
         slice = Slice(task_id=task_id, illustration=description, type=2)
         slice.save()
     except:
@@ -78,8 +83,5 @@ def delete_slice(id):
 
 
 if __name__ == '__main__':
-    Slice.drop_table()
-    # 建表
-    Slice.create_table()
-    create_text_slice(1, 5)
-    create_pic_slice(1, "ha")
+    create_pic_slice(6, "标注坐标（30，100）")
+    create_layer_slice(7, "。。。。。。")

DAPlatform/Dao/Task.py

@@ -126,6 +126,16 @@ def delete_task(task_id):
         return Result(1, "delete success", {}, 0)
 
 
+# 根据id查找任务
+def find_task_by_id(task_id):
+    try:
+        t = Task.get(Task.task_id == task_id)
+    except:
+        return t
+    else:
+        return t
+
+
 # 添加分片
 def add_slice(task_id):
     try:
@@ -139,8 +149,4 @@ def add_slice(task_id):
 
 
 if __name__ == '__main__':
-    # 建表
-    # Task.create_table()
-    # 添加数据
-    # Task.create_task(publish_id, task_name)
-    add_slice(1)
+    delete_task(10)
\ No newline at end of file

DAPlatform/Dao/User.py

@@ -167,5 +167,6 @@ def find_id(id):
 
 
 if __name__ == '__main__':
-    print(find_id(1))
+    register('456', '456', '456')
+    set_type(4, 1)