Parcourir la source

develop publish_single_video_pool_videos.py

luojunhui il y a 3 mois
Parent
commit
222a78ba5c
1 fichiers modifiés avec 3 ajouts et 8 suppressions
  1. 3 8
      coldStartTasks/publish/publish_single_video_pool_videos.py

+ 3 - 8
coldStartTasks/publish/publish_single_video_pool_videos.py

@@ -42,6 +42,7 @@ class PublishSingleVideoPoolVideos:
             from single_video_transform_queue
             where status = 0 and platform = '{platform}'
             order by score desc
+            limit 2000
         """
         fetch_response = self.db_client.fetch(query=fetch_query, cursor_type=DictCursor)
         return fetch_response
@@ -72,7 +73,7 @@ class PublishSingleVideoPoolVideos:
             vid_list = [task['pq_vid'] for task in task_list]
             if vid_list:
                 # create video crawler plan
-                plan_name = f"{platform_name_map[platform]}-历史数据-视频数量: {len(vid_list)}"
+                plan_name = f"{platform_name_map[platform]}-{datetime.datetime.today().strftime('%Y-%m-%d')}-视频数量: {len(vid_list)}"
                 crawler_plan_response = aiditApi.auto_create_single_video_crawler_task(
                     plan_name=plan_name,
                     plan_tag="单视频供给冷启动",
@@ -103,10 +104,4 @@ class PublishSingleVideoPoolVideos:
                 aiditApi.bind_crawler_task_to_generate_task(
                     crawler_task_list=crawler_task_list,
                     generate_task_id=generate_plan_id,
-                )
-
-
-
-if __name__ == '__main__':
-    task = PublishSingleVideoPoolVideos()
-    task.deal()
+                )