Update popularity.py

qqhsx · May 25, 2024 · 93077c5 · 93077c5
1 parent d353fea
commit 93077c5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git 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)