requirements and db config

backend_refactor/Dockerfile

@@ -7,7 +7,7 @@ WORKDIR /pose
 RUN sed -i 's/archive.ubuntu.com/mirrors.nju.edu.cn/g' /etc/apt/sources.list && \ 
     sed -i 's/security.ubuntu.com/mirrors.nju.edu.cn/g' /etc/apt/sources.list && \
     apt update && \
-    apt install -y libgl1-mesa-glx libglib2.0-dev && \
+    apt install -y libgl1-mesa-glx libglib2.0-dev python3-mysqldb && \
     rm -rf /var/lib/apt/lists/*
 
 COPY requirements.txt .

backend_refactor/app.py

@@ -2,7 +2,7 @@ import logging
 from nturl2path import url2pathname
 from flask import Flask
 import route.upload, route.analyze, route.result
-from config import VERSION_PREFIX, MAX_CONTENT_LENGTH
+from config import VERSION_PREFIX, MAX_CONTENT_LENGTH, MYSQL_URI
 from route.util import make_response
 
 from db import db
@@ -11,7 +11,7 @@ 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_DATABASE_URI'] = MYSQL_URI
 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
 db.init_app(app)
 with app.app_context():

backend_refactor/config/__init__.py

@@ -1,5 +1,9 @@
+import os
+
 VERSION_PREFIX = "/v1/"
 UPLOAD_DIR = "/data/"
 MAX_CONTENT_LENGTH = 100 * 1024 * 1024
 
+MYSQL_URI = f'mysql://{os.environ["MYSQL_USER"]}:{os.environ["MYSQL_PASSWORD"]}@{os.environ["MYSQL_ADDRESS"]}/{os.environ["MYSQL_DATABASE"]}'
+
 CACHE_DIR = "/tmp/"

backend_refactor/requirements.txt

@@ -5,3 +5,4 @@ numpy==1.21.6
 opencv-python==4.6.0.66
 matplotlib==3.5.3
 scipy==1.7.3
+flask_sqlalchemy==2.5.1