新增战旗TV直播

huya.py

@@ -16,7 +16,7 @@ def get_real_url(rid):
     result = re.findall(pattern, response.text, re.I)
     if result:
         real_url = result[0]
-        real_url = re.sub(r'_\d{3,4}.m3u8', '.flv', result[0])
+        real_url = re.sub(r'_1200[\s\S]*.m3u8', '.m3u8', result[0])
     else:
         real_url = '未开播或直播间不存在'
     return real_url

zhanqi.py

@@ -0,0 +1,24 @@
+# 获取战旗直播（战旗TV）的真实流媒体地址。
+# 默认画质为超清
+
+
+import requests
+
+
+def get_real_url(rid):
+    room_url = 'https://m.zhanqi.tv/api/static/v2.1/room/domain/' + str(rid) + '.json'
+    try:
+        response = requests.get(url=room_url).json()
+        videoId = response.get('data').get('videoId')
+        if videoId:
+            real_url = 'https://dlhdl-cdn.zhanqi.tv/zqlive/' + str(videoId) + '.flv'
+        else:
+            real_url = '未开播'
+    except:
+        real_url = '直播间不存在'
+    return real_url
+
+
+rid = input('请输入战旗直播房间号：\n')
+real_url = get_real_url(rid)
+print('该直播间源地址为：\n' + real_url)