Add .env configuration for Confluence settings
This commit is contained in:
18
main.py
18
main.py
@@ -14,10 +14,14 @@ from src.parser import HandoverLogParser
|
||||
from src.database import DailyLogsDatabase
|
||||
from src.report import DailyReportGenerator
|
||||
|
||||
# 配置
|
||||
CONF_BASE_URL = 'https://confluence.westwell-lab.com/rest/api'
|
||||
CONF_TOKEN = 'NDE1NTcwMDE1ODQ0OiinqS5HLm12v2orWEYyjJcI1bl5'
|
||||
CONF_CONTENT_ID = '155764524'
|
||||
# 加载环境变量
|
||||
from dotenv import load_dotenv
|
||||
load_dotenv()
|
||||
|
||||
# 配置(从环境变量读取)
|
||||
CONF_BASE_URL = os.getenv('CONFLUENCE_BASE_URL')
|
||||
CONF_TOKEN = os.getenv('CONFLUENCE_TOKEN')
|
||||
CONF_CONTENT_ID = os.getenv('CONFLUENCE_CONTENT_ID')
|
||||
|
||||
DEBUG_DIR = 'debug'
|
||||
|
||||
@@ -35,6 +39,10 @@ def get_timestamp():
|
||||
|
||||
def fetch_html():
|
||||
"""获取HTML内容"""
|
||||
if not CONF_BASE_URL or not CONF_TOKEN or not CONF_CONTENT_ID:
|
||||
print('错误:未配置 Confluence 信息,请检查 .env 文件')
|
||||
sys.exit(1)
|
||||
|
||||
print('正在从 Confluence 获取 HTML 内容...')
|
||||
client = ConfluenceClient(CONF_BASE_URL, CONF_TOKEN)
|
||||
html = client.get_html(CONF_CONTENT_ID)
|
||||
@@ -187,7 +195,7 @@ def main():
|
||||
fetch-save 获取、提取、解析并保存到数据库
|
||||
fetch-debug 获取、提取并保存带时间戳的debug文件
|
||||
report 生成日报(默认今天)
|
||||
report-today 生成今天日报
|
||||
report-today 生成今日日报
|
||||
parse-test 解析测试(使用已有的layout_output.txt)
|
||||
stats 显示今日统计
|
||||
|
||||
|
||||
Reference in New Issue
Block a user