BedwarLive/chatscript_gbk.py at master · noexcept2005/BedwarLive · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import threading
import time
import tkinter.simpledialog  # 使用Tkinter前需要先导入
from tkinter import END, messagebox

import requests

# 全局变量，用于标识线程是否退出
is_exit = True

my_room_id = 31196635		#		30972823
cd_seconds = 2.0

dlist = []
# B站获取弹幕对象
class ChatManager():
    def __init__(self, room_id):
        # 弹幕url
        self.url = 'https://api.live.bilibili.com/xlive/web-room/v1/dM/gethistory'
        # 请求头
        self.headers = {
            'Host': 'api.live.bilibili.com',
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:78.0) Gecko/20100101 Firefox/78.0',
        }
        # 定义POST传递的参数
        self.data = {
            'roomid': room_id,
            'csrf_token': '',
            'csrf': '',
            'visit_id': '',
        }
        # 日志写对象
        self.log_file_write = open('chat.log', mode='a', encoding='utf-8')
        # 读取日志
        log_file_read = open('chat.log', mode='r', encoding='utf-8')
        self.log = log_file_read.readlines()

    def get_chat(self):
        global dlist
        # 暂停0.5防止cpu占用过高
        # 获取直播间弹幕
        html = requests.post(url=self.url, headers=self.headers, data=self.data).json()
        # 解析弹幕列表
        rl = []
        for content in html['data']['room']:
            # 获取昵称
            nickname = content['nickname']
            # 获取发言
            text = content['text']
            # 获取发言时间
            timeline = content['timeline']
            dr = ChatRecord(nickname, text, timeline)

            if dr.gt() not in dlist:
                dlist.append(dr.gt())
                #print("新的弹幕:{} : {}  ({})".format(nickname, text, timeline))
                rl.append(dr)
                #print(text)
            # 记录发言
            msg = timeline + ' ' + nickname + ': ' + text
            # 清空变量缓存
            nickname = ''
            text = ''
            timeline = ''
            msg = ''
        return rl
class ChatRecord:
    def __init__(self, a, b, c):
        self.nickname = a
        self.text = b
        self.time = c
    def gt(self):
        return "[time {} nickname {} content {}]".format(self.time, self.nickname, self.text)

# 线程对象
def bilibili(room_id):
    # 创建bChatManager实例
    bChatManager = ChatManager(room_id)
    # 获取弹幕
    return bChatManager.get_chat()
if __name__ == '__main__':
    while 1:
        time.sleep(cd_seconds)
        wtd = ""
        for i in bilibili(my_room_id):
            print(i.nickname, '>> ', i.text, '        ', i.time, sep='')
            wtd += i.text + '`' + i.nickname + '\n'
        f = open("commands.txt", "a")
        f.write(wtd)
        f.close()