From f8d31de046c29be21e514491d179fb15be87081a Mon Sep 17 00:00:00 2001
From: wbt5 <wubangtu@qq.com>
Date: Fri, 31 Jul 2020 19:40:29 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E8=89=BE=E7=B1=B3=E7=9B=B4?=
 =?UTF-8?q?=E6=92=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 imifun.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 imifun.py

diff --git a/imifun.py b/imifun.py
new file mode 100644
index 0000000..3135676
--- /dev/null
+++ b/imifun.py
@@ -0,0 +1,23 @@
+# 艾米直播：https://www.imifun.com/
+import requests
+import re
+
+
+def imifun(rid):
+    with requests.Session() as s:
+        res = s.get('https://www.imifun.com/' + str(rid)).text
+    roomid = re.search(r"roomId:\s'([\w-]+)'", res)
+    if roomid:
+        status = re.search(r"isLive:(\d),", res).group(1)
+        if status == '1':
+            real_url = 'https://wsmd.happyia.com/ivp/{}.flv'.format(roomid.group(1))
+            return real_url
+        else:
+            raise Exception('未开播')
+    else:
+        raise Exception('直播间不存在')
+
+
+if __name__ == '__main__':
+    r = input('输入艾米直播房间号：\n')
+    print(imifun(r))