소스 검색

feat: 添加profile调试

SongZihuan 2 년 전
부모
커밋
aa0170a3fa
2개의 변경된 파일7개의 추가작업 그리고 0개의 파일을 삭제
  1. 6 0
      app/app.py
  2. 1 0
      configure/__init__.py

+ 6 - 0
app/app.py

@@ -22,6 +22,9 @@ from .oss import oss
 from .auth import auth
 from .about_me import about_me
 
+if conf["DEBUG_PROFILE"]:
+    from werkzeug.middleware.profiler import ProfilerMiddleware
+
 
 class HBlogFlask(Flask):
     def __init__(self, import_name: str, *args, **kwargs):
@@ -29,6 +32,9 @@ class HBlogFlask(Flask):
         self.about_me = ""
         self.update_configure()
 
+        if conf["DEBUG_PROFILE"]:
+            self.wsgi_app = ProfilerMiddleware(self.wsgi_app)
+
         self.register_blueprint(index, url_prefix="/")
         self.register_blueprint(archive, url_prefix="/archive")
         self.register_blueprint(docx, url_prefix="/docx")

+ 1 - 0
configure/__init__.py

@@ -3,6 +3,7 @@ import logging
 import os
 
 conf = {
+    "DEBUG_PROFILE": False,
     "SECRET_KEY": "HBlog-R-Salt",
     "BLOG_NAME": "HBlog",
     "BLOG_DESCRIBE": "Huan Blog.",