diff --git a/popularity.py b/popularity.py index 22b82362..fc14b01a 100644 --- a/popularity.py +++ b/popularity.py @@ -1,8 +1,9 @@ +import os import pywencai -# 定义变量 -query = '退市股票' -filename = f'{query}.csv' +# 从环境变量读取查询词 +query = os.getenv('CSV_FILE_NAME ', '退市股票') # 如果环境变量不存在,则使用默认值 '退市股票' +filename = f'{CSV_FILE_NAME }.csv' # 使用变量进行查询和保存结果 res = pywencai.get(question=query, loop=True)