eric/gitlab-instance-0a899031_dy2ytlive
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

's'

Browse Source
This commit is contained in:
eric
2025-09-13 12:26:59 +08:00
parent 6bbf547ab2
commit 752f11e9a6
2 changed files with 233 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
live.py
View File

@@ -11,11 +11,10 @@ import shutil
import spider
import stream
# ----------------- 自适应推流器 -----------------
class AdaptiveHLS2YouTube:
def __init__(self, max_bitrate=5000, min_bitrate=4000, audio_bitrate=192,
delay_sec=30, segment_time=2, check_interval=5):
self.processes = {} # room_name -> ffmpeg process
def __init__(self, max_bitrate=5000, min_bitrate=3000, audio_bitrate=192,
delay_sec=40, segment_time=2, check_interval=5):
self.processes = {}
self.stop_flag = threading.Event()
self.delay_sec = delay_sec
self.segment_time = segment_time
@@ -26,24 +25,14 @@ class AdaptiveHLS2YouTube:
self.audio_bitrate = audio_bitrate
self.current_bitrate = max_bitrate
def start_stream(self, real_urls, youtube_key, room_name=None):
"""
支持 real_urls 列表，主备流分别推送
"""
def start_stream(self, real_url, youtube_key, room_name=None):
room_name = room_name or f"抖音_{str(uuid.uuid4())[:8]}"
print(f"[{room_name}] 启动推流，目标 YouTube Key: {youtube_key}")
# 启动拉流线程（可支持多源）
for idx, url in enumerate(real_urls):
threading.Thread(target=self._pull_thread, args=(url, idx), daemon=True).start()
# 推送线程
threading.Thread(target=self._pull_thread, args=(real_url,), daemon=True).start()
threading.Thread(target=self._push_thread, args=(youtube_key, room_name), daemon=True).start()
# ABR线程
threading.Thread(target=self._abr_thread, daemon=True).start()
def _pull_thread(self, real_url, idx):
def _pull_thread(self, real_url):
cmd = [
"ffmpeg",
"-rw_timeout", "50000000",
@@ -62,15 +51,15 @@ class AdaptiveHLS2YouTube:
]
try:
proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL)
print(f"[拉流 {idx}] FFmpeg 启动成功")
print("[拉流] FFmpeg 启动成功")
while not self.stop_flag.is_set():
data = proc.stdout.read(188*50)
data = proc.stdout.read(188*500)
if not data:
time.sleep(0.05)
continue
self.buffer_queue.put(data)
except Exception as e:
print(f"[拉流 {idx}] 异常: {e}")
print(f"拉流异常: {e}")
def _push_thread(self, youtube_key, room_name):
min_buffer = max(1, self.delay_sec // self.segment_time)
@@ -89,9 +78,10 @@ class AdaptiveHLS2YouTube:
"-tune", "zerolatency",
"-b:v", f"{self.current_bitrate}k",
"-maxrate", f"{self.current_bitrate}k",
"-bufsize", f"{self.current_bitrate*2}k",
"-bufsize", f"{self.current_bitrate*4}k",
"-r", "30",
"-g", "60",
# 不固定分辨率，保持抖音原始分辨率
"-c:a", "aac",
"-b:a", f"{self.audio_bitrate}k",
"-ar", "44100",
@@ -112,15 +102,12 @@ class AdaptiveHLS2YouTube:
threading.Thread(target=read_stderr, daemon=True).start()
while not self.stop_flag.is_set():
if not self.buffer_queue.empty():
data = self.buffer_queue.get()
try:
proc.stdin.write(data)
proc.stdin.flush()
except Exception:
break
else:
time.sleep(0.01)
try:
data = self.buffer_queue.get(timeout=1)
proc.stdin.write(data)
proc.stdin.flush()
except queue.Empty:
continue
except Exception as e:
print(f"[{room_name}] 推流异常: {e}")
@@ -128,12 +115,11 @@ class AdaptiveHLS2YouTube:
while not self.stop_flag.is_set():
queue_len = self.buffer_queue.qsize()
if queue_len > 100:
new_bitrate = max(self.min_bitrate, self.current_bitrate - 200)
new_bitrate = max(self.min_bitrate, self.current_bitrate - 300)
elif queue_len < 30:
new_bitrate = min(self.max_bitrate, self.current_bitrate + 200)
new_bitrate = min(self.max_bitrate, self.current_bitrate + 300)
else:
new_bitrate = self.current_bitrate
if new_bitrate != self.current_bitrate:
print(f"[ABR] 调整码率: {self.current_bitrate}k -> {new_bitrate}k")
self.current_bitrate = new_bitrate
@@ -146,6 +132,7 @@ class AdaptiveHLS2YouTube:
if proc.poll() is None:
proc.terminate()
# ----------------- 抖音解析逻辑 -----------------
async def get_douyin_real_url(record_url, record_quality="原画"):
if 'v.douyin.com' not in record_url and '/user/' not in record_url:
@@ -155,25 +142,23 @@ async def get_douyin_real_url(record_url, record_quality="原画"):
port_info = await stream.get_douyin_stream_url(json_data, record_quality, None)
return port_info.get("record_url")
# ----------------- 启动推流 -----------------
def start_douyin_to_youtube(record_url, youtube_key, room_name=None, delay_sec=30):
def start_douyin_to_youtube(record_url, youtube_key, room_name=None, delay_sec=40):
try:
# 获取主/备流
real_url_main = asyncio.run(get_douyin_real_url(record_url))
real_url_backup = real_url_main.replace("stage", "backup") # 简单替换为备份流
real_url = asyncio.run(get_douyin_real_url(record_url))
except Exception as e:
print(f"获取真实直播流失败: {e}")
return
if not real_url_main:
if not real_url:
print("获取真实直播流失败")
return
print(f"实际直播流主: {real_url_main}")
print(f"实际直播流备: {real_url_backup}")
print(f"实际直播流: {real_url}")
hls2yt = AdaptiveHLS2YouTube(delay_sec=delay_sec)
hls2yt.start_stream([real_url_main, real_url_backup], youtube_key, room_name)
hls2yt.start_stream(real_url, youtube_key, room_name)
return hls2yt
# ----------------- 控制台监控面板 -----------------
def monitor_panel_thread(hls_instance, interval=2, max_width=50):
term_width, _ = shutil.get_terminal_size()
@@ -191,12 +176,13 @@ def monitor_panel_thread(hls_instance, interval=2, max_width=50):
print(f"[监控] 队列: [{queue_bar}] {queue_len:4} | 码率: {bitrate:4}k | 推流活跃: {active_push}", end="")
time.sleep(interval)
# ----------------- 示例 -----------------
# ----------------- 主程序 -----------------
if __name__ == "__main__":
DOUYIN_URL = "https://live.douyin.com/364595817295"
DOUYIN_URL = "https://live.douyin.com/851742623054"
YOUTUBE_KEY = "qxvb-r47b-r5ju-6ud3-6k7z"
hls_instance = start_douyin_to_youtube(DOUYIN_URL, YOUTUBE_KEY, delay_sec=30)
hls_instance = start_douyin_to_youtube(DOUYIN_URL, YOUTUBE_KEY, delay_sec=40)
if not hls_instance:
print("推流实例创建失败，程序退出")
sys.exit(1)