# 秀色直播：https://www.showself.com/

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


class ShowSelf:

    def __init__(self, rid):
        self.rid = rid
        self.headers = {
            'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) '
                          'Chrome/95.0.4638.69 Safari/537.36 '
        }
        self.s = requests.Session()

    def get_real_url(self):
        res = self.s.get('https://service.showself.com/v2/custuser/visitor', headers=self.headers).json()
        uid = res['data']['uid']
        accesstoken = sessionid = res['data']['sessionid']
        params = {
            'accessToken': accesstoken,
            'tku': uid,
            '_st1': int(time.time() * 1000)
        }
        payload = {
            'groupid': '999',
            'roomid': self.rid,
            'sessionid': sessionid,
            'sessionId': sessionid
        }
        # 合并两个字典
        data = dict(params, **payload)
        data = f'{urlencode(sorted(data.items(), key=lambda d: d[0]))}sh0wselfh5'
        _ajaxData1 = hashlib.md5(data.encode('utf-8')).hexdigest()
        payload['_ajaxData1'] = _ajaxData1
        url = f'https://service.showself.com/v2/rooms/{self.rid}/members?{urlencode(params)}'
        res = self.s.post(url, json=payload, headers=self.headers)
        if res.status_code == 200:
            res = res.json()
            statuscode = res['status']['statuscode']
            if statuscode == '0':
                if res['data']['roomInfo']['live_status'] == '1':
                    anchor, = res['data']['roomInfo']['anchor']
                    real_url = anchor['media_url']
                    return real_url
                else:
                    raise Exception('未开播')
            else:
                raise Exception('房间不存在')
        else:
            raise Exception('参数错误')


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


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