浏览代码

修改项目启动代码

liyan 9 月之前
父节点
当前提交
d2ff61dd54
共有 2 个文件被更改,包括 7 次插入0 次删除
  1. 2 0
      watermark_generate/__init__.py
  2. 5 0
      watermark_generate/run.py

+ 2 - 0
watermark_generate/__init__.py

@@ -1,4 +1,5 @@
 import logging
+import os
 import sys
 from logging.handlers import RotatingFileHandler
 
@@ -7,6 +8,7 @@ VERSION = '1.0.0'
 log_format = '%(asctime)s - %(levelname)s - [%(filename)s] - [%(funcName)s] - line:[%(lineno)d] - %(message)s'
 log_size = 20 * 1024 * 1024  # 默认为 1MB
 log_level = logging.INFO
+os.makedirs('logs', exist_ok=True)
 log_name = 'logs/watermark_generate.log'
 
 for arg in sys.argv:  # 跳过脚本名称(sys.argv[0])

+ 5 - 0
watermark_generate/run.py

@@ -1,3 +1,8 @@
+import os
+import sys
+
+rootpath = str(os.path.abspath(os.path.join(os.path.dirname(__file__), '../')))
+sys.path.append(rootpath)
 from watermark_generate.app import create_app
 
 # 运行