's'
This commit is contained in:
197
66666
Normal file
197
66666
Normal file
@@ -0,0 +1,197 @@
|
||||
import platform
|
||||
import time
|
||||
import datetime
|
||||
import subprocess
|
||||
|
||||
# ====================== 抖音 → YouTube 转推核心代码(终极稳定版)======================
|
||||
try:
|
||||
# 时间戳和标题设置
|
||||
timestamp_str = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())
|
||||
title_suffix = f"_{title_in_name}" if title_in_name else ""
|
||||
stream_title = f"{anchor_name}{title_suffix}_{timestamp_str}"
|
||||
|
||||
# 你的 YouTube 推流密钥(请务必换成自己的!)
|
||||
YT_STREAM_KEY = "x04z-564w-aks7-embw-30y4" # food频道
|
||||
youtube_rtmp = f"rtmp://a.rtmp.youtube.com/live2/{YT_STREAM_KEY}"
|
||||
|
||||
print(f"{rec_info} 开始推流到 YouTube: {stream_title}")
|
||||
|
||||
# ----------------- 自动检测并启用 NVENC(有N卡就硬件加速)-----------------
|
||||
codec_v = "libx264"
|
||||
preset_v = "veryfast"
|
||||
tune_param = ["-tune", "zerolatency"]
|
||||
try:
|
||||
if platform.system().lower() in ["windows", "linux"]:
|
||||
check = subprocess.run(["nvidia-smi"], capture_output=True, timeout=8)
|
||||
if check.returncode == 0:
|
||||
enc = subprocess.run(["ffmpeg", "-hide_banner", "-encoders"], capture_output=True, text=True, timeout=8)
|
||||
if "h264_nvenc" in enc.stdout:
|
||||
codec_v = "h264_nvenc"
|
||||
preset_v = "p5"
|
||||
tune_param = ["-tune", "ll", "-rc", "vbr", "-cq", "23"]
|
||||
print("检测到 NVIDIA GPU → 使用 NVENC 硬件加速")
|
||||
except Exception as e:
|
||||
print(f"NVENC检测异常: {e},使用CPU编码")
|
||||
|
||||
# ----------------- FFmpeg 推流命令(已极致优化)-----------------
|
||||
ffmpeg_command = [
|
||||
"ffmpeg",
|
||||
"-rw_timeout", "15000000",
|
||||
"-reconnect", "1",
|
||||
"-reconnect_at_eof", "1",
|
||||
"-reconnect_streamed", "1",
|
||||
"-reconnect_delay_max", "8",
|
||||
"-fflags", "+genpts+discardcorrupt+igndts",
|
||||
"-err_detect", "ignore_err",
|
||||
"-thread_queue_size", "2048",
|
||||
"-analyzeduration", "3000000",
|
||||
"-probesize", "3000000",
|
||||
"-i", real_url,
|
||||
"-vf", "scale=720:1280:force_original_aspect_ratio=decrease,pad=720:1280:(ow-iw)/2:(oh-ih)/2:black",
|
||||
"-c:v", codec_v, "-preset", preset_v, *tune_param,
|
||||
"-b:v", "2500k", "-maxrate", "2500k", "-bufsize", "5000k",
|
||||
"-g", "50", "-r", "25", "-pix_fmt", "yuv420p", "-bf", "0",
|
||||
"-c:a", "aac", "-b:a", "128k", "-ar", "44100", "-ac", "2",
|
||||
"-af", "aresample=async=1:first_pts=0",
|
||||
"-flags", "+low_delay+global_header",
|
||||
"-fflags", "+genpts+nobuffer",
|
||||
"-threads", "8",
|
||||
"-f", "flv", youtube_rtmp
|
||||
]
|
||||
|
||||
proc = None
|
||||
start_time = time.time() # 本次推流开始时间
|
||||
|
||||
while True:
|
||||
try:
|
||||
# 清理上一次残留进程
|
||||
if proc and proc.poll() is None:
|
||||
proc.terminate()
|
||||
try: proc.wait(timeout=6)
|
||||
except: proc.kill()
|
||||
|
||||
print(f"[{datetime.datetime.now().strftime('%H:%M:%S')}] 启动YouTube推流 → {stream_title}")
|
||||
proc = subprocess.Popen(
|
||||
ffmpeg_command,
|
||||
stdin=subprocess.DEVNULL,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
text=True,
|
||||
bufsize=1,
|
||||
universal_newlines=True
|
||||
)
|
||||
|
||||
# 时间戳属性
|
||||
proc.last_out_ts = time.time()
|
||||
last_frame_time = time.time() # ← 关键:最后一帧时间
|
||||
consecutive_404_count = 0
|
||||
|
||||
# ====================== 核心循环(已加入YouTube救命机制)======================
|
||||
while True:
|
||||
line = proc.stderr.readline()
|
||||
if line == '' and proc.poll() is not None:
|
||||
break
|
||||
if not line:
|
||||
time.sleep(0.01)
|
||||
continue
|
||||
|
||||
line = line.strip()
|
||||
if line:
|
||||
proc.last_out_ts = time.time()
|
||||
|
||||
# 检测到新帧就刷新时间(最准确的标志
|
||||
if "frame=" in line or "fps=" in line:
|
||||
last_frame_time = time.time()
|
||||
|
||||
print(f"FFmpeg: {line}")
|
||||
|
||||
line_lower = line.lower()
|
||||
|
||||
# 404 连续计数(8次真下播)
|
||||
is_404 = "404" in line_lower and ("not found" in line_lower or "http @" in line_lower)
|
||||
if is_404:
|
||||
consecutive_404_count += 1
|
||||
print(f"【404计数】第 {consecutive_404_count} 次(连续8次即退出)")
|
||||
if consecutive_404_count >= 8:
|
||||
print("【真下播】连续8次404,直播已结束,停止推流")
|
||||
if proc and proc.poll() is None:
|
||||
proc.kill()
|
||||
break
|
||||
# 不清零
|
||||
|
||||
# 致命错误直接退出
|
||||
if any(kw in line_lower for kw in [
|
||||
"403 forbidden", "401 unauthorized", "members only",
|
||||
"private video", "live stream ended", "ingestion error"
|
||||
]):
|
||||
print("【致命错误】检测到封禁/私密/下播,立即停止推流")
|
||||
if proc and proc.poll() is None:
|
||||
proc.kill()
|
||||
break
|
||||
|
||||
# YouTube救命机制:12秒无新帧就必须处决!(YouTube最多忍15~20秒)
|
||||
if time.time() - start_time > 25: # 25秒后进入战斗模式
|
||||
if time.time() - last_frame_time > 12: # 12秒没新帧 → 立刻重启
|
||||
print(f"【YouTube救命】已 {time.time()-last_frame_time:.1f}秒无新帧,立即重启FFmpeg避免直播被杀!")
|
||||
if proc and proc.poll() is None:
|
||||
proc.kill()
|
||||
break
|
||||
|
||||
# 双保险:20秒完全没日志输出也处决
|
||||
if time.time() - start_time > 20:
|
||||
if time.time() - proc.last_out_ts > 20:
|
||||
print("【真死流】20秒无任何输出,强制重启进程")
|
||||
if proc and proc.poll() is None:
|
||||
proc.kill()
|
||||
break
|
||||
|
||||
# 进程退出判断
|
||||
returncode = proc.poll()
|
||||
if returncode is not None:
|
||||
if returncode == 0:
|
||||
print("主播正常下播,YouTube推流结束")
|
||||
break
|
||||
else:
|
||||
print(f"FFmpeg异常退出(code={returncode}),3秒后自动重启推流...")
|
||||
time.sleep(3)
|
||||
continue
|
||||
|
||||
except Exception as e:
|
||||
print(f"推流过程中出现异常: {e}")
|
||||
time.sleep(5)
|
||||
continue
|
||||
|
||||
# ====================== 彻底清理 ======================
|
||||
if proc and proc.poll() is None:
|
||||
print("发现残留FFmpeg进程,执行kill")
|
||||
proc.kill()
|
||||
|
||||
try: recording.discard(record_name)
|
||||
except: pass
|
||||
try: recording_time_list.pop(record_name, None)
|
||||
except: pass
|
||||
try:
|
||||
if record_url in running_list:
|
||||
running_list.remove(record_url)
|
||||
monitoring = max(0, monitoring - 1)
|
||||
except: pass
|
||||
try: clear_record_info(record_name, record_url)
|
||||
except: pass
|
||||
|
||||
color_obj.print_colored(f"[{record_name}] YouTube推流已彻底停止并清理完毕\n", color_obj.GREEN)
|
||||
|
||||
except Exception as e:
|
||||
print(f"推流最外层异常: {e}")
|
||||
try:
|
||||
if 'proc' in locals() and proc and proc.poll() is None:
|
||||
proc.kill()
|
||||
except: pass
|
||||
try:
|
||||
recording.discard(record_name)
|
||||
recording_time_list.pop(record_name, None)
|
||||
if record_url in running_list:
|
||||
running_list.remove(record_url)
|
||||
monitoring = max(0, monitoring - 1)
|
||||
clear_record_info(record_name, record_url)
|
||||
except: pass
|
||||
color_obj.print_colored(f"[{record_name}] 异常退出但已清理完毕\n", color_obj.RED)
|
||||
Reference in New Issue
Block a user