231 lines
7.6 KiB
Python
231 lines
7.6 KiB
Python
# douyin_srs_ffplay.py
|
||
|
||
import queue
|
||
import unicodedata
|
||
import traceback
|
||
import subprocess
|
||
import threading
|
||
import re
|
||
import time
|
||
import datetime
|
||
|
||
class StreamEnded(Exception):
|
||
pass
|
||
|
||
def sanitize_title(text: str, max_len: int = 120) -> str:
|
||
if not text:
|
||
return ""
|
||
text = unicodedata.normalize("NFKC", text)
|
||
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:
|
||
line = proc.stderr.readline()
|
||
if not line:
|
||
break
|
||
q.put(line)
|
||
except:
|
||
pass
|
||
finally:
|
||
try:
|
||
q.put(None)
|
||
except:
|
||
pass
|
||
|
||
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 start_douyin_srs_ffplay(
|
||
title,
|
||
anchor_name,
|
||
real_url,
|
||
rec_info,
|
||
record_name,
|
||
recording,
|
||
recording_time_list,
|
||
running_list,
|
||
monitoring,
|
||
clear_record_info,
|
||
color_obj
|
||
):
|
||
# ====================== 配置参数 ======================
|
||
SRS_RTMP_URL = "rtmp://127.0.0.1/live/douyin"
|
||
SRS_PLAY_URL = "http://127.0.0.1:8080/live/douyin.flv"
|
||
|
||
MAX_RETRY_DELAY = 120
|
||
INITIAL_RETRY_DELAY = 3
|
||
NO_FRAME_TIMEOUT = 180
|
||
STALE_OUTPUT_SECONDS = 180
|
||
FFPLAY_RESTART_INTERVAL = 3600 * 6
|
||
|
||
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)
|
||
stream_title = sanitize_title(f"{anchor_name}{('_' + _clean_title) if _clean_title else ''}_{timestamp_str}")
|
||
print(f"[INFO] {rec_info} 启动 SRS → ffplay HDMI 转播（低CPU原画版）: {stream_title}")
|
||
|
||
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"
|
||
)
|
||
|
||
# ====================== FFmpeg 推流到 SRS（直抄原画，低CPU） ======================
|
||
ffmpeg_command = [
|
||
"ffmpeg", "-y", "-loglevel", "info", "-nostdin",
|
||
"-headers", douyin_headers,
|
||
"-rw_timeout", "30000000",
|
||
"-reconnect", "1", "-reconnect_at_eof", "1", "-reconnect_streamed", "1",
|
||
"-reconnect_delay_max", "7",
|
||
"-fflags", "+genpts+discardcorrupt+igndts",
|
||
"-err_detect", "ignore_err",
|
||
"-i", real_url,
|
||
"-c:v", "copy",
|
||
"-c:a", "copy",
|
||
"-f", "flv", SRS_RTMP_URL
|
||
]
|
||
|
||
# ====================== ffplay 最佳参数（低CPU + 零漂移 + 全屏） ======================
|
||
ffplay_cmd = [
|
||
"ffplay",
|
||
"-fflags", "nobuffer+genpts",
|
||
"-flags", "low_delay",
|
||
"-framedrop",
|
||
"-sync", "ext",
|
||
"-vf", "transpose=2",
|
||
"-autoexit",
|
||
"-fs",
|
||
"-loglevel", "quiet",
|
||
"-i", SRS_PLAY_URL
|
||
]
|
||
|
||
ffmpeg_proc = None
|
||
ffplay_proc = None
|
||
stderr_q = queue.Queue()
|
||
reader_t = None
|
||
retry_delay = INITIAL_RETRY_DELAY
|
||
last_ffplay_restart = time.time()
|
||
|
||
try:
|
||
while True:
|
||
current_time = time.time()
|
||
|
||
if ffplay_proc and (current_time - last_ffplay_restart > FFPLAY_RESTART_INTERVAL):
|
||
print("[SCHEDULE] 6小时预防性重启 ffplay")
|
||
cleanup_proc(ffplay_proc, "ffplay")
|
||
ffplay_proc = None
|
||
|
||
if ffplay_proc is None or ffplay_proc.poll() is not None:
|
||
if ffplay_proc is not None:
|
||
print("[WARN] ffplay 退出，重启中...")
|
||
|
||
print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 ffplay 播放 SRS 原画流")
|
||
ffplay_proc = subprocess.Popen(
|
||
ffplay_cmd,
|
||
stdin=subprocess.DEVNULL,
|
||
stdout=subprocess.DEVNULL,
|
||
stderr=subprocess.PIPE,
|
||
text=True
|
||
)
|
||
time.sleep(2)
|
||
last_ffplay_restart = time.time()
|
||
|
||
if retry_delay > INITIAL_RETRY_DELAY:
|
||
print(f"[WARN] 等待 {retry_delay}s 后重连抖音源...")
|
||
time.sleep(retry_delay)
|
||
retry_delay = min(retry_delay * 2, MAX_RETRY_DELAY)
|
||
|
||
cleanup_proc(ffmpeg_proc, "FFmpeg")
|
||
|
||
print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 FFmpeg 直抄推流到 SRS")
|
||
ffmpeg_proc = subprocess.Popen(
|
||
ffmpeg_command,
|
||
stdin=subprocess.DEVNULL,
|
||
stdout=subprocess.DEVNULL,
|
||
stderr=subprocess.PIPE,
|
||
text=True,
|
||
bufsize=1,
|
||
universal_newlines=True
|
||
)
|
||
ffmpeg_proc.last_out_ts = time.time()
|
||
|
||
stderr_q = queue.Queue()
|
||
reader_t = threading.Thread(target=_stderr_reader_thread, args=(ffmpeg_proc, stderr_q))
|
||
reader_t.daemon = True
|
||
reader_t.start()
|
||
|
||
last_frame_time = time.time()
|
||
|
||
while True:
|
||
try:
|
||
line = stderr_q.get(timeout=1.5)
|
||
except queue.Empty:
|
||
if time.time() - ffmpeg_proc.last_out_ts > STALE_OUTPUT_SECONDS:
|
||
print("[WARN] FFmpeg 僵死，重启")
|
||
break
|
||
continue
|
||
|
||
if line is None:
|
||
break
|
||
|
||
line = line.strip()
|
||
if not line:
|
||
continue
|
||
|
||
ffmpeg_proc.last_out_ts = time.time()
|
||
print(f"[FFmpeg] {line}")
|
||
line_lower = line.lower()
|
||
|
||
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() - last_frame_time > NO_FRAME_TIMEOUT:
|
||
print(f"[INFO] {NO_FRAME_TIMEOUT}s 无新帧，判定下播")
|
||
raise StreamEnded()
|
||
|
||
except StreamEnded:
|
||
print(f"[INFO] 主播已下播，停止 SRS → ffplay HDMI 转播 → {record_name}")
|
||
except KeyboardInterrupt:
|
||
print("[INFO] 用户手动中断")
|
||
except Exception as e:
|
||
print(f"[FATAL] 未知异常: {e}")
|
||
traceback.print_exc()
|
||
time.sleep(10)
|
||
finally:
|
||
cleanup_proc(ffmpeg_proc, "FFmpeg")
|
||
cleanup_proc(ffplay_proc, "ffplay")
|
||
try:
|
||
recording.discard(record_name)
|
||
recording_time_list.pop(record_name, None)
|
||
if record_name in running_list:
|
||
running_list.remove(record_name)
|
||
# global monitoring
|
||
# monitoring = max(0, monitoring - 1)
|
||
monitoring[0] = max(0, monitoring[0] - 1) # 修改为列表元素
|
||
clear_record_info(record_name, real_url)
|
||
except:
|
||
pass
|
||
color_obj.print_colored(f"[{record_name}] SRS → ffplay HDMI 转播已彻底停止并清理完毕\n", color_obj.GREEN)
|