update

main/run_shipinhao_recommend.py

@@ -19,7 +19,7 @@ class Main:
         while True:
             if 20 >= datetime.datetime.now().hour >= 10:
                 Recommend.start_wechat('recommend')
-                Download.run_download_publish('recommend', env)
+                # Download.run_download_publish('recommend', env)
                 Common.del_logs('recommend')
                 time.sleep(5)
             else:

main/shipinhao_get_url.py

@@ -5,6 +5,7 @@ import json
 import os
 import sys
 import time
+import psutil as psutil
 # import atomacos
 from appium import webdriver
 from selenium.webdriver.common.by import By
@@ -16,6 +17,24 @@ from main.common import Common
 
 
 class GetUrl:
+    @classmethod
+    def kill_pid(cls):
+        os.system('chcp 65001')  # 将cmd的显示字符编码从默认的GBK改为UTF-8
+        list_process = list()
+        pid_list = psutil.pids()
+        for sub_pid in pid_list:
+            try:
+                process_info = psutil.Process(sub_pid)
+                # print(process_info.name())
+                if process_info.name() == 'WechatBrowser.exe' or process_info.name() == 'WeChatPlayer.exe':
+                    list_process.append(sub_pid)
+            except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
+                pass
+        # print('\n')
+        # print(list_process)
+        for pid in list_process:
+            os.system('taskkill /f /pid ' + str(pid))
+
     @classmethod
     def click_video(cls, log_type, video_title):
         try:
@@ -38,12 +57,11 @@ class GetUrl:
             # print(f'点击视频{video_title}')
             driver.find_elements(By.ID, '42.852020.3.487')[-1].click()
 
-            # 切换 windows 窗口
-            # windows = driver.window_handles
-            # driver.switch_to.window(windows[0])
+            Common.logger(log_type).info('退出视频号')
+            time.sleep(2)
+            cls.kill_pid()
 
-            Common.logger(log_type).info('休眠 3s 后，退出微信')
-            time.sleep(3)
+            Common.logger(log_type).info('退出微信')
             driver.quit()
 
         except Exception as e: