zhangyong 1 month ago
parent
commit
ad26c4e86e
3 changed files with 26 additions and 14 deletions
  1. 1 1
      utils/gpt4o_mini_help.py
  2. 13 11
      workers/consumption_spider_work.py
  3. 12 2
      workers/select_spider_work.py

+ 1 - 1
utils/gpt4o_mini_help.py

@@ -12,7 +12,7 @@ class GPT4oMini:
             "prompt": (
             "prompt": (
                 "你是一个专业的标题分析师,请你通过视频标题来判断视频内容是否属于早安祝福内容。"
                 "你是一个专业的标题分析师,请你通过视频标题来判断视频内容是否属于早安祝福内容。"
                 "早安祝福视频标题标准如下:"
                 "早安祝福视频标题标准如下:"
-                "1.标题中含有“早”、“早上好”、“早安”等带有早安祝福的字眼"
+                "1.标题中含有“早”、“早上好”、“早安”等带有早安祝福的字眼,则符合"
                 "2.标题中带有早安祝福含义"
                 "2.标题中带有早安祝福含义"
                 f"请严格按照示例输出,标题:{title}。"
                 f"请严格按照示例输出,标题:{title}。"
             ),
             ),

+ 13 - 11
workers/consumption_spider_work.py

@@ -54,18 +54,18 @@ class ConsumptionSpiderRecommend(object):
             PQ.video_tag(task['video_id'], "lev-供给,rol-机器,#str-早安内容固定时段上推荐base_63")
             PQ.video_tag(task['video_id'], "lev-供给,rol-机器,#str-早安内容固定时段上推荐base_63")
             logger.info('[处理] 不属于早安祝福内容')
             logger.info('[处理] 不属于早安祝福内容')
             return
             return
-        if not title_status:
+        if title_status:
+            logger.info('[处理] 开始上推荐处理')
+            code = PQ.pq_recommended(task['video_id'])
+            if code:
+                logger.info('[处理] 上推荐处理成功')
+                PQ.video_tag(task['video_id'], "lev-供给,rol-机器,#str-早安内容固定时段上推荐实验_62")
+                logger.info('[处理] 视频打固定标签成功')
+            return
+        else:
             PQ.video_tag(task['video_id'], "lev-供给,rol-机器,#str-早安内容固定时段上推荐base_63")
             PQ.video_tag(task['video_id'], "lev-供给,rol-机器,#str-早安内容固定时段上推荐base_63")
             logger.info('[处理] 非当日日期不推荐')
             logger.info('[处理] 非当日日期不推荐')
             return
             return
-        logger.info('[处理] 开始上推荐处理')
-        code = PQ.pq_recommended(task['video_id'])
-        if code:
-            logger.info('[处理] 上推荐处理成功')
-            PQ.video_tag(task['video_id'], "lev-供给,rol-机器,#str-早安内容固定时段上推荐实验_62")
-            logger.info('[处理] 视频打固定标签成功')
-        return
-
 
 
 
 
 
 
@@ -90,5 +90,7 @@ async def run():
 
 
 if __name__ == '__main__':
 if __name__ == '__main__':
     # asyncio.run(ConsumptionSpiderRecommend.run())
     # asyncio.run(ConsumptionSpiderRecommend.run())
-    loop = asyncio.get_event_loop()
-    loop.run_until_complete(run())
+    # loop = asyncio.get_event_loop()
+    # loop.run_until_complete(run())
+    title = ConsumptionSpiderRecommend.judgment_time("🔴早上好❗2月13号的第一声问候,送给亲爱的你")
+    print(title)

+ 12 - 2
workers/select_spider_work.py

@@ -1,10 +1,13 @@
 import json
 import json
+import random
 import sys
 import sys
 import time
 import time
 
 
+import orjson
 import schedule
 import schedule
 from loguru import logger
 from loguru import logger
 
 
+from utils.piaoquan import PQ
 
 
 sys.path.append('/app')
 sys.path.append('/app')
 from utils.sql_help import sqlCollect
 from utils.sql_help import sqlCollect
@@ -20,8 +23,15 @@ def bot_get_video_id():
         data = sqlCollect.select_hour_data()
         data = sqlCollect.select_hour_data()
         if not data:
         if not data:
             return
             return
-        RedisHelper().get_client().rpush(redis_task, *(json.dumps(item) for item in data))
-        logger.info(f"[R] 早上好数据写入Redis 成功 共写入 {len(data)} 条")
+        data = [json.dumps(item) for item in data]
+        random_sample = random.sample(data, 13)
+        RedisHelper().get_client().rpush(redis_task, *random_sample)
+        logger.info(f"[R] 早上好数据写入Redis 成功 共写入 {len(random_sample)} 条")
+        remaining_data = [item for item in data if item not in random_sample]
+        for remaining in remaining_data:
+            task = orjson.loads(remaining)
+            PQ.video_tag(task['video_id'], "lev-供给,rol-机器,#str-早安内容固定时段上推荐base_63")
+            time.sleep(1)
     except Exception as e:
     except Exception as e:
         logger.error(f"[R] 早上好数据写入Redis写入失败,失败信息{e}")
         logger.error(f"[R] 早上好数据写入Redis写入失败,失败信息{e}")