wangkun 1 rok temu
rodzic
commit
9151766f5f

+ 2 - 1
common/common.py

@@ -314,10 +314,11 @@ class Common:
                         for chunk in response.iter_content(chunk_size=10240):
                             f.write(chunk)
                     cls.logger(log_type, crawler).info("==========视频下载完成==========")
-                    break
+                    return True
                 except Exception as e:
                     cls.logger(log_type, crawler).error(f"视频下载失败:{e}\n")
                     time.sleep(1)
+            return False
 
         # 下载音频
         elif text == "audio":

+ 4 - 1
jixiangxingfu/jixiangxingfu_recommend/jixiangxingfu_recommend.py

@@ -314,7 +314,10 @@ class JixiangxingfuRecommend:
     @classmethod
     def download_publish(cls, log_type, crawler, video_dict, env, driver: WebDriver):
         # 下载视频
-        Common.download_method(log_type=log_type, crawler=crawler, text='video', title=video_dict['video_title'], url=video_dict['video_url'])
+        download_res = Common.download_method(log_type=log_type, crawler=crawler, text='video', title=video_dict['video_title'], url=video_dict['video_url'])
+        if download_res is False:
+            driver.press_keycode(AndroidKey.BACK)
+            return
         ffmpeg_dict = Common.ffmpeg(log_type, crawler, f"./{crawler}/videos/{video_dict['video_title']}/video.mp4")
         if ffmpeg_dict is None:
             md_title = md5(video_dict['video_title'].encode('utf8')).hexdigest()