wangkun hace 1 año
padre
commit
cb70410ab5
Se han modificado 1 ficheros con 6 adiciones y 9 borrados
  1. 6 9
      main/publish.py

+ 6 - 9
main/publish.py

@@ -4,7 +4,6 @@
 import json
 import os
 import random
-import shutil
 import sys
 import time
 import oss2
@@ -285,7 +284,7 @@ class Publish:
             return "CRAWLER"
 
     @classmethod
-    def local_file_path(cls, crawler):
+    def local_file_path(cls):
         # local_file_path = f'./{crawler}/videos'
         local_file_path = f'./videos'
         video_file = 'video'
@@ -313,11 +312,11 @@ class Publish:
         Common.logger(log_type).info("upload_and_publish starting...")
         today = time.strftime("%Y%m%d", time.localtime())
         # videos 目录下的所有视频文件夹
-        files = os.listdir(cls.local_file_path(crawler)["local_file_path"])
+        files = os.listdir(cls.local_file_path()["local_file_path"])
         for fv in files:
             try:
                 # 单个视频文件夹
-                fi_d = os.path.join(cls.local_file_path(crawler)["local_file_path"], fv)
+                fi_d = os.path.join(cls.local_file_path()["local_file_path"], fv)
                 # 确认为视频文件夹
                 if os.path.isdir(fi_d):
                     Common.logger(log_type).info('dir = {}'.format(fi_d))
@@ -336,7 +335,7 @@ class Publish:
                         fi_path = fi_d + '/' + fi
                         Common.logger(log_type).info('dir fi_path = {}'.format(fi_path))
                         # 读取 info.txt,赋值给 data
-                        if cls.local_file_path(crawler)["info_file"] in fi:
+                        if cls.local_file_path()["info_file"] in fi:
                             f = open(fi_path, "r", encoding="UTF-8")
                             # 读取数据 数据准确性写入的时候保证 读取暂不处理
                             for i in range(14):
@@ -363,7 +362,7 @@ class Publish:
                         fi_path = fi_d + '/' + fi
                         # Common.logger(log_type).info('dir fi_path = {}'.format(fi_path))
                         # 上传oss
-                        if cls.local_file_path(crawler)["video_file"] in fi:
+                        if cls.local_file_path()["video_file"] in fi:
                             global oss_video_file
                             if env == "dev":
                                 oss_video_file = cls.oss_file_path_video.format("dev", today, data['crawlerSrcId'])
@@ -375,7 +374,7 @@ class Publish:
                             cls.put_file(log_type, oss_endpoint, oss_video_file, fi_path)
                             data['videoPath'] = oss_video_file
                             Common.logger(log_type).info("videoPath = {}".format(oss_video_file))
-                        elif cls.local_file_path(crawler)["image_file"] in fi:
+                        elif cls.local_file_path()["image_file"] in fi:
                             global oss_image_file
                             if env == "dev":
                                 oss_image_file = cls.oss_file_path_image.format("env", today, data['crawlerSrcId'])
@@ -406,6 +405,4 @@ class Publish:
                 else:
                     Common.logger(log_type).error('file not a dir = {}'.format(fi_d))
             except Exception as e:
-                # 删除视频文件夹
-                shutil.rmtree(f"./{crawler}/videos/{fv}/")
                 Common.logger(log_type).exception('upload_and_publish error', e)