Ver código fonte

视频不满足下载规则后,删除其所在文件夹

wangkun 2 anos atrás
pai
commit
6da5cf377b
4 arquivos alterados com 11 adições e 5 exclusões
  1. 0 3
      logs/__init__.py
  2. 6 1
      main/common.py
  3. 4 0
      main/recommend.py
  4. 1 1
      main/run_recommend.py

+ 0 - 3
logs/__init__.py

@@ -1,3 +0,0 @@
-# -*- coding: utf-8 -*-
-# @Author: wangkun
-# @Time: 2022/4/25

+ 6 - 1
main/common.py

@@ -64,6 +64,9 @@ class Common:
         :return: 保留最近 6 个日志
         """
         logs_dir = "./logs/"
+        if not os.path.exists(logs_dir):
+            os.mkdir(logs_dir)
+
         all_files = sorted(os.listdir(logs_dir))
         all_logs = []
         for log in all_files:
@@ -87,9 +90,11 @@ class Common:
         视频封面,或视频播放地址:d_url
         下载保存路径:"./files/{d_title}/"
         """
+        videos_dir = "./videos/"
+        if not os.path.exists(videos_dir):
+            os.mkdir(videos_dir)
         # 首先创建一个保存该视频相关信息的文件夹
         video_dir = "./videos/" + d_name + "/"
-        # video_dir = "./videos/"
         if not os.path.exists(video_dir):
             os.mkdir(video_dir)
 

+ 4 - 0
main/recommend.py

@@ -296,12 +296,16 @@ class Recommend:
 
                     # 视频时长<60s,直接删除
                     if int(download_video_duration) < 60:
+                        # 删除视频文件夹
                         shutil.rmtree("./videos/" + download_video_title + "/")
+                        # 删除云文档recommend_feeds中的记录
                         Feishu.dimension_range(log_type, "bszf", "CcHgO7", "ROWS", i + 1, i + 1)
                         Common.logger(log_type).info("时长:{}<60秒,删除成功\n", int(download_video_duration))
                         return
                     # 竖版视频不下载,写入竖版视频表
                     elif int(video_info[0]) < int(video_info[1]):
+                        # 删除视频文件夹
+                        shutil.rmtree("./videos/" + download_video_title + "/")
                         # 删除在 recommend_feeds 的记录
                         Feishu.dimension_range(log_type, "bszf", "CcHgO7", "ROWS", i + 1, i + 1)
                         Common.logger(log_type).info("宽:{}<高:{},删除成功", int(video_info[0]), int(video_info[1]))

+ 1 - 1
main/run_recommend.py

@@ -19,7 +19,7 @@ class Main:
         while True:
             while True:
                 main_time = datetime.datetime.now()
-                if 12 >= main_time.hour >= 8:
+                if 15 >= main_time.hour >= 8:
                     # 抓取视频
                     Common.logger("recommend").info("开始抓取本山祝福视频\n")
                     for i in range(10):