1
0
mirror of https://github.com/wbt5/real-url.git synced 2025-06-16 15:59:57 +08:00
zhibo-url/imifun.py
wbt5 c0018d0fb1
🐛 Fix 艾米直播
-更新正则以正确获取room_id
-优化代码
2021-11-21 01:56:50 +08:00

39 lines
991 B
Python
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

# 艾米直播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))