github/DouyinLiveRecorder
1
0
Fork 0
mirror of https://github.com/ihmily/DouyinLiveRecorder.git synced 2025-12-26 05:48:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: update shopee live record

Browse Source
This commit is contained in:
ihmily 2024-11-15 11:51:00 +08:00
parent 310390b3c2
commit 9d292cf865
3 changed files with 26 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config/config.ini
View File

@ -24,7 +24,7 @@ ts录制完成后自动转为mp4格式 = 是
生成时间字幕文件 = 否
是否录制完成后执行自定义脚本 = 否
自定义脚本执行命令 =
使用代理录制的平台(逗号分隔) = tiktok, sooplive, pandalive, winktv, flextv, popkontv, twitch, liveme, showroom, chzzk
使用代理录制的平台(逗号分隔) = tiktok, sooplive, pandalive, winktv, flextv, popkontv, twitch, liveme, showroom, chzzk, shopee, shp
额外使用代理录制的平台(逗号分隔) =
[推送配置]

20
douyinliverecorder/spider.py
View File

@ -2941,17 +2941,20 @@ def get_shopee_stream_url(url: str, proxy_addr: OptionalStr = None, cookies: Opt
headers['Cookie'] = cookies
result = {"anchor_name": "", "is_live": False}
if 'live.shopee' not in url:
url = get_req(url, proxy_addr=proxy_addr, headers=headers, redirect_url=True)
url = get_req(url, proxy_addr=proxy_addr, headers=headers, redirect_url=True, abroad=True)
uid = get_params(url, 'uid')
host_suffix = url.split('/')[2].rsplit('.', maxsplit=1)[1]
api_host = f'https://live.shopee.{host_suffix}'
if uid:
json_str = get_req(f'https://live.shopee.sg/api/v1/shop_page/live/ongoing?uid={uid}',
proxy_addr=proxy_addr, headers=headers)
json_str = get_req(f'{api_host}/api/v1/shop_page/live/ongoing?uid={uid}',
proxy_addr=proxy_addr, headers=headers, abroad=True)
json_data = json.loads(json_str)
if not json_data['data']['ongoing_live']:
json_str = get_req(
f'https://live.shopee.sg/api/v1/shop_page/live/replay_list?offset=0&limit=1&uid={uid}',
proxy_addr=proxy_addr, headers=headers)
json_str = get_req(f'{api_host}/api/v1/shop_page/live/replay_list?offset=0&limit=1&uid={uid}',
proxy_addr=proxy_addr, headers=headers, abroad=True)
json_data = json.loads(json_str)
result['anchor_name'] = json_data['data']['replay'][0]['nick_name']
return result
@ -2960,8 +2963,7 @@ def get_shopee_stream_url(url: str, proxy_addr: OptionalStr = None, cookies: Opt
else:
session_id = get_params(url, 'session')
json_str = get_req(f'https://live.shopee.sg/api/v1/session/{session_id}',
proxy_addr=proxy_addr, headers=headers)
json_str = get_req(f'{api_host}/api/v1/session/{session_id}', proxy_addr=proxy_addr, headers=headers, abroad=True)
json_data = json.loads(json_str)
if not json_data.get('data'):
print("Fetch shopee live data failed, please update the address of the live broadcast room and try again.")
@ -2977,4 +2979,4 @@ def get_shopee_stream_url(url: str, proxy_addr: OptionalStr = None, cookies: Opt
result['title'] = json_data['data']['session']['title']
result['flv_url'] = flv_url
result['record_url'] = flv_url
return result
return result

21
main.py
View File

@ -37,8 +37,8 @@ from msg_push import (
version = "v4.0.1"
platforms = ("\n国内站点：抖音|快手|虎牙|斗鱼|YY|B站|小红书|bigo|blued|网易CC|千度热播|猫耳FM|Look|TwitCasting|百度|微博|"
"酷狗|花椒|流星|Acfun|畅聊|映客|音播|知乎|嗨秀|VV星球|17Live|浪Live|漂漂|六间房|乐嗨|花猫|shopee"
"\n海外站点：TikTok|SOOP|PandaTV|WinkTV|FlexTV|PopkonTV|TwitchTV|LiveMe|ShowRoom|CHZZK")
"酷狗|花椒|流星|Acfun|畅聊|映客|音播|知乎|嗨秀|VV星球|17Live|浪Live|漂漂|六间房|乐嗨|花猫"
"\n海外站点：TikTok|SOOP|PandaTV|WinkTV|FlexTV|PopkonTV|TwitchTV|LiveMe|ShowRoom|CHZZK|shopee")
recording = set()
error_count = 0
@ -403,6 +403,7 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
record_quality, record_url, anchor_name = url_data
proxy_address = proxy_addr
platform = '未知平台'
live_domain = '/'.join(record_url.split('/')[0:3])
if proxy_addr:
proxy_address = None
@ -928,6 +929,9 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
if enable_https_recording and real_url.startswith("http://"):
real_url = real_url.replace("http://", "https://")
if platform == 'shopee':
real_url = real_url.replace("https://", "http://")
user_agent = ("Mozilla/5.0 (Linux; Android 11; SAMSUNG SM-G973U) AppleWebKit/537.36 ("
"KHTML, like Gecko) SamsungBrowser/14.2 Chrome/87.0.4280.141 Mobile "
"Safari/537.36")
@ -975,6 +979,7 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
'千度热播': 'referer:https://qiandurebo.com',
'17Live': 'referer:https://17.live/en/live/6302408',
'浪Live': 'referer:https://www.lang.live',
'shopee': f'origin:{live_domain}',
}
headers = record_headers.get(platform)
@ -1015,11 +1020,13 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
try:
flv_url = port_info.get('flv_url')
if flv_url:
_filepath, _ = urllib.request.urlretrieve(real_url, save_file_path)
_filepath, _ = urllib.request.urlretrieve(flv_url, save_file_path)
record_finished = True
recording.discard(record_name)
print(
f"\n{anchor_name} {time.strftime('%Y-%m-%d %H:%M:%S')} 直播录制完成\n")
else:
logger.debug("未找到FLV直播流，跳过录制")
except Exception as e:
print(
f"\n{anchor_name} {time.strftime('%Y-%m-%d %H:%M:%S')} 直播录制出错,请检查网络\n")
@ -1494,7 +1501,7 @@ while True:
custom_script = read_config_value(config, '录制设置', '自定义脚本执行命令', "") if is_run_script else None
enable_proxy_platform = read_config_value(
config, '录制设置', '使用代理录制的平台(逗号分隔)',
'tiktok, soop, pandalive, winktv, flextv, popkontv, twitch, liveme, showroom, chzzk')
'tiktok, soop, pandalive, winktv, flextv, popkontv, twitch, liveme, showroom, chzzk, shopee, shp')
enable_proxy_platform_list = enable_proxy_platform.replace('，', ',').split(',') if enable_proxy_platform else None
extra_enable_proxy = read_config_value(config, '录制设置', '额外使用代理录制的平台(逗号分隔)', '')
extra_enable_proxy_platform_list = extra_enable_proxy.replace('，', ',').split(',') if extra_enable_proxy else None
@ -1693,8 +1700,6 @@ while True:
"m.6.cn",
'www.lehaitv.com',
'h.catshow168.com',
'live.shopee.',
'.shp.ee',
]
overseas_platform_host = [
'www.tiktok.com',
@ -1708,7 +1713,9 @@ while True:
'www.liveme.com',
'www.showroom-live.com',
'chzzk.naver.com',
'm.chzzk.naver.com'
'm.chzzk.naver.com',
'live.shopee.',
'.shp.ee',
]
platform_host.extend(overseas_platform_host)