mpv-grok
This commit is contained in:
Your Name
310
main.py
310
main.py
@@ -1622,15 +1622,21 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
|
||||
error_window.append(1)
|
||||
|
||||
else:
|
||||
# ====================== 抖音 → HDMI 转播（FIFO + mpv 常驻 + 卡死自动重启 + 超强稳定重连） ======================
|
||||
# ====================== 抖音 → HDMI 终极无人值守版（365×24h 永不卡死） ======================
|
||||
import queue
|
||||
import unicodedata
|
||||
import traceback
|
||||
# import os
|
||||
# import datetime
|
||||
# import time
|
||||
# import subprocess
|
||||
# import threading
|
||||
# import re
|
||||
# import signal
|
||||
|
||||
class StreamEnded(Exception):
|
||||
"""主播真实下播时抛出，用于彻底退出循环"""
|
||||
pass
|
||||
|
||||
def sanitize_title(text: str, max_len: int = 120) -> str:
|
||||
if not text:
|
||||
return ""
|
||||
@@ -1638,6 +1644,7 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
|
||||
text = ''.join(ch for ch in text if unicodedata.category(ch)[0] != 'C')
|
||||
text = ''.join(ch for ch in text if ord(ch) <= 0xFFFF)
|
||||
return text.strip()[:max_len]
|
||||
|
||||
def _stderr_reader_thread(proc, q):
|
||||
try:
|
||||
while True:
|
||||
@@ -1645,171 +1652,140 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
|
||||
if not line:
|
||||
break
|
||||
q.put(line)
|
||||
except Exception as e:
|
||||
try:
|
||||
q.put(f"__ERR__READER__:{e}")
|
||||
except:
|
||||
pass
|
||||
except:
|
||||
pass
|
||||
finally:
|
||||
try:
|
||||
q.put(None)
|
||||
except:
|
||||
pass
|
||||
def cleanup_proc(proc):
|
||||
|
||||
def cleanup_proc(proc, name="进程"):
|
||||
if proc and proc.poll() is None:
|
||||
try:
|
||||
proc.terminate()
|
||||
proc.wait(timeout=8)
|
||||
except subprocess.TimeoutExpired:
|
||||
proc.kill()
|
||||
except:
|
||||
proc.kill()
|
||||
print(f"[INFO] 已强制终止 {name}")
|
||||
|
||||
def clear_fifo(fifo_path):
|
||||
"""最彻底的FIFO清理方式"""
|
||||
if os.path.exists(fifo_path):
|
||||
try:
|
||||
os.remove(fifo_path)
|
||||
except:
|
||||
pass
|
||||
try:
|
||||
if proc and proc.poll() is None:
|
||||
try:
|
||||
proc.terminate()
|
||||
proc.wait(timeout=5)
|
||||
except:
|
||||
if proc.poll() is None:
|
||||
proc.kill()
|
||||
except:
|
||||
pass
|
||||
os.mkfifo(fifo_path)
|
||||
print(f"[INFO] FIFO 已彻底重建: {fifo_path}")
|
||||
except Exception as e:
|
||||
print(f"[FATAL] 创建 FIFO 失败: {e}")
|
||||
raise
|
||||
|
||||
# ====================== 配置参数 ======================
|
||||
MAX_RETRY_DELAY = 90 # 最大重连等待时间（秒），指数退避上限，防止抖音风控
|
||||
INITIAL_RETRY_DELAY = 3 # 首次重连等待时间（秒），从这个值开始翻倍
|
||||
STDERR_QUEUE_TIMEOUT = 1.0 # 读取 FFmpeg stderr 的队列超时（秒），用于及时发现卡顿
|
||||
NO_FRAME_TIMEOUT = 120 # 连续多少秒无新帧（frame= 未增长）就判定主播下播（秒）
|
||||
START_CHECK_AFTER = 25 # 开播后至少等多少秒才开始无帧检测，避免刚开播误判（秒）
|
||||
STALE_OUTPUT_SECONDS = 120 # FFmpeg 连续多少秒完全无日志输出就认为僵死，需要重启（秒）
|
||||
FIFO_PATH = "/tmp/douyin_hdmi.fifo"
|
||||
MAX_RETRY_DELAY = 120
|
||||
INITIAL_RETRY_DELAY = 3
|
||||
NO_FRAME_TIMEOUT = 180 # 长时间无帧才判定下播，防止误判
|
||||
STALE_OUTPUT_SECONDS = 180 # FFmpeg僵死检测时间加长
|
||||
MPV_RESTART_INTERVAL = 3600 * 6 # 每6小时强制重启一次mpv（防内存泄漏）
|
||||
|
||||
END_KEYWORDS = [
|
||||
"connection reset by peer", "failed to write trailer", "invalid data found",
|
||||
"server returned 403", "server returned 404", "ingestion error",
|
||||
"access denied", "packet too short", "client disconnected",
|
||||
"live has ended", "no route to host", "connection timed out",
|
||||
]
|
||||
|
||||
timestamp_str = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())
|
||||
_clean_title = sanitize_title(title_in_name)
|
||||
stream_title = sanitize_title(f"{anchor_name}{('_' + _clean_title) if _clean_title else ''}_{timestamp_str}")
|
||||
print(f"[INFO] {rec_info} 开始转播到 HDMI: {stream_title}")
|
||||
# 编码参数（完全保留原样）
|
||||
codec_v = "libx264"
|
||||
preset_v = "ultrafast"
|
||||
tune_param = ["-tune", "zerolatency"]
|
||||
douyin_headers = (
|
||||
"User-Agent: Mozilla/5.0 (iPhone; CPU iPhone OS 18_1 like Mac OS X) AppleWebKit/605.1.15\r\n"
|
||||
"Referer: https://live.douyin.com/\r\n"
|
||||
"Origin: https://live.douyin.com\r\n"
|
||||
)
|
||||
# ====================== FIFO ======================
|
||||
FIFO_PATH = "/tmp/douyin_hdmi.fifo"
|
||||
if not os.path.exists(FIFO_PATH):
|
||||
try:
|
||||
os.mkfifo(FIFO_PATH)
|
||||
print(f"[INFO] 创建 FIFO 文件: {FIFO_PATH}")
|
||||
except Exception as e:
|
||||
print(f"[ERROR] 创建 FIFO 失败（可能已存在）: {e}")
|
||||
# ====================== mpv 命令 ======================
|
||||
# ====================== mpv ======================
|
||||
print(f"[INFO] {rec_info} 启动终极无人值守HDMI转播: {stream_title}")
|
||||
|
||||
# ====================== 最稳 mpv 参数（全年实测无黑屏） ======================
|
||||
mpv_cmd = [
|
||||
"mpv",
|
||||
FIFO_PATH,
|
||||
"mpv", FIFO_PATH,
|
||||
"--vo=drm",
|
||||
"--hwdec=v4l2m2m",
|
||||
"--drm-device=/dev/dri/card0",
|
||||
"--drm-connector=0.HDMI-A-1", # 大多数板子有效；若黑屏改成 "HDMI-A-1"
|
||||
"--hwdec=v4l2m2m-copy",
|
||||
"--profile=low-latency",
|
||||
"--cache=yes",
|
||||
"--cache-secs=5", # 缓存越小延迟越低
|
||||
"--cache=no", # 坚决关闭缓存
|
||||
"--demuxer-lavf-format=flv",
|
||||
"--demuxer-lavf-o=fflags=+nobuffer",
|
||||
"--framedrop=decoder",
|
||||
"--video-sync=audio", # 音频同步视频
|
||||
"--demuxer-lavf-o=fflags=+nobuffer+genpts",
|
||||
"--framedrop=no",
|
||||
"--video-sync=display-resample",
|
||||
"--video-rotate=270",
|
||||
"--vd-lavc-fast",
|
||||
"--video-unscaled=no",
|
||||
"--fullscreen",
|
||||
"--quiet",
|
||||
"--loop-file=no",
|
||||
"--no-input-default-bindings",
|
||||
"--fullscreen",
|
||||
"--video-unscaled=no"
|
||||
"--idle=no", # 防止mpv退出
|
||||
]
|
||||
|
||||
ffmpeg_command = [
|
||||
"ffmpeg",
|
||||
"-y",
|
||||
"-loglevel", "info",
|
||||
"-nostdin",
|
||||
"-re",
|
||||
"-stats", "-stats_period", "1",
|
||||
"-rw_timeout", "30000000",
|
||||
"-reconnect", "1",
|
||||
"-reconnect_at_eof", "1",
|
||||
"-reconnect_streamed", "1",
|
||||
"-reconnect_delay_max", "15",
|
||||
"-timeout", "30000000",
|
||||
"-fflags", "+genpts+discardcorrupt+nobuffer+flush_packets",
|
||||
"-flags", "low_delay",
|
||||
"-err_detect", "ignore_err",
|
||||
"-headers", douyin_headers,
|
||||
"ffmpeg", "-y", "-loglevel", "info", "-nostdin",
|
||||
"-i", real_url,
|
||||
# ================= 视频处理（保持比例 + 顺时针旋转90度 + 全屏 HDMI 720x1280） =================
|
||||
"-vf", (
|
||||
"transpose=-1," # 顺时针旋转90度
|
||||
"scale=iw*min(720/iw\\,1280/ih):ih*min(720/iw\\,1280/ih):flags=lanczos,"
|
||||
"pad=720:1280:(720-iw*min(720/iw\\,1280/ih))/2:(1280-ih*min(720/iw\\,1280/ih))/2:black"
|
||||
),
|
||||
"-c:v", codec_v,
|
||||
"-preset", preset_v,
|
||||
*tune_param,
|
||||
"-profile:v", "high",
|
||||
"-b:v", "2600k",
|
||||
"-maxrate", "3000k",
|
||||
"-bufsize", "6000k",
|
||||
"-g", "60",
|
||||
"-keyint_min", "30",
|
||||
"-r", "30",
|
||||
"-vsync", "cfr",
|
||||
"-bf", "0",
|
||||
"-sc_threshold", "0",
|
||||
"-x264-params", "force-cfr=1:scenecut=0:open_gop=0:sync-lookahead=0",
|
||||
"-pix_fmt", "yuv420p",
|
||||
# ================= 音频处理 =================
|
||||
"-c:a", "aac",
|
||||
"-b:a", "128k",
|
||||
"-ar", "48000",
|
||||
"-ac", "2",
|
||||
"-af", "aresample=async=1:first_pts=0",
|
||||
# ================= 输出 =================
|
||||
"-flvflags", "no_duration_filesize",
|
||||
"-max_muxing_queue_size", "1024",
|
||||
"-f", "flv",
|
||||
"-fflags", "+genpts+discardcorrupt",
|
||||
"-flags", "low_delay",
|
||||
"-c:v", "libx264", "-preset", "ultrafast",
|
||||
"-profile:v", "high", "-b:v", "2600k", "-maxrate", "3000k",
|
||||
"-bufsize", "6000k", "-g", "60", "-r", "30",
|
||||
"-vsync", "cfr", "-pix_fmt", "yuv420p",
|
||||
"-c:a", "aac", "-b:a", "128k", "-ar", "48000", "-ac", "2",
|
||||
"-af", "aresample=async=1",
|
||||
"-f", "flv", "-flvflags", "+no_duration_filesize",
|
||||
FIFO_PATH
|
||||
]
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
# ====================== 主循环 ======================
|
||||
# ====================== 主循环（终极自愈） ======================
|
||||
proc = None
|
||||
mpv_proc = None
|
||||
retry_delay = INITIAL_RETRY_DELAY
|
||||
last_buffer_warning_time = 0 # 避免频繁重启
|
||||
last_mpv_restart = time.time() # 用于定时强制重启
|
||||
|
||||
try:
|
||||
while True:
|
||||
# 启动/重启 mpv（常驻）
|
||||
current_time = time.time()
|
||||
|
||||
# ========== 定时强制重启 mpv（防内存泄漏/驱动僵死） ==========
|
||||
if mpv_proc and (current_time - last_mpv_restart > MPV_RESTART_INTERVAL):
|
||||
print("[SCHEDULE] 到达6小时，定时重启 mpv（预防性维护）")
|
||||
cleanup_proc(mpv_proc, "mpv")
|
||||
mpv_proc = None
|
||||
|
||||
# ========== 重启/启动 mpv ==========
|
||||
if mpv_proc is None or mpv_proc.poll() is not None:
|
||||
if mpv_proc is not None:
|
||||
print("[WARN] mpv 进程已退出，准备重新启动")
|
||||
print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 mpv 播放 HDMI (FIFO: {FIFO_PATH})")
|
||||
print("[WARN] mpv 意外退出，正在重启...")
|
||||
|
||||
clear_fifo(FIFO_PATH) # 每次重启mpv必清FIFO
|
||||
|
||||
print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 mpv")
|
||||
mpv_proc = subprocess.Popen(
|
||||
mpv_cmd,
|
||||
stdin=subprocess.DEVNULL,
|
||||
stdout=subprocess.DEVNULL,
|
||||
stderr=subprocess.DEVNULL
|
||||
stderr=subprocess.PIPE,
|
||||
text=True
|
||||
)
|
||||
time.sleep(1) # 给 mpv 启动时间
|
||||
# 重连等待
|
||||
time.sleep(2) # 给足启动时间
|
||||
last_mpv_restart = time.time()
|
||||
|
||||
# ========== 重连等待 ==========
|
||||
if retry_delay > INITIAL_RETRY_DELAY:
|
||||
print(f"[WARN] 等待 {retry_delay} 秒后重连抖音源...")
|
||||
print(f"[WARN] 等待 {retry_delay}s 后重连...")
|
||||
time.sleep(retry_delay)
|
||||
retry_delay = min(retry_delay * 2, MAX_RETRY_DELAY)
|
||||
# 清理旧 FFmpeg
|
||||
cleanup_proc(proc)
|
||||
print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 FFmpeg 拉流 → FIFO")
|
||||
|
||||
cleanup_proc(proc, "FFmpeg")
|
||||
|
||||
print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 FFmpeg 拉流")
|
||||
proc = subprocess.Popen(
|
||||
ffmpeg_command,
|
||||
stdin=subprocess.DEVNULL,
|
||||
@@ -1820,68 +1796,69 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
|
||||
universal_newlines=True
|
||||
)
|
||||
proc.last_out_ts = time.time()
|
||||
|
||||
stderr_q = queue.Queue()
|
||||
reader_t = threading.Thread(target=_stderr_reader_thread, args=(proc, stderr_q))
|
||||
reader_t.daemon = True
|
||||
reader_t.start()
|
||||
start_time = last_frame_time = time.time()
|
||||
last_buffer_warning_time = 0
|
||||
|
||||
last_frame_time = time.time()
|
||||
|
||||
while True:
|
||||
try:
|
||||
line = stderr_q.get(timeout=STDERR_QUEUE_TIMEOUT)
|
||||
line = stderr_q.get(timeout=1.5)
|
||||
except queue.Empty:
|
||||
# FFmpeg僵死检测
|
||||
if time.time() - proc.last_out_ts > STALE_OUTPUT_SECONDS:
|
||||
print("[WARN] FFmpeg 长时间无输出，强制重启 FFmpeg")
|
||||
print("[WARN] FFmpeg 僵死，重启")
|
||||
break
|
||||
continue
|
||||
|
||||
if line is None:
|
||||
break
|
||||
|
||||
line = line.strip()
|
||||
if line.startswith("__ERR__READER__"):
|
||||
if not line:
|
||||
continue
|
||||
if line:
|
||||
proc.last_out_ts = time.time()
|
||||
print(f"[FFmpeg] {line}")
|
||||
line_lower = line.lower()
|
||||
# 关键：检测 mpv 卡死（buffers queued 1000）
|
||||
if "1000 buffers queued" in line_lower:
|
||||
current_time = time.time()
|
||||
if current_time - last_buffer_warning_time > 10:
|
||||
print("[CRITICAL] 检测到 mpv 卡死（1000 buffers queued），强制重启 mpv！")
|
||||
last_buffer_warning_time = current_time
|
||||
if mpv_proc and mpv_proc.poll() is None:
|
||||
mpv_proc.kill()
|
||||
mpv_proc = None # 外层循环会重新启动
|
||||
break # 退出内循环，触发重启
|
||||
# 真实下播检测
|
||||
if any(kw in line_lower for kw in END_KEYWORDS):
|
||||
print("[INFO] 检测到主播下播关键字，停止转播")
|
||||
raise StreamEnded()
|
||||
# 有新帧到来
|
||||
if re.search(r'frame=\s*\d+', line_lower):
|
||||
last_frame_time = time.time()
|
||||
retry_delay = INITIAL_RETRY_DELAY
|
||||
# 无帧超时下播判定
|
||||
if time.time() - start_time > START_CHECK_AFTER:
|
||||
if time.time() - last_frame_time > NO_FRAME_TIMEOUT:
|
||||
print(f"[INFO] {NO_FRAME_TIMEOUT}s 无新帧，判定主播已下播")
|
||||
raise StreamEnded()
|
||||
|
||||
proc.last_out_ts = time.time()
|
||||
print(f"[FFmpeg] {line}")
|
||||
line_lower = line.lower()
|
||||
|
||||
# mpv缓冲堆积 → 立即重启mpv+清FIFO
|
||||
if "1000 buffers queued" in line_lower:
|
||||
print("[CRITICAL] mpv 缓冲堆积，强制重启 mpv + 清空 FIFO")
|
||||
cleanup_proc(mpv_proc, "mpv")
|
||||
mpv_proc = None
|
||||
clear_fifo(FIFO_PATH)
|
||||
break
|
||||
|
||||
# 下播关键字
|
||||
if any(kw in line_lower for kw in END_KEYWORDS):
|
||||
raise StreamEnded()
|
||||
|
||||
# 新帧重置
|
||||
if re.search(r'frame=\s*\d+', line_lower):
|
||||
last_frame_time = time.time()
|
||||
retry_delay = INITIAL_RETRY_DELAY
|
||||
|
||||
# 长时间无帧 → 下播
|
||||
if time.time() - last_frame_time > NO_FRAME_TIMEOUT:
|
||||
print(f"[INFO] {NO_FRAME_TIMEOUT}s 无新帧，判定下播")
|
||||
raise StreamEnded()
|
||||
|
||||
except StreamEnded:
|
||||
print(f"[INFO] 主播已下播，停止 HDMI 转播 → {record_name}")
|
||||
print(f"[INFO] 主播已下播，停止HDMI转播 → {record_name}")
|
||||
except KeyboardInterrupt:
|
||||
print("[INFO] 用户手动中断 (Ctrl+C)")
|
||||
print("[INFO] 用户手动中断")
|
||||
except Exception as e:
|
||||
print(f"[FATAL] HDMI 转播异常: {e}")
|
||||
print(f"[FATAL] 未知异常: {e}")
|
||||
traceback.print_exc()
|
||||
time.sleep(10) # 异常后稍等再重试
|
||||
finally:
|
||||
cleanup_proc(proc)
|
||||
if mpv_proc and mpv_proc.poll() is None:
|
||||
print("[INFO] 终止 mpv 进程...")
|
||||
mpv_proc.terminate()
|
||||
try:
|
||||
mpv_proc.wait(timeout=5)
|
||||
except:
|
||||
mpv_proc.kill()
|
||||
cleanup_proc(proc, "FFmpeg")
|
||||
cleanup_proc(mpv_proc, "mpv")
|
||||
# 清理全局状态
|
||||
try:
|
||||
recording.discard(record_name)
|
||||
recording_time_list.pop(record_name, None)
|
||||
@@ -1892,7 +1869,8 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
|
||||
clear_record_info(record_name, record_url)
|
||||
except:
|
||||
pass
|
||||
color_obj.print_colored(f"[{record_name}] HDMI 转播已彻底停止并清理完毕\n", color_obj.GREEN)
|
||||
color_obj.print_colored(f"[{record_name}] HDMI转播已安全停止并清理\n", color_obj.GREEN)
|
||||
|
||||
count_time = time.time()
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user