修改后端数据库结构

DAPlatform/Dao/Data.py

 from peewee import *
+
 mysql = MySQLDatabase(
     'sys',
     host='127.0.0.1',
@@ -9,35 +10,37 @@ mysql = MySQLDatabase(
 
 mysql.connect()
 
+
 class Data(Model):
-    name = CharField(null=False)
-    url = CharField(null=False,unique=True)
+    data_id = AutoField(primary_key=True, unique=True)
+    slice_id = IntegerField(null=False)
+    url = CharField(null=False, unique=True)
+
     class Meta:
         database = mysql
 
-#向数据库中添加原始数据
-def create_data(name,url):
-    if not Data.table_exists():
-        Data.create_table()
+
+# 向数据库中添加原始数据
+def create_data(slice_id, url):
     try:
-        data = Data(name=name,url=url)
+        data = Data(slice_id=slice_id, url=url)
         data.save()
     except Exception:
         print(0)
     else:
         print(1)
 
-#根据数据id寻找数据url
+
+# 根据数据id寻找数据url
 def find_data(id):
-    if not Data.table_exists():
-        Data.create_table()
     try:
-        data = Data.get(Data.id == id)
+        data = Data.get(Data.data_id == id)
     except Exception:
         print(0)
     else:
         print(data.url)
 
+
 if __name__ == '__main__':
-    create_data('ha','123')
-    find_data(1)
+    Data.drop_table()
+    Data.create_table()
\ No newline at end of file

DAPlatform/Dao/Relation.py

 from peewee import *
-from Dao import Slice, User, Data
 import datetime
 
 mysql = MySQLDatabase(

DAPlatform/Dao/Slice.py

@@ -28,16 +28,38 @@ class Slice(Model):
     illustration = TextField(null=True)
     Annotaion_number = IntegerField(null=False, default=0)
     document_number = IntegerField(null=False, default=0)
-    type = FixedCharField(null=False)
+    type = FixedCharField(null=0)
 
     class Meta:
         database = mysql
 
 
-# 创建分片
-def create_slice(task_id, model_id, document_number, type):
+# 创建文本分片 文本分片类型值为0
+def create_text_slice(task_id, model_id):
     try:
-        slice = Slice(task_id=task_id, model_id=model_id, document_number=document_number, type=type)
+        slice = Slice(task_id=task_id, model_id=model_id, type=0)
+        slice.save()
+    except:
+        return Result(0, "create failed", {})
+    else:
+        return Result(1, "create success", slice.slice_id)
+
+
+# 创建图片分片 图片分片类型值为1
+def create_pic_slice(task_id, description):
+    try:
+        slice = Slice(task_id=task_id, illustration=description, type=1)
+        slice.save()
+    except:
+        return Result(0, "create failed", {})
+    else:
+        return Result(1, "create success", slice.slice_id)
+
+
+# 创建图层分片 图层分片类型值为2
+def create_layer_slice(task_id, description):
+    try:
+        slice = Slice(task_id=task_id, illustration=description, type=2)
         slice.save()
     except:
         return Result(0, "create failed", {})
@@ -59,4 +81,5 @@ if __name__ == '__main__':
     Slice.drop_table()
     # 建表
     Slice.create_table()
-    create_slice(1, 5, 10, False)
+    create_text_slice(1, 5)
+    create_pic_slice(1, "ha")

DAPlatform/Dao/Task.py

 from peewee import *
-from Dao import User
 import datetime
 import math
 
@@ -41,13 +40,10 @@ class Task(Model):
 
 
 # 新建任务
-def create_task(publish_id, slice_number, task_name):
+def create_task(publish_id, task_name):
     try:
-        if User.find_id(publish_id):
-            task = Task(publish_id=publish_id, slice_number=slice_number, task_name=task_name)
-            task.save()
-        else:
-            return Result(0, "User not exist", {}, 0)
+        task = Task(publish_id=publish_id, task_name=task_name)
+        task.save()
     except:
         return Result(0, "create failed", {}, 0)
     else:
@@ -130,11 +126,21 @@ def delete_task(task_id):
         return Result(1, "delete success", {}, 0)
 
 
+# 添加分片
+def add_slice(task_id):
+    try:
+        t = Task.get(Task.task_id == task_id)
+    except:
+        return False
+    else:
+        t.slice_number += 1
+        t.save()
+        return True
+
+
 if __name__ == '__main__':
     # 建表
     # Task.create_table()
     # 添加数据
-    # Task.create_task(publish_id, slice_number, task_name)
-    r = find_task()
-    for row in r.data:
-        print(row)
+    # Task.create_task(publish_id, task_name)
+    add_slice(1)