feat(profiler): add support for profiler

README.md

@@ -10,7 +10,7 @@
     - `conda install faiss-gpu cuda90 -c pytorch # For CUDA9.0`
     - `conda install flask`
     - `pip install flask-restful flask_sqlalchemy flask_script`
-    - `pip install pymysql environs`
+    - `pip install pymysql environs flask-profiler`
 
 ### Create Database
 

pyengine/engine/.env.example

@@ -2,6 +2,7 @@ DEBUG=True
 SQLALCHEMY_TRACK_MODIFICATIONS=False
 SECRET_KEY=test
 SQLALCHEMY_DATABASE_URI=mysql+pymysql://vecwise@127.0.0.1:3306/vecdata
+PROFILER_STORAGE_DB_URL=mysql+pymysql://vecwise@127.0.0.1:3306/vecdata
 
 ROW_LIMIT=10000000
 DATABASE_DIRECTORY=/tmp

pyengine/engine/__init__.py

@@ -2,9 +2,11 @@
 from engine import settings
 from flask import Flask
 from flask_sqlalchemy import SQLAlchemy
+import flask_profiler
 
 app = Flask(__name__)
 app.config.from_object(settings)
+app.config['flask_profiler'] = settings.FLASK_PROFILER_CONFIG
 
 #创建数据库对象
 print ("Create database instance")
@@ -14,3 +16,5 @@ from engine.model.group_table import GroupTable
 from engine.model.file_table import FileTable
 
 from engine.controller import views
+
+flask_profiler.init_app(app)

pyengine/engine/settings.py

@@ -10,3 +10,16 @@ SQLALCHEMY_DATABASE_URI = env.str('SQLALCHEMY_DATABASE_URI')
 
 ROW_LIMIT = env.int('ROW_LIMIT')
 DATABASE_DIRECTORY = env.str('DATABASE_DIRECTORY')
+
+FLASK_PROFILER_CONFIG = {
+    "enabled": DEBUG,
+    "storage": {
+        "engine": "sqlalchemy",
+        "db_url": env.str("PROFILER_STORAGE_DB_URL")
+    },
+    "basicAuth": {
+        "enabled": True,
+        "username": env.str("PROFILER_BASIC_AUTH_USERNAME", "admin"),
+        "password": env.str("PROFILER_BASIC_AUTH_PASSWORD", "admin"),
+    }
+}