基于go-cqhttp实现QQ机器人

前言

本篇文章原文：http://www.7yue.top/rabbitbot/

本篇文章记录一下自己在编写QQ机器人的时候所遇到的一些问题和核心功能的实现。

QQ机器人RabbitBot采用python编写，由于是个人学习使用，故目前不会开源完整代码，只会放出核心代码供学习参考。

使用的go-cqhttp项目：https://github.com/Mrs4s/go-cqhttp
go-cqhttp是基于 Mirai 以及 MiraiGo 的 cqhttp golang 原生实现。
RabbitBot在读取、发送QQ信息时采用的是HTTP API和反向HTTP POST接口。

github项目地址:https://github.com/Yang9999999/Go-CQHTTP-YesBot

初期配置

go-cqhttp的安装和完整配置可自行查看文档，这里描述一下关键部分。

配置文件：

{uin: 0password: 0encrypt_password: falsepassword_encrypted: ""enable_db: falseaccess_token: ""relogin: {enabled: truerelogin_delay: 3max_relogin_times: 0}_rate_limit: {enabled: falsefrequency: 1bucket_size: 1}ignore_invalid_cqcode: falseforce_fragmented: truefix_url: falseproxy_rewrite: ""heartbeat_interval: -1http_config: {enabled: truehost: 0.0.0.0port: 5700timeout: 0post_urls: {"127.0.0.1:5710":secret}}ws_config: {enabled: falsehost: 0.0.0.0port: 6700}ws_reverse_servers: [{enabled: falsereverse_url: ws://you_websocket_universal.serverreverse_api_url: ws://you_websocket_api.serverreverse_event_url: ws://you_websocket_event.serverreverse_reconnect_interval: 3000}]post_message_format: stringuse_sso_address: falsedebug: falselog_level: ""web_ui: null
}

其中uin为机器人QQ号
password为密码
enable_db为是否使用数据库，由于个人学习使用，这里并不开启数据库功能
heartbeat_interval为心跳间隔时间，默认开启，值小于0则关闭。
http_config进行http接口设置，post_urls设置上传接收端口
ws_config和ws_reverse_servers是Websocket的正向和反向接口，由于RabbitBot采用http接口，故这里也全部设置了false

RabbitBot开启后的显示结果：
在这里插入图片描述

信息发送和接收

RabbitBot利用python的socket库来进行数据的发送和接收，socket库的使用方法请自行学习。

5700为go-cqhttp默认端口，设置http post上报器端口为5710。

常用的HTTP API：
发送私聊信息：/send_private_msg
发送群聊信息：/send_group_msg

服务端接收信息配置：

ListenSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
ListenSocket.bind(('127.0.0.1', 5710))
ListenSocket.listen(100)HttpResponseHeader = '''HTTP/1.1 200 OK
Content-Type: text/html
'''#定位有效信息
def request_to_json(msg):for i in range(len(msg)):if msg[i]=="{" and msg[-1]=="}":return json.loads(msg[i:])return None#需要循环执行，返回值为json格式
def rev_msg():# json or Noneconn, Address = ListenSocket.accept()Request = conn.recv(1024).decode(encoding='utf-8')#print(Request)rev_json=request_to_json(Request)#print(rev_json)conn.sendall((HttpResponseHeader).encode(encoding='utf-8'))conn.close()return rev_json

接收到的群聊信息格式：

{‘anonymous’: None, ‘font’: 0, ‘group_id’: ×××××, ‘message’: ‘爱你’, ‘message_id’: 1425567994, ‘message_seq’: 2170, ‘message_type’: ‘group’, ‘post_type’: ‘message’, ‘raw_message’: ‘爱你’, ‘self_id’: ×××××, ‘sender’: {‘age’: 0, ‘area’: ‘’, ‘card’: ‘’, ‘level’: ‘’, ‘nickname’: ‘七月’, ‘role’: ‘owner’, ‘sex’: ‘unknown’, ‘title’: ‘’, ‘user_id’: ×××××}, ‘sub_type’: ‘normal’, ‘time’: 1611721421, ‘user_id’: ×××××}

group_id为群号，user_id为发送者QQ号，message为接收到的信息，message_id为信息编号，随机生成，message_type为信息类型。

接收到的私聊信息格式：

{‘font’: 0, ‘message’: ‘[CQ:face,id=107]’, ‘message_id’: -730420846, ‘message_type’: ‘private’, ‘post_type’: ‘message’, ‘raw_message’: ‘[CQ:face,id=107]’, ‘self_id’: ×××××, ‘sender’: {‘age’: 0, ‘nickname’: ‘七月’, ‘sex’: ‘unknown’, ‘user_id’: ×××××}, ‘sub_type’: ‘friend’, ‘time’: 1611726456, ‘user_id’: ×××××}

user_id为通话者QQ号，message为接收到的信息，**‘[CQ:face,id=107]’**为表情数据，message_id为信息编号，随机生成，message_type为信息类型。

客户端发送信息配置：

client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)#从客户端发送给服务端
def client_to_conn():label = get_message_type()number = get_number()msg = get_raw_message()if flag == 0:msg = txt_msg(get_raw_message())if label == 'group':payload = "GET /send_group_msg?group_id=" + str(number) + "&message=" + msg + " HTTP/1.1\r\nHost: 127.0.0.1:5700\r\nConnection: close\r\n\r\n"elif label == 'private':payload = "GET /send_private_msg?user_id=" + str(number) + "&message=××××" + " HTTP/1.1\r\nHost: 127.0.0.1:5700\r\nConnection: close\r\n\r\n"client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)client.connect(('127.0.0.1',5700))client.send(payload.encode("utf-8"))client.close()

信息处理

RabbitBot的学习信息保存至一个txt文档中，个人学习问题不大，但做成项目的话不建议这么使用。

获取信息和处理信息的一些函数：

#获取信息类型 群聊/私聊 group/private
def get_message_type():return all_message['message_type']#获取群号/私聊qq号
def get_number():if get_message_type() == 'group':return all_message['group_id']elif get_message_type() == 'private':return all_message['user_id']else:print('出错啦！找不到群号/QQ号')exit()
# 获取信息发送者的QQ号
def get_user_id():return all_message['user_id']#获取发送的信息
def get_raw_message():return all_message['raw_message']#查找txt文本数据库
def txt_msg(msg):fp = open("/机器人/txt.txt", "r",encoding='utf-8')while 1:s = fp.readline()if not s:fp.close()if flag == 2:returnreturn error()s = s.strip('\n')s1 = s.split(' ')[0]s2 = s.split(' ')[1]if '[CQ:at,qq=×××××] ' + s1 == msg:fp.close()return s2

发送信息的一些函数：

#帮助界面
def help_interface():number = get_number()payload = "GET /send_group_msg?group_id=" + str(number) + "&message=学习方式：%0a私聊rabbit酱，发送学习信息。%0a学习格式：%27学习%27%20%2b%20发送信息%20%2b%20回复信息，以空格分开%0a例：学习%20我爱你%20我也爱你" + " HTTP/1.1\r\nHost: 127.0.0.1:5700\r\nConnection: close\r\n\r\n"client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)client.connect(('127.0.0.1',5700))client.send(payload.encode("utf-8"))client.close()#错误
def error():client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)client.connect(('127.0.0.1',5700))rand = random.randint(1,4)number = get_number()if rand == 1:msg = "我听不懂你在说什么哦"elif rand == 2:msg = "我好笨，听不懂呜呜呜"elif rand == 3:msg = "啊？发生了什么"elif rand == 4:msg = "干啥呢干啥呢"payload = "GET /send_group_msg?group_id=" + str(number) + "&message=" + msg + " HTTP/1.1\r\nHost: 127.0.0.1:5700\r\nConnection: close\r\n\r\n"client.send(payload.encode("utf-8"))client.close()#发送猫猫图
，图片保存在本地
def send_cat_pic():global flagflag = 1cat_list = os.listdir("/data/catpic")all_message['raw_message'] = "[CQ:image,file=file:///data/catpic/"+ random.choice(cat_list)+"]"client_to_conn()#发送setu，图片从API内获取
def send_setu_pic():apikey = '×××××××××××××××'req_url="https://api.lolicon.app/setu/"params = {"apikey":apikey}res=requests.get(req_url,params=params)setu_url=res.json()['data'][0]['url']all_message['raw_message'] ="[CQ:image,file="+setu_url+"]"client_to_conn()

私聊机器人学习数据：

#调教机器人
#这块代码也有点bug，需要后期调整。
def training_message():s = get_raw_message()if s.split(' ')[0] != '学习':returns2 = s.split(' ')[1]s3 = s.split(' ')[2]s = s2 + ' ' + s3fp = open("/机器人/txt.txt", "a",encoding='utf-8')fp.write('\n')fp.write(s)fp.close()client_to_conn()

获取数据的第一时间判断信息内容：

#首次判断信息内容
def first_judgement():if get_message_type() == 'private':training_message()if get_raw_message() == '[CQ:at,qq=×××××××××] help':help_interface()returnif get_raw_message() == '[CQ:at,qq=×××××××××] setu':send_setu()returnelif get_raw_message() == '[CQ:at,qq=×××××××××] 猫猫图':send_cat_pic()returnelif len(get_raw_message()) < 20:  #即使不@，也有15%概率回复信息rand = random.randint(1,20)if rand <= 3:global flagflag = 2all_message['raw_message'] = '[CQ:at,qq=×××××××××] ' + all_message['raw_message']client_to_conn()else:returnelif get_raw_message()[0:20] != '[CQ:at,qq=×××××××××]':returnclient_to_conn()

循环部分：

#flag为全局变量
#flag = 0 正常
#flag = 1 数据不通过数据库
#flag = 2 退出
#使用try、except语句保证程序不会因部分错误退出。
while 1:global flagflag = 0all_message = rev_msg()#print(all_message)try:first_judgement()except:continue