s
This commit is contained in:
Your Name
406
main.py
406
main.py
@@ -1618,223 +1618,239 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
|
||||
with max_request_lock:
|
||||
error_count += 1
|
||||
error_window.append(1)
|
||||
|
||||
|
||||
else:
|
||||
# ====================== 抖音 → SRS → ffplay HDMI 终极低CPU清晰版 ======================
|
||||
import queue
|
||||
import unicodedata
|
||||
import traceback
|
||||
# import time
|
||||
# import datetime
|
||||
# import subprocess
|
||||
# import threading
|
||||
# import re
|
||||
|
||||
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}")
|
||||
|
||||
# ====================== 配置参数 ======================
|
||||
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_in_name)
|
||||
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"
|
||||
from douyin_srs_ffplay import start_douyin_srs_ffplay
|
||||
# 调用
|
||||
start_douyin_srs_ffplay(
|
||||
title=title_in_name,
|
||||
anchor_name=anchor_name,
|
||||
real_url=real_url,
|
||||
rec_info=rec_info,
|
||||
record_name=record_name,
|
||||
recording=recording,
|
||||
recording_time_list=recording_time_list,
|
||||
running_list=running_list,
|
||||
monitoring=monitoring,
|
||||
clear_record_info=clear_record_info,
|
||||
color_obj=color_obj
|
||||
)
|
||||
# else:
|
||||
# # ====================== 抖音 → SRS → ffplay HDMI 终极低CPU清晰版 ======================
|
||||
# import queue
|
||||
# import unicodedata
|
||||
# import traceback
|
||||
# # import time
|
||||
# # import datetime
|
||||
# # import subprocess
|
||||
# # import threading
|
||||
# # import re
|
||||
|
||||
# ====================== 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", # 直抄视频，CPU极低，画质取决于源（抖音通常给原画）
|
||||
"-c:a", "copy", # 直抄音频
|
||||
"-f", "flv", SRS_RTMP_URL
|
||||
]
|
||||
# class StreamEnded(Exception):
|
||||
# pass
|
||||
|
||||
# ====================== 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
|
||||
]
|
||||
# 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]
|
||||
|
||||
ffmpeg_proc = None
|
||||
ffplay_proc = None
|
||||
stderr_q = queue.Queue()
|
||||
reader_t = None
|
||||
retry_delay = INITIAL_RETRY_DELAY
|
||||
last_ffplay_restart = time.time()
|
||||
# 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
|
||||
|
||||
try:
|
||||
while True:
|
||||
current_time = time.time()
|
||||
# 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}")
|
||||
|
||||
if ffplay_proc and (current_time - last_ffplay_restart > FFPLAY_RESTART_INTERVAL):
|
||||
print("[SCHEDULE] 6小时预防性重启 ffplay")
|
||||
cleanup_proc(ffplay_proc, "ffplay")
|
||||
ffplay_proc = None
|
||||
# # ====================== 配置参数 ======================
|
||||
# SRS_RTMP_URL = "rtmp://127.0.0.1/live/douyin"
|
||||
# SRS_PLAY_URL = "http://127.0.0.1:8080/live/douyin.flv"
|
||||
|
||||
if ffplay_proc is None or ffplay_proc.poll() is not None:
|
||||
if ffplay_proc is not None:
|
||||
print("[WARN] ffplay 退出，重启中...")
|
||||
# MAX_RETRY_DELAY = 120
|
||||
# INITIAL_RETRY_DELAY = 3
|
||||
# NO_FRAME_TIMEOUT = 180
|
||||
# STALE_OUTPUT_SECONDS = 180
|
||||
# FFPLAY_RESTART_INTERVAL = 3600 * 6
|
||||
|
||||
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()
|
||||
# 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",
|
||||
# ]
|
||||
|
||||
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)
|
||||
# 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} 启动 SRS → ffplay HDMI 转播（低CPU原画版）: {stream_title}")
|
||||
|
||||
cleanup_proc(ffmpeg_proc, "FFmpeg")
|
||||
# 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"
|
||||
# )
|
||||
|
||||
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()
|
||||
# # ====================== 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", # 直抄视频，CPU极低，画质取决于源（抖音通常给原画）
|
||||
# "-c:a", "copy", # 直抄音频
|
||||
# "-f", "flv", SRS_RTMP_URL
|
||||
# ]
|
||||
|
||||
stderr_q = queue.Queue()
|
||||
reader_t = threading.Thread(target=_stderr_reader_thread, args=(ffmpeg_proc, stderr_q))
|
||||
reader_t.daemon = True
|
||||
reader_t.start()
|
||||
# # ====================== 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
|
||||
# ]
|
||||
|
||||
last_frame_time = time.time()
|
||||
# ffmpeg_proc = None
|
||||
# ffplay_proc = None
|
||||
# stderr_q = queue.Queue()
|
||||
# reader_t = None
|
||||
# retry_delay = INITIAL_RETRY_DELAY
|
||||
# last_ffplay_restart = 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
|
||||
# try:
|
||||
# while True:
|
||||
# current_time = time.time()
|
||||
|
||||
if line is None:
|
||||
break
|
||||
# if ffplay_proc and (current_time - last_ffplay_restart > FFPLAY_RESTART_INTERVAL):
|
||||
# print("[SCHEDULE] 6小时预防性重启 ffplay")
|
||||
# cleanup_proc(ffplay_proc, "ffplay")
|
||||
# ffplay_proc = None
|
||||
|
||||
line = line.strip()
|
||||
if not line:
|
||||
continue
|
||||
# if ffplay_proc is None or ffplay_proc.poll() is not None:
|
||||
# if ffplay_proc is not None:
|
||||
# print("[WARN] ffplay 退出，重启中...")
|
||||
|
||||
ffmpeg_proc.last_out_ts = time.time()
|
||||
print(f"[FFmpeg] {line}")
|
||||
line_lower = line.lower()
|
||||
# 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 any(kw in line_lower for kw in END_KEYWORDS):
|
||||
print("[INFO] 检测到下播关键字，停止转播")
|
||||
raise StreamEnded()
|
||||
# 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)
|
||||
|
||||
if re.search(r'frame=\s*\d+', line_lower):
|
||||
last_frame_time = time.time()
|
||||
retry_delay = INITIAL_RETRY_DELAY
|
||||
# cleanup_proc(ffmpeg_proc, "FFmpeg")
|
||||
|
||||
if time.time() - last_frame_time > NO_FRAME_TIMEOUT:
|
||||
print(f"[INFO] {NO_FRAME_TIMEOUT}s 无新帧，判定下播")
|
||||
raise StreamEnded()
|
||||
# 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()
|
||||
|
||||
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_url in running_list:
|
||||
running_list.remove(record_url)
|
||||
global monitoring
|
||||
monitoring = max(0, monitoring - 1)
|
||||
clear_record_info(record_name, record_url)
|
||||
except:
|
||||
pass
|
||||
color_obj.print_colored(f"[{record_name}] SRS → ffplay HDMI 转播已彻底停止并清理完毕\n", color_obj.GREEN)
|
||||
# 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_url in running_list:
|
||||
# running_list.remove(record_url)
|
||||
# global monitoring
|
||||
# monitoring = max(0, monitoring - 1)
|
||||
# clear_record_info(record_name, record_url)
|
||||
# except:
|
||||
# pass
|
||||
# color_obj.print_colored(f"[{record_name}] SRS → ffplay HDMI 转播已彻底停止并清理完毕\n", color_obj.GREEN)
|
||||
|
||||
count_time = time.time()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user