liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Bump to 3.7.0.30.25

Browse Source
This commit is contained in:
Changhua
2023-05-06 10:20:34 +08:00
parent 3e7b62451e
commit 57ab82c645
1 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
robot.py
View File

@@ -5,7 +5,7 @@ import re
import time import time
import xml.etree.ElementTree as ET import xml.etree.ElementTree as ET
from wcferry import Wcf from wcferry import Wcf, WxMsg
from configuration import Config from configuration import Config
from func_chatgpt import ChatGPT from func_chatgpt import ChatGPT
@@ -30,14 +30,14 @@ class Robot(Job):
if chatgpt: if chatgpt:
self.chat = ChatGPT(chatgpt.get("key"), chatgpt.get("api"), chatgpt.get("proxy")) self.chat = ChatGPT(chatgpt.get("key"), chatgpt.get("api"), chatgpt.get("proxy"))
def toAt(self, msg: Wcf.WxMsg) -> bool: def toAt(self, msg: WxMsg) -> bool:
"""处理被 @ 消息 """处理被 @ 消息
:param msg: 微信消息结构 :param msg: 微信消息结构
:return: 处理状态，`True` 成功，`False` 失败 :return: 处理状态，`True` 成功，`False` 失败
""" """
return self.toChitchat(msg) return self.toChitchat(msg)
def toChengyu(self, msg: Wcf.WxMsg) -> bool: def toChengyu(self, msg: WxMsg) -> bool:
""" """
处理成语查询/接龙消息 处理成语查询/接龙消息
:param msg: 微信消息结构 :param msg: 微信消息结构
@@ -64,7 +64,7 @@ class Robot(Job):
return status return status
def toChitchat(self, msg: Wcf.WxMsg) -> bool: def toChitchat(self, msg: WxMsg) -> bool:
"""闲聊，接入 ChatGPT """闲聊，接入 ChatGPT
""" """
if not self.chat: # 没接 ChatGPT，固定回复 if not self.chat: # 没接 ChatGPT，固定回复
@@ -84,7 +84,7 @@ class Robot(Job):
self.LOG.error(f"无法从 ChatGPT 获得答案") self.LOG.error(f"无法从 ChatGPT 获得答案")
return False return False
def processMsg(self, msg: Wcf.WxMsg) -> None: def processMsg(self, msg: WxMsg) -> None:
"""当接收到消息的时候，会调用本方法。如果不实现本方法，则打印原始消息。 """当接收到消息的时候，会调用本方法。如果不实现本方法，则打印原始消息。
此处可进行自定义发送的内容,如通过 msg.content 关键字自动获取当前天气信息，并发送到对应的群组@发送者 此处可进行自定义发送的内容,如通过 msg.content 关键字自动获取当前天气信息，并发送到对应的群组@发送者
群号：msg.roomid 微信ID：msg.sender 消息内容：msg.content 群号：msg.roomid 微信ID：msg.sender 消息内容：msg.content
@@ -116,13 +116,14 @@ class Robot(Job):
elif msg.type == 0x01: # 文本消息 elif msg.type == 0x01: # 文本消息
# 让配置加载更灵活，自己可以更新配置。也可以利用定时任务更新。 # 让配置加载更灵活，自己可以更新配置。也可以利用定时任务更新。
if msg.from_self() and msg.content == "^更新$": if msg.from_self():
self.config.reload() if msg.content == "^更新$":
self.LOG.info("已更新") self.config.reload()
self.LOG.info("已更新")
else: else:
self.toChitchat(msg) # 闲聊 self.toChitchat(msg) # 闲聊
def onMsg(self, msg: Wcf.WxMsg) -> int: def onMsg(self, msg: WxMsg) -> int:
try: try:
self.LOG.info(msg) # 打印信息 self.LOG.info(msg) # 打印信息
self.processMsg(msg) self.processMsg(msg)
@@ -172,7 +173,7 @@ class Robot(Job):
self.runPendingJobs() self.runPendingJobs()
time.sleep(1) time.sleep(1)
def autoAcceptFriendRequest(self, msg: Wcf.WxMsg) -> None: def autoAcceptFriendRequest(self, msg: WxMsg) -> None:
try: try:
xml = ET.fromstring(msg.content) xml = ET.fromstring(msg.content)
v3 = xml.attrib["encryptusername"] v3 = xml.attrib["encryptusername"]
@@ -182,7 +183,7 @@ class Robot(Job):
except Exception as e: except Exception as e:
self.LOG.error(f"同意好友出错：{e}") self.LOG.error(f"同意好友出错：{e}")
def sayHiToNewFriend(self, msg: Wcf.WxMsg) -> None: def sayHiToNewFriend(self, msg: WxMsg) -> None:
nickName = re.findall(r"你已添加了(.*)，现在可以开始聊天了。", msg.content) nickName = re.findall(r"你已添加了(.*)，现在可以开始聊天了。", msg.content)
if nickName: if nickName:
# 添加了好友，更新好友列表 # 添加了好友，更新好友列表