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

feature：1.点歌功能，2.签到功能

Browse Source
This commit is contained in:
liuwei
2025-03-05 09:01:08 +08:00
parent 5ed4c94094
commit 9937115ad3
14 changed files with 524 additions and 200 deletions
Download Patch File Download Diff File Expand all files Collapse all files

119
robot.py
View File

@@ -9,6 +9,8 @@ from threading import Thread
from datetime import datetime, timedelta
import random
import redis
from base.func_doubao import Doubao
from base.func_epic import is_friday, get_free
from base.func_zhipu import ZhiPu
@@ -29,16 +31,19 @@ from game_task.game_task_encyclopedia import game_process_message, setup_schedul
run_random_task_assignment
from group_auto.group_auto_invite import get_first_group_id, process_command
from group_auto.group_member_change import GroupMemberChange
from message_sign.main import SignInSystem
from message_storage.message_to_db import MessageStorage
from music.bot_music import BotMusic
from robot_cmd.robot_command import GroupBotManager
from job_mgmt import Job
from robot_cmd.robot_command import Feature
from robot_cmd.robot_command import PermissionStatus
import mysql.connector.pooling
__version__ = "39.2.4.0"
from message_report.process_message import process_message
from message_report.write_db import write_to_db, generate_and_send_ranking
from message_storage.message_to_db import archive_message, get_messages
from message_summary.message_summary_4o import message_summary
from sehuatang.shehuatang import pdf_file_path
from xiuren.meitu_dl import meitu_dowload_pic, meitu_dowload_pub_pic, meitu_dowload_heisi_pic
@@ -55,11 +60,27 @@ class Robot(Job):
self.config = config
self.LOG = logging.getLogger("Robot")
self.wxid = self.wcf.get_self_wxid()
self.allContacts = self.getAllContacts()
self.allContacts = self.get_all_contacts()
self.LOG.info(f"DB+REDIS 连接池开始初始化")
# db 配置加载
self.db_pool = mysql.connector.pooling.MySQLConnectionPool(self.config.mariadb)
self.LOG.info(f"DB连接池加载完成: {self.config.mariadb}")
self.redis_pool = redis.ConnectionPool(self.config.redis)
self.LOG.info(f"REDIS连接池加载完成: {self.config.redis}")
self.groups = {} # 存储按group_id分组的消息列表，每个group_id最多保留10条消息
GroupBotManager.load_local_cache()
# 消息存档模块初始化，自动完成入库动作
self.message_storage = MessageStorage(self.db_pool, self.redis_pool)
# 权限模块加载
self.gbm = GroupBotManager()
self.gmc = GroupMemberChange(wcf)
# 群成员变更模块加载
self.gmc = GroupMemberChange(wcf, self.redis_pool)
# 点歌模块加载
self.music = BotMusic(wcf, self.gbm)
# 签到模块加载
self.signin = SignInSystem(wcf, self.gbm, self.allContacts, self.db_pool, self.redis_pool)
if ChatType.is_in_chat_types(chat_type):
if chat_type == ChatType.TIGER_BOT.value and TigerBot.value_check(self.config.TIGERBOT):
self.chat = TigerBot(self.config.TIGERBOT)
@@ -132,13 +153,13 @@ class Robot(Job):
if cy.isChengyu(text):
rsp = cy.getNext(text)
if rsp:
self.sendTextMsg(rsp, msg.roomid)
self.send_text_msg(rsp, msg.roomid)
status = True
elif flag in ["?", "？"]: # 查词
if cy.isChengyu(text):
rsp = cy.getMeaning(text)
if rsp:
self.sendTextMsg(rsp, msg.roomid)
self.send_text_msg(rsp, msg.roomid)
status = True
return status
@@ -172,15 +193,15 @@ class Robot(Job):
player_id = resp["player_id"]
print(f"消息: {message}")
print(f"玩家ID: {player_id}")
self.sendTextMsg(message, msg.roomid, msg.sender)
self.send_text_msg(message, msg.roomid, msg.sender)
except Exception as e:
self.LOG.error(f"game_message_load error：{e}")
return True
if q == "#今日百度新闻":
self.newsBaiduReport((msg.roomid if msg.from_group() else msg.sender))
self.news_baidu_report((msg.roomid if msg.from_group() else msg.sender))
return True
elif q in ["nbc", "cnn", "abc", "fox", "bbc"]:
self.newsEnReport(q, (msg.roomid if msg.from_group() else msg.sender))
self.news_en_report(q, (msg.roomid if msg.from_group() else msg.sender))
return True
elif q == '#总结':
self.message_summary_robot((msg.roomid if msg.from_group() else msg.sender))
@@ -221,7 +242,7 @@ class Robot(Job):
# 如果是群消息，并且群没开启AI，则不处理该动作
if msg.from_group() and self.gbm.get_group_permission(msg.roomid,
Feature.AI_CAPABILITY) == PermissionStatus.DISABLED:
self.sendTextMsg("群AI功能未开启", msg.roomid, msg.sender)
self.send_text_msg("群AI功能未开启", msg.roomid, msg.sender)
return True
else:
if msg.type == 1: # 只处理类型为1的消息提供的问题，引用@不予以对话
@@ -230,9 +251,9 @@ class Robot(Job):
return True
if rsp:
if msg.from_group():
self.sendTextMsg(rsp, msg.roomid, msg.sender)
self.send_text_msg(rsp, msg.roomid, msg.sender)
else:
self.sendTextMsg(rsp, msg.sender)
self.send_text_msg(rsp, msg.sender)
return True
else:
@@ -259,7 +280,7 @@ class Robot(Job):
# 聊天记录入库动作：
try:
now_time = str(datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
archive_message(msg.roomid, now_time, msg.sender, msg.content, msg.type, msg.extra)
self.message_storage.archive_message(msg.roomid, now_time, msg.sender, msg.content, msg.type, msg.extra)
except Exception as e:
self.LOG.error(f"archive_message error: {e}")
@@ -268,7 +289,7 @@ class Robot(Job):
if msg.from_self():
self.revoke_receive_message(msg)
rsp = self.gbm.handle_command(msg.roomid, msg.content)
self.sendTextMsg(rsp, msg.roomid, msg.sender)
self.send_text_msg(rsp, msg.roomid, msg.sender)
return
except Exception as e:
self.LOG.error(f"revoke_receive_message error: {e}")
@@ -288,7 +309,7 @@ class Robot(Job):
player_id = resp["player_id"]
print(f"消息: {message}")
print(f"玩家ID: {player_id}")
self.sendTextMsg(message, msg.roomid, msg.sender)
self.send_text_msg(message, msg.roomid, msg.sender)
return
except Exception as e:
self.LOG.error(f"game_message_load error：{e}")
@@ -298,25 +319,25 @@ class Robot(Job):
# 判断是否没有变化
if "$NO_CHANGE$" not in result:
self.LOG.info(f"检测到群成员变化，进行相关内容输出:{result}")
self.sendTextMsg(result, msg.roomid)
self.send_text_msg(result, msg.roomid)
except Exception as e:
self.LOG.error(f"group_member_change error: {e}")
try:
self.music.get_music(message=msg)
except Exception as e:
self.LOG.error(f"get_music error: {e}")
if msg.is_at(self.wxid): # 被@
self.toAt(msg)
else: # 其他消息
self.toChengyu(msg)
return # 处理完群聊信息，后面就不需要处理了
# 非群聊信息，按消息类型进行处理
if msg.type == 37: # 好友请求
self.LOG.info(f"收到好友请求：{msg}")
self.autoAcceptFriendRequest(msg)
self.auto_accept_friend_request(msg)
elif msg.type == 10000: # 系统信息
self.sayHiToNewFriend(msg)
self.say_hi_to_new_friend(msg)
elif msg.type == 0x01: # 文本消息
# 让配置加载更灵活，自己可以更新配置。也可以利用定时任务更新。
@@ -326,15 +347,15 @@ class Robot(Job):
self.gbm.load_local_cache()
self.LOG.info("已更新")
if msg.content == "今日百度新闻":
self.newsBaiduReport()
self.news_baidu_report()
if msg.content == '聊天排行榜':
self.generateAndSendRanking()
self.generate_and_send_ranking()
if msg.content == '聊天数据入库':
self.messageCountToDB()
self.message_count_to_db()
if msg.content == 'PDF':
self.generateSehuatangPdf()
self.generate_sehuatang_pdf()
if msg.content == 'GROUP_LIST':
self.sendTextMsg(self.gbm.get_group_list(), msg.sender)
self.send_text_msg(self.gbm.get_group_list(), msg.sender)
if msg.content.startswith('#加群配置'):
# msg_content = "# 加群配置|add 原生鱼 xxx@room"
parts = msg.content.split('|')
@@ -344,8 +365,8 @@ class Robot(Job):
resp = process_command(after_pipe)
else:
resp = process_command("help")
self.sendTextMsg(resp, msg.sender)
self.sendTextMsg(f"指令：{msg.content} 已执行", msg.sender)
self.send_text_msg(resp, msg.sender)
self.send_text_msg(f"指令：{msg.content} 已执行", msg.sender)
else:
self.toChitchat(msg) # 闲聊
@@ -376,7 +397,7 @@ class Robot(Job):
self.wcf.enable_receiving_msg()
Thread(target=innerProcessMsg, name="GetMessage", args=(self.wcf,), daemon=True).start()
def sendTextMsg(self, msg: str, receiver: str, at_list: str = "") -> None:
def send_text_msg(self, msg: str, receiver: str, at_list: str = "") -> None:
""" 发送消息
:param msg: 消息字符串
:param receiver: 接收人wxid或者群id
@@ -405,7 +426,7 @@ class Robot(Job):
self.LOG.info(f"To {receiver}: {ats}\r{msg}")
self.wcf.send_text(f"{ats}\n\n{msg}", receiver, at_list)
def getAllContacts(self) -> dict:
def get_all_contacts(self) -> dict:
"""
获取联系人（包括好友、公众号、服务号、群成员……）
格式: {"wxid": "NickName"}
@@ -413,7 +434,7 @@ class Robot(Job):
contacts = self.wcf.query_sql("MicroMsg.db", "SELECT UserName, NickName FROM Contact;")
return {contact["UserName"]: contact["NickName"] for contact in contacts}
def keepRunningAndBlockProcess(self) -> None:
def keep_running_and_block_process(self) -> None:
"""
保持机器人运行，不让进程退出
"""
@@ -421,7 +442,7 @@ class Robot(Job):
self.runPendingJobs()
time.sleep(1)
def autoAcceptFriendRequest(self, msg: WxMsg) -> None:
def auto_accept_friend_request(self, msg: WxMsg) -> None:
try:
xml = ET.fromstring(msg.content)
v3 = xml.attrib["encryptusername"]
@@ -432,12 +453,12 @@ class Robot(Job):
except Exception as e:
self.LOG.error(f"同意好友出错：{e}")
def sayHiToNewFriend(self, msg: WxMsg) -> None:
def say_hi_to_new_friend(self, msg: WxMsg) -> None:
nickName = re.findall(r"你已添加了(.*)，现在可以开始聊天了。", msg.content)
if nickName:
# 添加了好友，更新好友列表
self.allContacts[msg.sender] = nickName[0]
self.sendTextMsg(f"Hi {nickName[0]}，我自动通过了你的好友请求。", msg.sender)
self.send_text_msg(f"Hi {nickName[0]}，我自动通过了你的好友请求。", msg.sender)
def send_group_txt_message(self, msg: str, feature: Feature):
try:
@@ -446,7 +467,7 @@ class Robot(Job):
return
for r in receivers:
if self.gbm.get_group_permission(r, feature) == PermissionStatus.ENABLED:
self.sendTextMsg(msg, r)
self.send_text_msg(msg, r)
except Exception as e:
self.LOG.error(f"send_group_txt_message:{feature.description} error：{e}")
@@ -497,28 +518,28 @@ class Robot(Job):
# ============================================== 业务内容==========================================================
def newsBaiduReportAuto(self) -> None:
def news_baidu_report_auto(self) -> None:
try:
news = News().get_baidu_news()
self.send_group_txt_message(news, Feature.DAILY_NEWS)
except Exception as e:
self.LOG.error(f"newsBaiduReportAuto error：{e}")
def newsBaiduReport(self, sender: str = None) -> None:
def news_baidu_report(self, sender: str = None) -> None:
try:
news = News().get_baidu_news()
self.sendTextMsg(news, sender)
self.send_text_msg(news, sender)
except Exception as e:
self.LOG.error(f"newsBaiduReport error：{e}")
def newsEnReport(self, website, sender: str = None) -> None:
def news_en_report(self, website, sender: str = None) -> None:
try:
news = News().get_eng_news(website)
self.sendTextMsg(news, sender)
self.send_text_msg(news, sender)
except Exception as e:
self.LOG.error(f"newsEnReport error：{e}")
def sendEpicFreeGames(self):
def send_epic_free_games(self):
try:
if is_friday():
games = get_free()
@@ -526,13 +547,13 @@ class Robot(Job):
except Exception as e:
self.LOG.error(f"sendEpicFreeGames error：{e}")
def messageCountToDB(self):
def message_count_to_db(self):
try:
write_to_db()
except Exception as e:
self.LOG.error(f"write_to_db error：{e}")
def generateSehuatangPdf(self):
def generate_sehuatang_pdf(self):
try:
path = pdf_file_path()
# 暂时只发4K群
@@ -540,14 +561,14 @@ class Robot(Job):
except Exception as e:
self.LOG.error(f"generateSehuatangPdf error：{e}")
def generateAndSendRanking(self):
def generate_and_send_ranking(self):
try:
receivers = self.gbm.get_group_list()
if not receivers:
return
for r in receivers:
if self.gbm.get_group_permission(r, Feature.DAILY_SUMMARY) == PermissionStatus.ENABLED:
self.sendTextMsg(generate_and_send_ranking(r, self.allContacts), r)
self.send_text_msg(generate_and_send_ranking(r, self.allContacts), r)
except Exception as e:
self.LOG.error(f"SendRanking error：{e}")
@@ -555,11 +576,11 @@ class Robot(Job):
try:
if self.gbm.get_group_permission(sender, Feature.SUMMARY_CAPABILITY) == PermissionStatus.ENABLED:
self.LOG.info(f"群： {sender} 消息总结开始执行！")
content = get_messages(sender, self.allContacts)
content = self.message_storage.get_messages(sender, self.allContacts)
summary = message_summary(content)
self.sendTextMsg(summary, sender)
self.send_text_msg(summary, sender)
else:
self.sendTextMsg("群发言总结功能未开启", sender)
self.send_text_msg("群发言总结功能未开启", sender)
except Exception as e:
self.LOG.error(f"message_summary_robot error：{e}")
@@ -574,7 +595,7 @@ class Robot(Job):
player_id = rep["player_id"]
print(f"消息: {message}")
print(f"玩家ID: {player_id}")
self.sendTextMsg(message, gid, player_id)
self.send_text_msg(message, gid, player_id)
except Exception as e:
self.LOG.error(f"message_summary_robot error：{e}")