add weixinzhishu: search_key

weixinzhishu/weixinzhishu_main/search_key.py

@@ -15,6 +15,7 @@
 
 import json
 import os
+import shutil
 import sys
 import time
 
@@ -38,10 +39,12 @@ class Searchkey:
             driver.implicitly_wait(10)
             # Common.logger(log_type, crawler).info('点击微信指数')
             driver.find_elements(By.NAME, '消息')[-1].click()
-            time.sleep(1)
-            time.sleep(1)
+            time.sleep(2)
             cls.kill_pid(log_type, crawler)
+            time.sleep(2)
             driver.quit()
+            time.sleep(2)
+            cls.rmtree_WMPFRuntime()
         except Exception as e:
             Common.logger(log_type, crawler).error(f'start_wechat异常:{e}\n')
 
@@ -67,6 +70,13 @@ class Searchkey:
         except Exception as e:
             Common.logger(log_type, crawler).error(f'kill_pid异常:{e}\n')
 
+    @classmethod
+    def rmtree_WMPFRuntime(cls):
+        WMPFRuntime_path = r"C:\Users\guosh\AppData\Roaming\Tencent\WeChat\XPlugin\Plugins\WMPFRuntime"
+        shutil.rmtree(WMPFRuntime_path)
+        if not os.path.exists(WMPFRuntime_path):
+            os.mkdir(WMPFRuntime_path)
+
     @classmethod
     def get_search_key(cls, log_type, crawler):
         try:

weixinzhishu/weixinzhishu_main/weixinzhishu.py

@@ -23,7 +23,8 @@ class Weixinzhishu:
             return search_key, openid
 
     @classmethod
-    def weixinzhishu(cls, log_type, crawler, query):
+    def weixinzhishu(cls, log_type, crawler):
+        search_word_list = cls.search_word()
         wechat_key = cls.wechat_key(log_type, crawler)
         search_key = wechat_key[0]
         openid = wechat_key[-1]
@@ -33,26 +34,30 @@ class Weixinzhishu:
         print(f"openid:{openid}")
         print(f"start_ymd:{start_ymd}")
         print(f"start_ymd:{end_ymd}")
-        url = "https://search.weixin.qq.com/cgi-bin/wxaweb/wxindex"
-        payload = json.dumps({
-            "openid": openid,
-            "search_key": search_key,
-            "cgi_name": "GetDefaultIndex",
-            "start_ymd": "20230206",
-            "end_ymd": "20230210",
-            "query": query
-        })
-        headers = {
-            'Host': 'search.weixin.qq.com',
-            'content-type': 'application/json',
-            'User-Agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 13_3_1 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148 MicroMessenger/8.0.32(0x1800202a) NetType/WIFI Language/zh_CN',
-            'Referer': 'https://servicewechat.com/wxc026e7662ec26a3a/42/page-frame.html'
-        }
-        response = requests.request("POST", url, headers=headers, data=payload)
-        print(response.text)
-        time_index = response.json()['content']['resp_list'][0]['indexes'][0]['time_indexes']
-        print(time_index)
+        for word in search_word_list:
+            print(f"word:{word}")
+            url = "https://search.weixin.qq.com/cgi-bin/wxaweb/wxindex"
+            payload = json.dumps({
+                "openid": openid,
+                "search_key": search_key,
+                "cgi_name": "GetDefaultIndex",
+                "start_ymd": start_ymd,
+                "end_ymd": end_ymd,
+                "query": word
+            })
+            headers = {
+                'Host': 'search.weixin.qq.com',
+                'content-type': 'application/json',
+                'User-Agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 13_3_1 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148 MicroMessenger/8.0.32(0x1800202a) NetType/WIFI Language/zh_CN',
+                'Referer': 'https://servicewechat.com/wxc026e7662ec26a3a/42/page-frame.html'
+            }
+            response = requests.request("POST", url, headers=headers, data=payload)
+            if response.json()['code'] == -10000:
+                print(response.text)
+            else:
+                time_index = response.json()['content']['resp_list'][0]['indexes'][0]['time_indexes']
+                print(time_index)
 
 
 if __name__ == "__main__":
-    Weixinzhishu.weixinzhishu('weixin', 'weixinzhishu', '狂飙')
+    Weixinzhishu.weixinzhishu('weixin', 'weixinzhishu')