# 乐嗨直播：https://www.lehaitv.com/

from urllib.parse import urlencode
from urllib.parse import unquote
import requests
import time
import hashlib


class LeHai:

    def __init__(self, rid):
        self.rid = rid

    def get_real_url(self):
        url = f'https://service.lehaitv.com/v2/room/{self.rid}/media/advanceInfoRoom'
        params = {
            '_st1': int(time.time() * 1e3),
            'accessToken': 's7FUbTJ%2BjILrR7kicJUg8qr025ZVjd07DAnUQd8c7g%2Fo4OH9pdSX6w%3D%3D',
            'tku': 3000006,
        }
        data = urlencode(params) + '1eha12h5'
        _ajaxData1 = hashlib.md5(data.encode('utf-8')).hexdigest()
        params['_ajaxData1'] = _ajaxData1
        params['accessToken'] = unquote(params['accessToken'])
        with requests.Session() as s:
            res = s.get(url, params=params)
        if res.status_code == 200:
            res = res.json()
            statuscode = res['status']['statuscode']
            if statuscode == '0':
                if res['data']['live_status'] == 1:
                    real_url = res['data']['medial_url_app_for_h264']
                    return real_url
                else:
                    raise Exception('未开播')
            else:
                raise Exception('房间不存在 或 权限检查错误')
        else:
            raise Exception('请求错误')


def get_real_url(rid):
    try:
        lh = LeHai(rid)
        return lh.get_real_url()
    except Exception as e:
        print('Exception：', e)
        return False


if __name__ == '__main__':
    r = input('输入乐嗨直播房间号：\n')
    print(get_real_url(r))