76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
定时排期生成器 v2
|
||
规则：
|
||
1. 第一条视频 **立即发布**（first_delay=0）
|
||
2. 第二条起，每条间隔 30~120 分钟（随机）
|
||
3. 若总时长超过 max_hours，按比例压缩
|
||
4. 支持各平台原生定时 API 所需的 datetime 对象
|
||
"""
|
||
import random
|
||
from datetime import datetime, timedelta
|
||
|
||
|
||
def generate_schedule(
|
||
n: int,
|
||
min_gap: int = 30,
|
||
max_gap: int = 120,
|
||
max_hours: float = 24.0,
|
||
first_delay: int = 0,
|
||
start_time: datetime = None,
|
||
) -> list[datetime]:
|
||
"""
|
||
返回 n 个 datetime：
|
||
- times[0] = now + first_delay（默认 0 = 立即）
|
||
- times[1..] = 前一条 + random(min_gap, max_gap) 分钟
|
||
"""
|
||
if n <= 0:
|
||
return []
|
||
|
||
base = start_time or datetime.now()
|
||
|
||
if n == 1:
|
||
return [base + timedelta(minutes=first_delay)]
|
||
|
||
gaps = [random.randint(min_gap, max_gap) for _ in range(n - 1)]
|
||
total_min = first_delay + sum(gaps)
|
||
max_min = max_hours * 60
|
||
|
||
if total_min > max_min:
|
||
ratio = max_min / total_min
|
||
first_delay = max(0, int(first_delay * ratio))
|
||
gaps = [max(5, int(g * ratio)) for g in gaps]
|
||
|
||
times = []
|
||
cur = base + timedelta(minutes=first_delay)
|
||
times.append(cur)
|
||
for g in gaps:
|
||
cur = cur + timedelta(minutes=g)
|
||
times.append(cur)
|
||
|
||
return times
|
||
|
||
|
||
def format_schedule(videos: list[str], times: list[datetime]) -> str:
|
||
"""格式化排期表"""
|
||
lines = [" 序号 | 发布时间 | 间隔 | 视频"]
|
||
lines.append(" " + "-" * 70)
|
||
for i, (v, t) in enumerate(zip(videos, times)):
|
||
gap = "立即" if i == 0 else ""
|
||
if i > 0:
|
||
delta = (t - times[i - 1]).total_seconds() / 60
|
||
gap = f"{delta:.0f}min"
|
||
name = v[:40] if len(v) > 40 else v
|
||
lines.append(f" {i+1:>4} | {t.strftime('%Y-%m-%d %H:%M')} | {gap:>7} | {name}")
|
||
|
||
total = (times[-1] - times[0]).total_seconds() / 3600 if len(times) > 1 else 0
|
||
lines.append(" " + "-" * 70)
|
||
lines.append(f" 总跨度: {total:.1f}h | 首条: {times[0].strftime('%H:%M')}(立即) | 末条: {times[-1].strftime('%H:%M')}")
|
||
return "\n".join(lines)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
schedule = generate_schedule(12)
|
||
names = [f"视频_{i+1}.mp4" for i in range(12)]
|
||
print(format_schedule(names, schedule))
|