# 艾米直播：https://www.imifun.com/

import requests
import re


class IMFun:

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

    def get_real_url(self):
        with requests.Session() as s:
            res = s.get(f'https://www.imifun.com/{self.rid}').text
        roomid = re.search(r'mixPkPlayUrl ="rtmp://wsmd.happyia.com/ivp/(\d+-\d+)"', res).group(1)
        if roomid:
            status = re.search(r"isLive:(\d),", res).group(1)
            if status == '1':
                real_url = f'https://wsmd.happyia.com/ivp/{roomid}.flv'
                return real_url
            else:
                raise Exception('未开播')
        else:
            raise Exception('直播间不存在')


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


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