连接数据库方式更改

DAPlatform/app.py

@@ -8,12 +8,10 @@ from views.task import task
 from views.slice import slice
 from views.file import data
 from views.imageAnnotation import image
-from playhouse.flask_utils import FlaskDB
 from dao.model import mysql
 from config.config import DevelopmentConfig
 
 app = Flask(__name__)
-FlaskDB(app, mysql)
 CORS(app)
 app.config.from_object(DevelopmentConfig)
 

DAPlatform/views/file.py

@@ -2,10 +2,22 @@ import json
 from flask import Blueprint
 from flask import request
 from dao import Data, Relation
+from dao.model import mysql
 
 data = Blueprint("data", __name__, url_prefix="/api/file")
 
 
+@data.before_request
+def before_request():
+    mysql.connect()
+
+
+@data.after_request
+def after_request(response):
+    mysql.close()
+    return response
+
+
 @data.route("/getFiles", methods=["GET"])
 def get_pieces():
     sliceId = request.args['pieceId']

DAPlatform/views/slice.py

@@ -2,10 +2,22 @@ import json
 from flask import Blueprint
 from flask import request
 from dao import Data, Slice, Relation, Task
+from dao.model import mysql
 
 slice = Blueprint("slice", __name__, url_prefix="/api/slice")
 
 
+@slice.before_request
+def before_request():
+    mysql.connect()
+
+
+@slice.after_request
+def after_request(response):
+    mysql.close()
+    return response
+
+
 # 获取分片列表的接口，接收参数taskId，返回相应task下的分片列表
 @slice.route("/getPieces", methods=["GET"])
 def get_pieces():

DAPlatform/views/task.py

@@ -2,10 +2,21 @@ import json
 from flask import Blueprint
 from flask import request
 from dao import Task, Slice, Data, Relation, User
-
+from dao.model import mysql
 task = Blueprint("task", __name__, url_prefix="/api/task")
 
 
+@task.before_request
+def before_request():
+    mysql.connect()
+
+
+@task.after_request
+def after_request(response):
+    mysql.close()
+    return response
+
+
 # 各个界面获取任务列表的接口
 @task.route("/getTasks", methods=["GET"])
 def get_tasks():

DAPlatform/views/user.py

@@ -2,10 +2,22 @@ import json
 from flask import Blueprint
 from flask import request
 from dao import User
+from dao.model import mysql
 
 user = Blueprint("user", __name__, url_prefix="/api/user")
 
 
+@user.before_request
+def before_request():
+    mysql.connect()
+
+
+@user.after_request
+def after_request(response):
+    mysql.close()
+    return response
+
+
 @user.route("/signup", methods=["Post"])
 def signup():
     # print("signup")