zhangyong 7 månader sedan
förälder
incheckning
d94f3f552c
2 ändrade filer med 10 tillägg och 8 borttagningar
  1. 8 3
      common/redis.py
  2. 2 5
      video_rewriting/video_processor.py

+ 8 - 3
common/redis.py

@@ -1,5 +1,7 @@
 import redis
 
+from common import Material
+
 
 class SyncRedisHelper:
     _pool: redis.ConnectionPool = None
@@ -33,7 +35,7 @@ class SyncRedisHelper:
             self._pool.disconnect(inuse_connections=True)
 
 
-def get_data(name, data):
+def get_data(name, feishu_id, feishu_sheet):
     task = f"task:{name}"
     lock = f"lock:{name}"
     helper = SyncRedisHelper()
@@ -42,10 +44,13 @@ def get_data(name, data):
         acquire_lock = client.set(lock, 1, ex=60, nx=True)
         if not acquire_lock:
             return None
-
+        if name == 'pl-gjc':
+            data = Material.get_keyword_data(feishu_id, feishu_sheet)
+        else:
+            data = Material.get_task_data(feishu_id, feishu_sheet)
         client.rpush(task, *data)
     ret = client.lpop(task)
-    if name == 'pl-gjc':
+    if name == 'pl-gjc' or name == 'dd-sp':
         client.rpush(task, ret)
     return ret
 

+ 2 - 5
video_rewriting/video_processor.py

@@ -556,12 +556,9 @@ class VideoProcessor:
         feishu_id = data["feishu_id"]
         feishu_sheet = data["feishu_sheet"]
         cookie_sheet = data["cookie_sheet"]
-        if mark == 'pl-gjc':
-            task_data = Material.get_keyword_data(feishu_id, feishu_sheet)
-        else:
-            task_data = Material.get_task_data(feishu_id, feishu_sheet)
+
         try:
-            data = get_data(mark, task_data)
+            data = get_data(mark, feishu_id, feishu_sheet)
             if not data:
                 Common.logger("redis").error(f"{mark}任务开始新的一轮\n")
                 return