# 获取抖音直播的真实流媒体地址，默认最高画质。
# 如果知道该直播间如“6779127643792280332”形式的room_id，则直接传入room_id。
# 如果不知道room_id，可以使用手机上打开直播间后，选择“分享--复制链接”，传入如“https://v.douyin.com/qyRqMp/”形式的分享链接。


import requests
import re


def get_real_url(rid):
    try:
        if 'v.douyin.com' in rid:
            room_id = re.findall(r'(\d{19})', requests.get(url=rid).url)[0]
        else:
            room_id = rid
        room_url = 'https://webcast-hl.amemv.com/webcast/room/reflow/info/?room_id={}&live_id=1'.format(room_id)
        response = requests.get(url=room_url).json()
        hls_pull_url = response.get('data').get('room').get('stream_url').get('hls_pull_url')
        rtmp_pull_url = response.get('data').get('room').get('stream_url').get('rtmp_pull_url')
        real_url = [rtmp_pull_url, hls_pull_url]
    except:
        real_url = '直播间不存在或未开播或参数错误'
    return real_url


rid = input('请输入抖音直播间room_id或分享链接：\n')
real_url = get_real_url(rid)
print('该直播间源地址为：')
print(real_url)