2019-02-19 23:15:00 +08:00
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
2019-02-14 18:36:37 +08:00
|
|
|
|
import json
|
2019-02-19 23:15:00 +08:00
|
|
|
|
import struct
|
2018-06-03 14:06:00 +08:00
|
|
|
|
import sys
|
2019-02-19 23:15:00 +08:00
|
|
|
|
from asyncio import get_event_loop, gather, sleep, CancelledError
|
|
|
|
|
from collections import namedtuple
|
|
|
|
|
from enum import IntEnum
|
|
|
|
|
# noinspection PyProtectedMember
|
|
|
|
|
from ssl import _create_unverified_context
|
|
|
|
|
|
2018-06-03 14:06:00 +08:00
|
|
|
|
import aiohttp
|
2019-02-19 23:15:00 +08:00
|
|
|
|
import websockets
|
|
|
|
|
from websockets.exceptions import ConnectionClosed
|
2018-05-13 21:57:36 +08:00
|
|
|
|
|
|
|
|
|
|
2019-02-19 23:15:00 +08:00
|
|
|
|
class Operation(IntEnum):
|
|
|
|
|
SEND_HEARTBEAT = 2
|
|
|
|
|
POPULARITY = 3
|
|
|
|
|
COMMAND = 5
|
|
|
|
|
AUTH = 7
|
|
|
|
|
RECV_HEARTBEAT = 8
|
2018-05-13 21:57:36 +08:00
|
|
|
|
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
|
|
|
|
class BLiveClient:
|
|
|
|
|
ROOM_INIT_URL = 'https://api.live.bilibili.com/room/v1/Room/room_init'
|
|
|
|
|
WEBSOCKET_URL = 'wss://broadcastlv.chat.bilibili.com:2245/sub'
|
|
|
|
|
|
|
|
|
|
HEADER_STRUCT = struct.Struct('>I2H2I')
|
|
|
|
|
HeaderTuple = namedtuple('HeaderTuple', ('total_len', 'header_len', 'proto_ver', 'operation', 'sequence'))
|
|
|
|
|
|
|
|
|
|
def __init__(self, room_id, ssl=True, loop=None):
|
|
|
|
|
"""
|
|
|
|
|
:param room_id: URL中的房间ID
|
|
|
|
|
:param ssl: True表示用默认的SSLContext验证,False表示不验证,也可以传入SSLContext
|
|
|
|
|
:param loop: 协程事件循环
|
|
|
|
|
"""
|
|
|
|
|
self._short_id = room_id
|
|
|
|
|
self._room_id = None
|
|
|
|
|
# 未登录
|
|
|
|
|
self._uid = 0
|
|
|
|
|
|
|
|
|
|
self._ssl = ssl if ssl else _create_unverified_context()
|
|
|
|
|
self._websocket = None
|
|
|
|
|
|
|
|
|
|
self._loop = loop or get_event_loop()
|
|
|
|
|
self._future = None
|
|
|
|
|
|
|
|
|
|
def start(self):
|
|
|
|
|
"""
|
|
|
|
|
创建相关的协程,不会执行事件循环
|
|
|
|
|
:return: True表示成功创建协程,False表示之前创建的协程未结束
|
|
|
|
|
"""
|
|
|
|
|
if self._future is not None:
|
2018-05-15 19:31:58 +08:00
|
|
|
|
return False
|
2019-02-19 23:15:00 +08:00
|
|
|
|
self._future = gather(
|
|
|
|
|
self._message_loop(),
|
|
|
|
|
self._heartbeat_loop(),
|
|
|
|
|
loop=self._loop
|
|
|
|
|
)
|
|
|
|
|
self._future.add_done_callback(self.__on_done)
|
2018-05-15 19:31:58 +08:00
|
|
|
|
return True
|
|
|
|
|
|
2019-02-19 23:15:00 +08:00
|
|
|
|
def stop(self):
|
|
|
|
|
"""
|
|
|
|
|
取消相关的协程,不会停止事件循环
|
|
|
|
|
"""
|
|
|
|
|
if self._future is not None:
|
|
|
|
|
self._future.cancel()
|
|
|
|
|
|
|
|
|
|
def __on_done(self, future):
|
|
|
|
|
self._future = None
|
|
|
|
|
self._on_stop(future.exception())
|
|
|
|
|
|
|
|
|
|
async def _get_room_id(self):
|
2019-02-14 18:36:37 +08:00
|
|
|
|
try:
|
2019-02-19 23:15:00 +08:00
|
|
|
|
async with aiohttp.ClientSession(loop=self._loop) as session:
|
|
|
|
|
async with session.get(self.ROOM_INIT_URL,
|
|
|
|
|
params={'id': self._short_id},
|
|
|
|
|
ssl=self._ssl) as res:
|
|
|
|
|
if res.status == 200:
|
|
|
|
|
data = await res.json()
|
|
|
|
|
if data['code'] == 0:
|
|
|
|
|
self._room_id = data['data']['room_id']
|
|
|
|
|
else:
|
|
|
|
|
raise ConnectionAbortedError('获取房间ID失败:' + data['msg'])
|
|
|
|
|
else:
|
|
|
|
|
raise ConnectionAbortedError('获取房间ID失败:' + res.reason)
|
|
|
|
|
except Exception as e:
|
|
|
|
|
if not self._handle_error(e):
|
|
|
|
|
self._future.cancel()
|
|
|
|
|
raise
|
|
|
|
|
|
|
|
|
|
def _make_packet(self, data, operation):
|
|
|
|
|
body = json.dumps(data).encode('utf-8')
|
|
|
|
|
header = self.HEADER_STRUCT.pack(
|
|
|
|
|
self.HEADER_STRUCT.size + len(body),
|
|
|
|
|
self.HEADER_STRUCT.size,
|
|
|
|
|
1,
|
|
|
|
|
operation,
|
|
|
|
|
1
|
|
|
|
|
)
|
|
|
|
|
return header + body
|
|
|
|
|
|
|
|
|
|
async def _send_auth(self):
|
|
|
|
|
auth_params = {
|
|
|
|
|
'uid': self._uid,
|
|
|
|
|
'roomid': self._room_id,
|
|
|
|
|
'protover': 1,
|
|
|
|
|
'platform': 'web',
|
|
|
|
|
'clientver': '1.4.0'
|
|
|
|
|
}
|
|
|
|
|
await self._websocket.send(self._make_packet(auth_params, Operation.AUTH))
|
|
|
|
|
|
|
|
|
|
async def _message_loop(self):
|
|
|
|
|
# 获取房间ID
|
|
|
|
|
if self._room_id is None:
|
|
|
|
|
await self._get_room_id()
|
|
|
|
|
|
2018-05-14 01:02:53 +08:00
|
|
|
|
while True:
|
2019-02-19 23:15:00 +08:00
|
|
|
|
try:
|
|
|
|
|
# 连接
|
|
|
|
|
async with websockets.connect(self.WEBSOCKET_URL,
|
|
|
|
|
ssl=self._ssl,
|
|
|
|
|
loop=self._loop) as websocket:
|
|
|
|
|
self._websocket = websocket
|
|
|
|
|
await self._send_auth()
|
2018-05-14 01:02:53 +08:00
|
|
|
|
|
2019-02-19 23:15:00 +08:00
|
|
|
|
# 处理消息
|
|
|
|
|
async for message in websocket:
|
|
|
|
|
await self._handle_message(message)
|
2018-05-13 23:49:32 +08:00
|
|
|
|
|
2019-02-19 23:15:00 +08:00
|
|
|
|
except CancelledError:
|
|
|
|
|
break
|
|
|
|
|
except ConnectionClosed:
|
|
|
|
|
self._websocket = None
|
|
|
|
|
# 重连
|
|
|
|
|
print('掉线重连中', file=sys.stderr)
|
|
|
|
|
try:
|
|
|
|
|
await sleep(5)
|
|
|
|
|
except CancelledError:
|
2019-02-17 17:01:52 +08:00
|
|
|
|
break
|
2018-05-14 01:02:53 +08:00
|
|
|
|
continue
|
2019-02-19 23:15:00 +08:00
|
|
|
|
except Exception as e:
|
|
|
|
|
if not self._handle_error(e):
|
|
|
|
|
self._future.cancel()
|
|
|
|
|
raise
|
|
|
|
|
continue
|
|
|
|
|
finally:
|
|
|
|
|
self._websocket = None
|
|
|
|
|
|
|
|
|
|
async def _heartbeat_loop(self):
|
|
|
|
|
while True:
|
|
|
|
|
try:
|
|
|
|
|
if self._websocket is None:
|
|
|
|
|
await sleep(0.5)
|
|
|
|
|
else:
|
|
|
|
|
await self._websocket.send(self._make_packet({}, Operation.SEND_HEARTBEAT))
|
|
|
|
|
await sleep(30)
|
|
|
|
|
|
|
|
|
|
except CancelledError:
|
|
|
|
|
break
|
|
|
|
|
except ConnectionClosed:
|
|
|
|
|
# 等待重连
|
|
|
|
|
continue
|
|
|
|
|
except Exception as e:
|
|
|
|
|
if not self._handle_error(e):
|
|
|
|
|
self._future.cancel()
|
|
|
|
|
raise
|
|
|
|
|
continue
|
|
|
|
|
|
|
|
|
|
async def _handle_message(self, message):
|
|
|
|
|
offset = 0
|
|
|
|
|
while offset < len(message):
|
|
|
|
|
try:
|
|
|
|
|
header = self.HeaderTuple(*self.HEADER_STRUCT.unpack_from(message, offset))
|
|
|
|
|
except struct.error:
|
|
|
|
|
break
|
|
|
|
|
|
|
|
|
|
if header.operation == Operation.POPULARITY:
|
|
|
|
|
popularity = int.from_bytes(message[offset + self.HEADER_STRUCT.size:
|
|
|
|
|
offset + self.HEADER_STRUCT.size + 4],
|
|
|
|
|
'big')
|
|
|
|
|
await self._on_get_popularity(popularity)
|
|
|
|
|
|
|
|
|
|
elif header.operation == Operation.COMMAND:
|
|
|
|
|
body = message[offset + self.HEADER_STRUCT.size: offset + header.total_len]
|
|
|
|
|
body = json.loads(body.decode('utf-8'))
|
|
|
|
|
await self._handle_command(body)
|
|
|
|
|
|
|
|
|
|
elif header.operation == Operation.RECV_HEARTBEAT:
|
|
|
|
|
await self._websocket.send(self._make_packet({}, Operation.SEND_HEARTBEAT))
|
|
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
body = message[offset + self.HEADER_STRUCT.size: offset + header.total_len]
|
|
|
|
|
print('未知包类型:', header, body, file=sys.stderr)
|
|
|
|
|
|
|
|
|
|
offset += header.total_len
|
|
|
|
|
|
|
|
|
|
async def _handle_command(self, command):
|
|
|
|
|
if isinstance(command, list):
|
|
|
|
|
for one_command in command:
|
|
|
|
|
await self._handle_command(one_command)
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
cmd = command['cmd']
|
|
|
|
|
# print(command)
|
|
|
|
|
|
|
|
|
|
if cmd == 'DANMU_MSG': # 收到弹幕
|
|
|
|
|
await self._on_get_danmaku(command['info'][1], command['info'][2][1])
|
|
|
|
|
|
|
|
|
|
elif cmd == 'SEND_GIFT': # 送礼物
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
elif cmd == 'WELCOME': # 欢迎
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
elif cmd == 'WELCOME_GUARD': # 欢迎房管
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
elif cmd == 'SYS_MSG': # 系统消息
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
elif cmd == 'PREPARING': # 房主准备中
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
elif cmd == 'LIVE': # 直播开始
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
elif cmd == 'WISH_BOTTLE': # 许愿瓶?
|
|
|
|
|
pass
|
|
|
|
|
|
2019-02-17 17:01:52 +08:00
|
|
|
|
else:
|
2019-02-19 23:15:00 +08:00
|
|
|
|
print('未知命令:', command, file=sys.stderr)
|
|
|
|
|
|
|
|
|
|
async def _on_get_popularity(self, popularity):
|
|
|
|
|
"""
|
|
|
|
|
获取到人气值
|
|
|
|
|
:param popularity: 人气值
|
|
|
|
|
"""
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
async def _on_get_danmaku(self, content, user_name):
|
|
|
|
|
"""
|
|
|
|
|
获取到弹幕
|
|
|
|
|
:param content: 弹幕内容
|
|
|
|
|
:param user_name: 弹幕作者
|
|
|
|
|
"""
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
def _on_stop(self, exc):
|
|
|
|
|
"""
|
|
|
|
|
协程结束后被调用
|
|
|
|
|
:param exc: 如果是异常结束则为异常,否则为None
|
|
|
|
|
"""
|
|
|
|
|
pass
|
2018-06-03 14:06:00 +08:00
|
|
|
|
|
2019-02-19 23:15:00 +08:00
|
|
|
|
def _handle_error(self, exc):
|
|
|
|
|
"""
|
|
|
|
|
处理异常时被调用
|
|
|
|
|
:param exc: 异常
|
|
|
|
|
:return: True表示异常被处理,False表示异常没被处理
|
|
|
|
|
"""
|
|
|
|
|
return False
|