更改消息通知

agc_assign_main.py

@@ -1,4 +1,6 @@
-from common import Material, Common
+import re
+
+from common import Material, Common, Feishu
 from video_agc.agc_video_method import AgcVidoe
 import concurrent.futures
 import schedule
@@ -11,6 +13,8 @@ def video_start(user_data):
     global returned_usernames_today
     user_data_mark = user_data["mark"]
     video_call = user_data["video_call"]
+    mark_name = user_data['mark_name']
+
     # 开始准备执行生成视频脚本
     if user_data_mark is not None and user_data_mark in returned_usernames_today:
         Common.logger("video").info(f"视频脚本参数中的用户名 {user_data_mark} 今天已经返回过，不再启动线程。今天已经返回的用户名：{returned_usernames_today}")
@@ -26,6 +30,12 @@ def video_start(user_data):
         if mark:
             Common.logger("video").info(f"返回用户名{mark}")
             returned_usernames_today.append(mark)
+            zd_count = user_data["zd_count"]  # 生成总条数
+            # 总条数
+            result = re.match(r'([^0-9]+)', mark).group()
+            all_count = AgcVidoe.get_link_gs_count(result)
+            if all_count >= int(zd_count):
+                Feishu.bot('recommend', 'AGC完成通知', '今日脚本跟随视频拼接任务完成啦～', mark.split("-")[0], mark_name)
 
 # gs_name_list = Material.feishu_gs_list()
 # video_start(gs_name_list[0])

video_agc/agc_video_method.py

@@ -566,12 +566,12 @@ class AgcVidoe():
         count = result[0][1] # 生成条数
         zd_count = ex_list["zd_count"] # 生成总条数
 
-        # 总条数
-        result = re.match(r'([^0-9]+)', mark).group()
-        all_count = cls.get_link_gs_count(result)
-        if all_count >= int(zd_count):
-            Feishu.bot('recommend', 'AGC完成通知', '今日脚本跟随视频拼接任务完成啦～', mark.split("-")[0], mark_name)
-            return mark
+        # # 总条数
+        # result = re.match(r'([^0-9]+)', mark).group()
+        # all_count = cls.get_link_gs_count(result)
+        # if all_count >= int(zd_count):
+        #     Feishu.bot('recommend', 'AGC完成通知', '今日脚本跟随视频拼接任务完成啦～', mark.split("-")[0], mark_name)
+        #     return mark
         # 获取音频类型+字幕+标题
         uid, srt, video_list = Material.get_all_data(feishu_id, link, mark)
         platform_list = ex_list["platform_list"] # 渠道