基于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: 0
    password: 0
    encrypt_password: false
    password_encrypted: ""
    enable_db: false
    access_token: ""
    relogin: { 
   
        enabled: true
        relogin_delay: 3
        max_relogin_times: 0
    }
    _rate_limit: { 
   
        enabled: false
        frequency: 1
        bucket_size: 1
    }
    ignore_invalid_cqcode: false
    force_fragmented: true
    fix_url: false
    proxy_rewrite: ""
    heartbeat_interval: -1
    http_config: { 
   
        enabled: true
        host: 0.0.0.0
        port: 5700
        timeout: 0
        post_urls: 
        { 
   
            "127.0.0.1:5710":secret
        }
    }
    ws_config: { 
   
        enabled: false
        host: 0.0.0.0
        port: 6700
    }
    ws_reverse_servers: [
        { 
   
            enabled: false
            reverse_url: ws://you_websocket_universal.server
            reverse_api_url: ws://you_websocket_api.server
            reverse_event_url: ws://you_websocket_event.server
            reverse_reconnect_interval: 3000
        }
    ]
    post_message_format: string
    use_sso_address: false
    debug: false
    log_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 None
	conn, 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:
				return
			return 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 flag
	flag = 1
	cat_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] != '学习':
		return
	s2 = s.split(' ')[1]
	s3 = s.split(' ')[2]
	s = s2 + ' ' + s3
	fp = 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()
		return
	if get_raw_message() == '[CQ:at,qq=×××××××××] setu':
		send_setu()
		return
	elif get_raw_message() == '[CQ:at,qq=×××××××××] 猫猫图':
		send_cat_pic()
		return
	elif len(get_raw_message()) < 20:  #即使不@，也有15%概率回复信息
		rand = random.randint(1,20)
		if rand <= 3:
			global flag
			flag = 2
			all_message['raw_message'] = '[CQ:at,qq=×××××××××] ' + all_message['raw_message']
			client_to_conn()
		else:
			return
	elif get_raw_message()[0:20] != '[CQ:at,qq=×××××××××]':
		return
	client_to_conn()

循环部分：

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