Bob Huang 4 лет назад
Родитель
Сommit
225eba55fc
3 измененных файлов с 22 добавлено и 0 удалено
  1. 8 0
      backend_refactor/app.py
  2. 3 0
      backend_refactor/db/__init__.py
  3. 11 0
      backend_refactor/db/task.py

+ 8 - 0
backend_refactor/app.py

@@ -7,11 +7,19 @@ import route.upload
 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
 
 pool = ThreadPoolExecutor(max_workers=2)
 
+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;