From fe41cd305d696ca948d9b15d0b5cbfdab28c416c Mon Sep 17 00:00:00 2001
From: wbt5 <wubangtu@qq.com>
Date: Sat, 18 Jul 2020 15:03:21 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9Epps=E5=A5=87=E7=A7=80?=
 =?UTF-8?q?=E7=9B=B4=E6=92=AD=E5=BC=B9=E5=B9=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 danmu/danmaku/pps.py | 73 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 danmu/danmaku/pps.py

diff --git a/danmu/danmaku/pps.py b/danmu/danmaku/pps.py
new file mode 100644
index 0000000..5ef44ad
--- /dev/null
+++ b/danmu/danmaku/pps.py
@@ -0,0 +1,73 @@
+import hashlib
+import urllib.parse
+import json
+
+
+class QiXiu:
+    heartbeat = None
+
+    @staticmethod
+    async def get_ws_info(url):
+        rid = url.split('/')[-1]
+        s = bytes([57, 77, 83, 73, 53, 86, 85, 71, 50, 81, 74, 80, 66, 52, 78, 54, 68, 48, 81,
+                   83, 89, 87, 69, 72, 67, 90, 83, 75, 84, 49, 77, 50, 84, 65, 75, 88]).decode('utf-8')
+        # ua = 'User-Agent'
+        # ak = deviceid = md5(str(int(time.time() * 1e3)) + ua + '0000')
+        ak = deviceid = '118d2ae703e62992263e6741afbb5627'
+        e = {
+            'ag': 1,
+            'ak': ak,
+            'at': 3,
+            'd': deviceid,
+            'n': 1,
+            'p': 1,
+            'r': rid,
+            'v': '1.01.0801'
+        }
+        i = ''
+        for k, v in e.items():
+            i += '{}={}|'.format(k, str(v))
+        e['sg'] = hashlib.md5((i + s).encode('utf-8')).hexdigest()
+        ws_url = 'ws://qx-ws.iqiyi.com/ws?' + urllib.parse.urlencode(e)
+        return ws_url, None
+
+    @staticmethod
+    def decode_msg(data):
+        message = json.loads(data)
+        msgs = []
+        msg = {'name': '', 'content': '', 'msg_type': 'other'}
+        for ms in message:
+            m = ms['ct']
+            type_ = ms['t']
+            # 200001：进场消息
+            # 300001：聊天信息
+            # 102001：礼物
+            # 1100002：礼物
+            # 400001：人气值
+            # 5000010：升级
+            # 700095：live_score
+            # 700091：排名
+            # 其他：系统消息
+            if type_ == 300001:
+                msg['name'] = m['op_userInfo']['nick_name']
+                msg['content'] = m['msg']
+                msg['msg_type'] = 'danmaku'
+            elif type_ == 102001:
+                msg['name'] = m['op_userInfo']['nick_name']
+                num = m['op_info']['num']
+                gift = m['op_info']['name']
+                msg['content'] = '送出{}个{}'.format(num, gift)
+                msg['msg_type'] = 'danmaku'
+            elif type_ in [200001, 1100002, 110001, 3019, 3022, 3002, 3024]:
+                msg['name'] = 'SYS'
+                info = m['op_info'].get('public_chat_msg', 0)
+                if not info:
+                    info = m['op_info']['roll_chat_msg']
+                content = ''
+                items = info['items']
+                for item in items:
+                    content += item.get('content', '')
+                msg['content'] = content
+                msg['msg_type'] = 'danmaku'
+            msgs.append(msg.copy())
+            return msgs