From cc4f2b88b2c8b94cd322409d5f615f54af61a3fb Mon Sep 17 00:00:00 2001 From: ihmily <961532186@qq.com> Date: Wed, 12 Jun 2024 12:38:27 +0800 Subject: [PATCH] feat: add showroom live record --- main.py | 18 ++++++++++++++---- spider.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++-- 2 files changed, 64 insertions(+), 6 deletions(-) diff --git a/main.py b/main.py index 1e5020e..3d89982 100644 --- a/main.py +++ b/main.py @@ -4,7 +4,7 @@ Author: Hmily GitHub: https://github.com/ihmily Date: 2023-07-17 23:52:05 -Update: 2024-06-11 22:21:00 +Update: 2024-06-12 12:41:00 Copyright (c) 2023-2024 by Hmily, All Rights Reserved. Function: Record live stream video. """ @@ -56,7 +56,8 @@ from spider import ( get_twitchtv_stream_data, get_liveme_stream_url, get_huajiao_stream_url, - get_liuxing_stream_url + get_liuxing_stream_url, + get_showroom_stream_data ) from web_rid import ( @@ -70,7 +71,7 @@ from utils import ( from msg_push import dingtalk, xizhi, tg_bot version = "v3.0.5" -platforms = "\n国内站点：抖音|快手|虎牙|斗鱼|YY|B站|小红书|bigo|blued|网易CC|千度热播|猫耳FM|Look|TwitCasting|百度|微博|酷狗|LiveMe|花椒|流星" \ +platforms = "\n国内站点：抖音|快手|虎牙|斗鱼|YY|B站|小红书|bigo|blued|网易CC|千度热播|猫耳FM|Look|TwitCasting|百度|微博|酷狗|LiveMe|花椒|流星|ShowRoom" \ "\n海外站点：TikTok|AfreecaTV|PandaTV|WinkTV|FlexTV|PopkonTV|TwitchTV" recording = set() @@ -1065,6 +1066,13 @@ def start_record(url_data: tuple, count_variable: int = -1): with semaphore: port_info = get_liuxing_stream_url( url=record_url, proxy_addr=proxy_address, cookies=liuxing_cookie) + + elif record_url.find("showroom-live.com/") > -1: + platform = 'ShowRoom' + with semaphore: + json_data = get_showroom_stream_data( + url=record_url, proxy_addr=proxy_address, cookies=showroom_cookie) + port_info = get_twitchtv_stream_url(json_data, record_quality) else: logger.error(f'{record_url} 未知直播地址') return @@ -1776,6 +1784,7 @@ while True: liveme_cookie = read_config_value(config, 'Cookie', 'liveme_cookie', '') huajiao_cookie = read_config_value(config, 'Cookie', 'huajiao_cookie', '') liuxing_cookie = read_config_value(config, 'Cookie', 'liuxing_cookie', '') + showroom_cookie = read_config_value(config, 'Cookie', 'showroom_cookie', '') if len(video_save_type) > 0: if video_save_type.upper().lower() == "FLV".lower(): @@ -1869,6 +1878,7 @@ while True: 'www.huajiao.com', 'www.7u66.com', 'wap.7u66.com', + 'www.showroom-live.com', ] overseas_platform_host = [ 'www.tiktok.com', @@ -1943,4 +1953,4 @@ while True: first_run = False - time.sleep(3) \ No newline at end of file + time.sleep(3) diff --git a/spider.py b/spider.py index 3364f88..75eb0c1 100644 --- a/spider.py +++ b/spider.py @@ -4,7 +4,7 @@ Author: Hmily GitHub: https://github.com/ihmily Date: 2023-07-15 23:15:00 -Update: 2024-06-11 22:16:51 +Update: 2024-06-12 12:35:17 Copyright (c) 2023 by Hmily, All Rights Reserved. Function: Get live stream data. """ @@ -137,6 +137,10 @@ def get_play_url_list(m3u8: str, proxy: Union[str, None] = None, header: Union[d for i in resp.split('\n'): if i.startswith('https://'): play_url_list.append(i.strip()) + if not play_url_list: + for i in resp.split('\n'): + if i.strip().endswith('m3u8'): + play_url_list.append(i.strip()) bandwidth_pattern = re.compile(r'BANDWIDTH=(\d+)') bandwidth_list = bandwidth_pattern.findall(resp) url_to_bandwidth = {url: int(bandwidth) for bandwidth, url in zip(bandwidth_list, play_url_list)} @@ -1980,6 +1984,47 @@ def get_liuxing_stream_url(url: str, proxy_addr: Union[str, None] = None, cookie return result +@trace_error_decorator +def get_showroom_stream_data(url: str, proxy_addr: Union[str, None] = None, cookies: Union[str, None] = None) -> \ + Dict[str, Any]: + headers = { + 'accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7', + 'accept-language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6', + 'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36 Edg/121.0.0.0', + } + if cookies: + headers['Cookie'] = cookies + + if '/room/profile' in url: + room_id = url.split('room_id=')[-1] + else: + html_str = get_req(url, proxy_addr=proxy_addr, headers=headers) + room_id = re.search('href="/room/profile\?room_id=(.*?)"', html_str).group(1) + info_api = f'https://www.showroom-live.com/api/live/live_info?room_id={room_id}' + json_str = get_req(info_api, proxy_addr=proxy_addr, headers=headers) + json_data = json.loads(json_str) + anchor_name = json_data['room_name'] + result = {"anchor_name": anchor_name, "is_live": False} + live_status = json_data['live_status'] + if live_status == 2: + result["is_live"] = True + web_api = f'https://www.showroom-live.com/api/live/streaming_url?room_id={room_id}&abr_available=1' + json_str = get_req(web_api, proxy_addr=proxy_addr, headers=headers) + if json_str: + json_data = json.loads(json_str) + streaming_url_list = json_data['streaming_url_list'] + + for i in streaming_url_list: + if i['type'] == 'hls_all': + m3u8_url = i['url'] + result['m3u8_url'] = m3u8_url + if m3u8_url: + m3u8_url_list = get_play_url_list(m3u8_url, proxy=proxy_addr, header=headers, abroad=False) + result['play_url_list'] = [f"{m3u8_url.rsplit('/', maxsplit=1)[0]}/{i}" for i in m3u8_url_list] + break + return result + + if __name__ == '__main__': # 尽量用自己的cookie，以避免默认的不可用导致无法获取数据 # 以下示例链接不保证时效性，请自行查看链接是否能正常访问 @@ -2018,6 +2063,8 @@ if __name__ == '__main__': # room_url = 'https://www.liveme.com/zh/v/17141937295821012854/index.html' # LiveMe # room_url = 'https://www.huajiao.com/user/223184650' # 花椒直播 # room_url = 'https://www.7u66.com/100960' # 流星直播 + # room_url = 'https://www.showroom-live.com/room/profile?room_id=511033' # showroom + # room_url = 'https://www.showroom-live.com/r/TPS0728' # showroom print(get_douyin_stream_data(room_url, proxy_addr='')) # print(get_tiktok_stream_data(room_url, proxy_addr='')) @@ -2046,4 +2093,5 @@ if __name__ == '__main__': # print(get_twitchtv_stream_data(room_url, proxy_addr='')) # print(get_liveme_stream_url(room_url, proxy_addr='')) # print(get_huajiao_stream_url(room_url, proxy_addr='')) - # print(get_liuxing_stream_url(room_url, proxy_addr='')) \ No newline at end of file + # print(get_liuxing_stream_url(room_url, proxy_addr='')) + # print(get_showroom_stream_data(room_url, proxy_addr=''))