添加在进行随机抽样前检查数据长度功能

workers/select_spider_work.py

@@ -25,7 +25,11 @@ def bot_get_video_id():
         if not data:
             return
         data = [json.dumps(item) for item in data]
-        random_sample = random.sample(data, 26)
+        # 检查 data 列表的长度
+        if len(data) < 26:
+            random_sample = data
+        else:
+            random_sample = random.sample(data, 26)
         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]