jack hace 11 meses
padre
commit
2f1a3f3e19
Se han modificado 4 ficheros con 16 adiciones y 4 borrados
  1. 2 1
      .gitignore
  2. 7 1
      ai_news/ai_news.py
  3. 1 1
      ai_news/config.json
  4. 6 1
      message/message_coin_detail.py

+ 2 - 1
.gitignore

@@ -60,4 +60,5 @@ docs/_build/
 # PyBuilder
 target/
 
-other/split_clash_config/split_config
+other/split_clash_config/split_config
+ai_news/save_txt

+ 7 - 1
ai_news/ai_news.py

@@ -241,7 +241,12 @@ class AINEWS:
             result_text = asyncio.run(self.get_htmls(target_url_list))
 
         # 保存文本
-        self.save_to_txt(result_text)
+        if result_text:
+            print(f'共获取 {len(result_text)} 个网址的数据')
+            self.save_to_txt(result_text)
+        else:
+            print('无数据, 程序退出')
+            exit(0)
 
         # # 如果只需要保存爬取数据, 不使用 AI, 注释下面
         # # 创建消息bot实例
@@ -261,5 +266,6 @@ if __name__ == "__main__":
 
     for key in key_list:
         target_url_list, prompt_words, role, use_browser, ai_host = ainews.load_config(key)
+        print(f'关键词 {key} 共有 {len(target_url_list)} 个网址')
         ainews.main(target_url_list, prompt_words, role, use_browser, ai_host)
     print('done!')

+ 1 - 1
ai_news/config.json

@@ -7,7 +7,7 @@
     "role": ""
   },
   "web3": {
-    "use_browser": 0,
+    "use_browser": 1,
     "ai_host": "http://home.erhe.link:36001",
     "target_url_list": [
       "https://wublock123.com",

+ 6 - 1
message/message_coin_detail.py

@@ -37,7 +37,12 @@ def fetch_coin_data(target):
         else:
             text = ''
             data = response.json()
-            target_data = eval(data['result'][0]['data'])
+            try:
+                target_data = eval(data['result'][0]['data'])
+            except Exception as e:
+                client.close()
+                raise Exception(f"Failed to parse data: {data}, error: {str(e)}")
+
             target_data = target_data[0]
 
             # print(target_data)