Kaynağa Gözat

feat: 允许设置域名

SongZihuan 3 yıl önce
ebeveyn
işleme
f3d9be41fe
2 değiştirilmiş dosya ile 5 ekleme ve 1 silme
  1. 1 0
      configure/__init__.py
  2. 4 1
      main.py

+ 1 - 0
configure/__init__.py

@@ -27,6 +27,7 @@ def configure(conf_file: str, encoding="utf-8"):
     conf["email_sender"] = str(_email["sender"])
 
     conf["secret-key"] = str(_conf.get("secret-key", "HBlog-R-Salt"))
+    conf["server-name"] = _conf.get("server-name")
 
     introduce = _conf["info"]["introduce"]
     introduce_list = []

+ 4 - 1
main.py

@@ -1,4 +1,4 @@
-from configure import configure
+from configure import configure, conf
 import os
 
 
@@ -19,6 +19,9 @@ from waitress import serve
 web = WebApp(__name__)
 app = web.get_app()
 
+if conf["server-name"] is not None:
+    app.config['SERVER_NAME'] = conf["server-name"]
+
 
 if __name__ == '__main__':
     print("已启动服务: 127.0.0.1:8080")