Update popularity.py

qqhsx · May 25, 2024 · 4609f70 · 4609f70
1 parent 5466112
commit 4609f70
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/popularity.py b/popularity.py
@@ -1,9 +1,8 @@
-import os
 import pywencai
 
-# 从环境变量读取查询词
-query = os.getenv('CSV_FILE_NAME ', '退市股票')  # 如果环境变量不存在，则使用默认值 '退市股票'
-filename = f'{CSV_FILE_NAME }'
+# 定义变量
+query = '退市股票'
+filename = f'{query}.csv'
 
 # 使用变量进行查询和保存结果
 res = pywencai.get(question=query, loop=True)