wss_url = 'wss://broadcastlv.chat.bilibili.com/sub'
heartbeat = b'\x00\x00\x00\x1f\x00\x10\x00\x01\x00\x00\x00\x02\x00\x00\x00\x01\x5b\x6f\x62\x6a\x65\x63\x74\x20' \
                b'\x4f\x62\x6a\x65\x63\x74\x5d '
  heartbeatInterval = 60

@staticmethod
async def get_ws_info(url):
    url = 'https://api.live.bilibili.com/room/v1/Room/room_init?id=' + url.split('/')[-1]
    reg_datas = []
    async with aiohttp.ClientSession() as session:
        async with session.get(url) as resp:
            room_json = json.loads(await resp.text())
            room_id = room_json['data']['room_id']
            data = json.dumps({
                'roomid': room_id,
                'uid': int(1e14 + 2e14 * random.random()),
                'protover': 1
            }, separators=(',', ':')).encode('ascii')
            data = (pack('>i', len(data) + 16) + b'\x00\x10\x00\x01' +
                    pack('>i', 7) + pack('>i', 1) + data)
            reg_datas.append(data)

    return Bilibili.wss_url, reg_datas

它連上了Bilibili的直播彈幕WSS地址，也就是WebSocket地址，然后偽裝成客戶端，接受彈幕推送。

OK，做完了第一步，下一步就是用消息隊列將彈幕發(fā)送出來。開啟單獨的消費者接收彈幕。

為了實現(xiàn)上盡量簡單，就不上那些專業(yè)的消息隊列了，這里用了redis的list作為隊列，將彈幕內(nèi)容放進去。

發(fā)送者核心代碼如下：

# 鏈接Redis
def init_redis():
    r = redis.Redis(host='localhost', port=6379, decode_responses=True)
    return r

# 消息發(fā)送者
async def printer(q, redis):
    while True:
        m = await q.get()
        if m['msg_type'] == 'danmaku':
            print(f'{m["name"]}：{m["content"]}')
            list_str = list(m["content"])
            print("彈幕拆分:", list_str)
            for char in list_str:
                if char.lower() in key_list:
                    print('推送隊列：', char.lower())
                    redis.rpush(list_name, char.lower())

完成了彈幕內(nèi)容的發(fā)送后，需要寫一個消費者，消費這些彈幕，把里面的指令都提取出來。

并且，在消費者收到彈幕后，如何消費呢？我們需要一個能夠用代碼指令控制電腦的辦法。

咱繼續(xù)本著不造輪子的原則，找到了一個Python的自動化控制庫PyAutoGUI

PyAutoGUI is a cross-platform GUI automation Python module for human beings. Used to programmatically control the mouse & keyboard.

安裝上這個庫，在代碼中引入，便可以通過他的API控制電腦鼠標和鍵盤執(zhí)行對應的操作。簡直是完美??！

消費者（控制電腦）核心Python代碼如下：

# 鏈接Redis
def init_redis():
    r = redis.Redis(host='localhost', port=6379, decode_responses=True)
    return r

# 消費者
def control(key_name):
    print("key_name =", key_name)
    if key_name == None:
        print("本次無指令發(fā)出")
        return
    key_name = key_name.lower()
    # 控制電腦指令
    if key_name in key_list:
        print("發(fā)出指令", key_name)
        pyautogui.keyDown(key_name)
        time.sleep(press_sec)
        pyautogui.keyUp(key_name)
        print("結束指令", key_name)

if __name__ == '__main__':
    r = init_redis()
    print("開始監(jiān)聽彈幕消息, loop_sec =", loop_sec)
    while True:
        key_name = r.lpop(list_name)
        control(key_name)
        time.sleep(loop_sec)

ok，大功告成，我們打開彈幕發(fā)送隊列和消費者，這個不斷循環(huán)消費的隊列就開始運行了。一旦彈幕中有wsad這種控制游戲常用的按鍵，電腦就會自己給自己發(fā)出指令。

初版運行中的問題

我興沖沖的打開自己的B站直播間，開始調(diào)試，結果發(fā)現(xiàn)我還是太天真了。這個初版代碼暴露了非常多的問題。我們一個個來說下是什么問題，我是如何解決的。

指令不人性化

水友們其實很喜歡發(fā)送類似www dddd這類重復單詞（疊詞），但初版的實現(xiàn)只支持單個字幕，水友們發(fā)現(xiàn)不得勁，沒有作用后，就從直播間走了。

這點很容易解決，把彈幕內(nèi)容拆分成每個單詞，然后再推送給隊列。

解決方法：拆解彈幕，把DDD，拆成D,D,D，發(fā)送個消費者。

危險指令

首先是玩家的指令超出了應該有的范圍。

在我把賽博朋克游戲打開，讓彈幕觀眾控制游戲里的開車時，有個神秘觀眾進了直播間，默默發(fā)了個“F”，然后。。。

然后游戲里的V（主角名）就從車里下來了，淦，我是讓你們開車的，不是讓你們下來和警察斗毆的。。。

解決方法：添加彈幕過濾器。

# 將彈幕進行拆分，只發(fā)送指定的指令給消費者
key_list = ('w', 's', 'a', 'd', 'j', 'k', 'u', 'i', 'z', 'x', 'f', 'enter', 'shift', 'backspace')
list_str = list(m["content"])
            print("彈幕拆分:", list_str)
            for char in list_str:
                if char.lower() in key_list:
                    print('推送隊列：', char.lower())
                    redis.rpush(list_name, char.lower())

上面兩個問題解決后，發(fā)送者就像下面這樣運行了：

彈幕指令堆積

這是個很大的問題，如果處理所有水友發(fā)送的全部彈幕指令，一定會存在消費不過來的問題。

解決方法：需要固定時間處理彈幕，其他拋棄。

if __name__ == '__main__':
    r = init_redis()
    print("開始監(jiān)聽彈幕消息, loop_sec =", loop_sec)
    while True:
        key_name = r.lpop(list_name)
        # 每次只取出一個指令，然后把list清空，也就是這個時間窗口內(nèi)其他彈幕都扔掉！
        r.delete(list_name)
        control(key_name)
        time.sleep(loop_sec)

彈幕從發(fā)出到觀眾看到結果有延遲

在最開始的視頻里，你們也能感受到了，從觀眾的指令發(fā)出，到最終被觀眾看到，大概要經(jīng)歷5秒的延遲。其中，起碼有3秒，都是網(wǎng)絡直播流的延遲，這一點，很難去優(yōu)化。