diff --git a/douyin_srs_ffplay.py b/douyin_srs_ffplay.py deleted file mode 100644 index 2790629..0000000 --- a/douyin_srs_ffplay.py +++ /dev/null @@ -1,230 +0,0 @@ -# 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) diff --git a/main.py b/main.py index 564130e..6f2da1f 100644 --- a/main.py +++ b/main.py @@ -1620,6 +1620,8 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None: error_window.append(1) else: + import sys + sys.path.insert(0, './dist') from douyin_srs_ffplay import start_douyin_srs_ffplay # 调用 start_douyin_srs_ffplay( @@ -1635,223 +1637,6 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None: 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 - - # 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" - # ) - - # # ====================== 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 - # ] - - # # ====================== 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_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()