Przeglądaj źródła

Merge branch 'dev' of http://gogs.seec.seecoder.cn/ZhaoFengShan/PoseCorrection into dev

LeoDu 4 lat temu
rodzic
commit
203a5fec8a

+ 8 - 0
backend_refactor/app.py

@@ -5,10 +5,18 @@ import route.upload, route.analyze, route.result
 from config import VERSION_PREFIX, MAX_CONTENT_LENGTH
 from route.util import make_response
 
+from db import db
+
 app = Flask(__name__)
 app.config['MAX_CONTENT_LENGTH'] = MAX_CONTENT_LENGTH
 
 
+app.config['SQLALCHEMY_DATABASE_URI'] = f'mysql://{MYSQL_USERNAME}:{MYSQL_PASSWORD}@{MYSQL_ADDRESS}/{MYSQL_DATABASE}'
+app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+db.init_app(app)
+with app.app_context():
+    # db.drop_all()
+    db.create_all()
 
 @app.route("/")
 def hello_world():

+ 3 - 0
backend_refactor/db/__init__.py

@@ -0,0 +1,3 @@
+from flask_sqlalchemy import SQLAlchemy
+
+db = SQLAlchemy()

+ 11 - 0
backend_refactor/db/task.py

@@ -0,0 +1,11 @@
+from db import db
+
+class Task:
+	uuid = db.Column(db.String(64), primary_key=True)
+	status = db.Column(db.Enum('QUEUEING','RUNNING','FINISHED','ERROR'))
+	result = db.Column(db.String(8192))
+
+	def __init__(self, uuid, status='QUEUEING', result='{}'):
+		self.uuid = uuid;
+		self.status = status;
+		self.result = result;