diff --git a/main.py b/main.py index 7228552..afff237 100644 --- a/main.py +++ b/main.py @@ -10,19 +10,21 @@ from robot import Robot def weather_report(robot: Robot) -> None: """模拟发送天气预报 """ + # 获取接收人 receivers = ["filehelper"] # 获取天气,需要自己实现,可以参考 https://gitee.com/lch0821/WeatherScrapy 获取天气。 report = "这就是获取到的天气情况了" - + for r in receivers: robot.sendTextMsg(report, r) + #robot.sendTextMsg(report, r, "nofity@all") # 发送消息并@所有人 def main(): wcf = Wcf() - + def handler(sig, frame): wcf.cleanup() # 退出前清理环境 exit(0) @@ -30,16 +32,21 @@ def main(): signal.signal(signal.SIGINT, handler) robot = Robot(wcf) - robot.LOG.info("机器人已启动") - + robot.LOG.info("正在启动机器人···") + # 机器人启动发送测试消息 + robot.sendTextMsg("机器人启动成功!", "filehelper") + # 接收消息 robot.enableRecvMsg() + # 每天7点发送天气预报 - robot.onEveryTime("07:00", weather_report, robot=robot) + robot.onEveryTime("22:30", weather_report, robot=robot) # 让机器人一直跑 robot.keepRunningAndBlockProcess() + + if __name__ == "__main__": diff --git a/robot.py b/robot.py index d3e28e7..096f7aa 100644 --- a/robot.py +++ b/robot.py @@ -12,6 +12,7 @@ from func_chengyu import cy from job_mgmt import Job + class Robot(Job): """个性化自己的机器人 """ @@ -21,8 +22,9 @@ 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.getAllContacts() + + def toAt(self, msg: Wcf.WxMsg) -> bool: """ 处理被 @ 消息,现在只固定回复: "你@我干嘛?" @@ -70,6 +72,15 @@ class Robot(Job): def processMsg(self, msg: Wcf.WxMsg) -> None: """当接收到消息的时候,会调用本方法。如果不实现本方法,则打印原始消息。 """ + + """ 此处可进行自定义发送的内容,如通过关键字自动获取当前天气信息,并发送到对应的群组@发送者 + 群号:msg.roomid 微信ID:msg.sender 消息内容:msg.content + content = "天气查询" + receivers = msg.roomid + self.sendTextMsg(content, receivers, msg.sender) + """ + + # 群聊消息 if msg.from_group(): # 如果在群里被 @,回复发信人:“收到你的消息了!” 并 @他 @@ -101,6 +112,7 @@ class Robot(Job): def onMsg(self, msg: Wcf.WxMsg) -> int: self.LOG.info(msg) # 打印信息 + try: self.processMsg(msg) except Exception as e: @@ -117,7 +129,14 @@ class Robot(Job): if at_list: wxids = at_list.split(",") for wxid in wxids: - # 这里偷个懒,直接 @昵称。有必要的话可以通过 MicroMsg.db 里的 ChatRoom 表,解析群昵称 + # 这里偷个懒,直接 @昵称 + + """ 若出现直接 @wxid,则要通过 MicroMsg.db 里的 ChatRoom 表解析群昵称,取消此处注释,并注释141行 + userInfo = self.getAllContacts() + everyUser = {'notify@all': '所有人'} + userInfo.update(everyUser) # 追加dict @所有人所需字段 + ats = f" @{userInfo[wxid]}" + """ ats = f" @{self.allContacts.get(wxid, '')}" self.LOG.info(f"To {receiver}: {msg}{ats}")