From 40299dade8f85d85b55346e4b06b957cd93c6fc1 Mon Sep 17 00:00:00 2001
From: wbt5 <wubangtu@qq.com>
Date: Sun, 24 Nov 2019 00:45:27 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E6=94=B6=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

---
 smzb.py | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 smzb.py

diff --git a/smzb.py b/smzb.py
new file mode 100644
index 0000000..601ed80
--- /dev/null
+++ b/smzb.py
@@ -0,0 +1,39 @@
+# 收米直播（http://smzb.tv:66/live/）所有直播间。
+
+
+import requests
+import re
+import json
+
+
+def get_all_rooms():
+    room_url = 'http://api.shoumilive.com:83/pc/anchor.json'
+    try:
+        response = requests.get(url=room_url).text
+        response_json = json.loads(re.findall(r'anchor\(([\s\S]*)\)', response)[0])
+        all_hot = response_json.get('data').get('hot')
+        all_rooms = dict()
+        for room in all_hot:
+            room_id  = room.get('room_num')
+            room_title = str(room_id) + ':' + room.get('title')
+            room_flv = get_real_url(str(room_id))
+            all_rooms[room_title] = room_flv
+    except:
+        all_rooms = '获取错误'
+    return all_rooms
+
+
+
+def get_real_url(rid):
+    room_url = 'http://api.shoumilive.com:83/pc/room/{}.json'.format(rid)
+    try:
+        response = requests.get(url=room_url).text
+        response_json = json.loads(re.findall(r'livePath\(([\s\S]*)\)', response)[0])
+        real_url = response_json.get('data').get('flv_hd')
+    except:
+        real_url = '获取错误'
+    return real_url
+
+
+if __name__ == "__main__":
+    print(get_all_rooms())