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