17
main.py
17
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__":
|
||||
|
||||
25
robot.py
25
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}")
|
||||
|
||||
Reference in New Issue
Block a user