update robot add some methods

README.md

@@ -18,6 +18,13 @@ pip install -r requirements.txt
 
 
 ## 更新
+- **2023.3.18**
+  - 更新`robot.py` 
+    - 新增  自动添加好友并发送自定义消息
+    - 新增  微信聊天的方式更新配置文件
+    -  新增 用chatgpt与好友聊天
+    -  新增  用chatgpt与群友聊天
+  - 更新 wcfree.py 用法
 - **2023.3.18** 
   - 更新`gpt.py` 将一些参数丢到了json文件中，方便修改
   - 新建了`job_all.py` 创建了定时类型的任务

main.py

@@ -42,16 +42,16 @@ def clean_all():
     robot.LOG.info("----正在启动机器人----")
 
     # 机器人启动后发送测试消息
-    robot.sendTextMsg("机器人启动成功！", "filehelper")
+    robot.send_text_msg("机器人启动成功！", "filehelper")
 
     # 接收消息
-    robot.action_to_Msg()
+    robot.enable_receive_message_service()
 
     # 每天7点定时发送天气预报
     robot.on_every_time('07:00', weather_report, robot=robot)
 
     # 让机器人一直跑
-    robot.keepRunning_and_block_the_process()
+    robot.keep_running_and_block_the_process()
 
 
 if __name__ == "__main__":

robot.py

@@ -28,8 +28,153 @@ def __init__(self, wcf: Wcf) -> None:
         self.config = Config()
         self.LOG = logging.getLogger("Robot")
         self.wxid = self.wcf.get_self_wxid()
-        self.allPeople = self.getallPeople()
+        self.allPeople = self.getall_people()
         self.chat = GptThread(self.config.CHAT_KEY)
 
-    def getallPeople(self):
-        pass
+    def enable_receive_message_service(self) -> None:
+        """
+        启动接收消息的服务
+        """
+        self.wcf.enable_recv_msg(self.action_msg)
+
+    def action_msg(self, msg: Wcf.WxMsg):
+        """
+        接收到消息的处理
+        :param msg: 微信消息体
+        """
+        try:
+            self.LOG.info(msg)
+            self.various_methods(msg)
+        except Exception as e:
+            self.LOG.error(f'出现错误，接收消息出粗：{e}')
+
+    def various_methods(self, msg: Wcf.WxMsg) -> None:
+        """
+        接收到消息的时候会调用这个方法，不调用，打印原始消息
+        群号: msg.roomid 微信ID: msg.sender 消息内容: msg.content
+        :param msg: 微信消息体 包含以上信息
+        """
+
+        if msg.from_group():  # 群聊消息
+            self.from_group_msg(msg)
+        else:
+            if msg.type == 37:  # 好友请求
+                self.from_friend_request(msg)
+            elif msg.type == 10000:  # 系统消息
+                self.from_system_info(msg)
+            elif msg.type == 0x01:  # 文本消息
+                self.from_text_msg(msg)
+
+    def from_group_msg(self, msg: Wcf.WxMsg) -> None:
+        """
+        处理群聊消息
+        :param msg: 微信消息体
+        """
+        if msg.roomid not in self.config.GROUPS:  # 不在允许的配置里面
+            self.LOG.info(f'收到群聊消息，但是{msg.roomid}没有在{self.config.GROUPS}中配置，不处理\n')
+            return
+        elif msg.is_at(self.wxid):  # 被@
+            self.at_me(msg)
+        else:
+            return
+
+    def from_friend_request(self, msg: Wcf.WxMsg) -> None:
+        """
+        处理好友请求
+        :param msg: 微信消息体
+        """
+        self.auto_accept_friend_request(msg)
+
+    def from_system_info(self, msg: Wcf.WxMsg) -> None:
+        """
+        处理系统消息
+        :param msg: 微信消息体
+        """
+        self.say_hi_to_new_friend(msg)
+
+    def from_text_msg(self, msg: Wcf.WxMsg) -> None:
+        """
+        处理文本消息
+        :param msg:  微信消息体
+        """
+        if msg.from_self() and msg.content == '更新':
+            self.config.reload()
+            self.LOG.info('更新配置成功')
+        else:
+            self.get_chat_gpt(msg)
+
+    def at_me(self, msg: Wcf.WxMsg) -> bool:
+        """
+        处理被@的消息
+        :param msg: 微信消息结构体
+        :return: 处理状态， True为处理成功， False为处理失败
+        """
+        return self.get_chat_gpt(msg)
+
+    def get_chat_gpt(self, msg: Wcf.WxMsg) -> bool:
+        """
+        获取聊天结构体中的gpt模型
+        :param msg: 微信消息结构体
+        :return: 处理状态， True为处理成功， False为处理失败
+        """
+        question = re.sub(r"@.*?[\u2005|\s]", "", msg.content)
+
+    def send_text_msg(self, msg: str, receiver: str, at_lists: str = '') -> None:
+        """
+        发送文本消息
+        :param msg: 消息内容
+        :param receiver: 接收人的wxid或者群roomid
+        :param at_lists: 要@的人的wxid列表，@所有人为：nofity@all
+        """
+        # msg 中需要有 @ 名单中一样数量的 @
+        ats = ""
+        if at_lists:
+            wxids = at_lists.split(",")
+            for wxid in wxids:
+                ats = f" @{self.allPeople.get(wxid, '')}"
+
+        self.LOG.info(f"To {receiver}: {msg}{ats}")
+        self.wcf.send_text(f"{msg}{ats}", receiver, at_lists)
+
+    def getall_people(self) -> dict:
+        """
+        获取联系人（包括好友、公众号、服务号、群成员……）
+        格式: {"wxid": "NickName"}
+        """
+        contacts = self.wcf.query_sql(
+            "MicroMsg.db", "SELECT UserName, NickName FROM Contact;")
+        return {contact["UserName"]: contact["NickName"]
+                for contact in contacts}
+
+    def auto_accept_friend_request(self, msg: Wcf.WxMsg):
+        """
+        自动接受好友请求
+        :param msg: 微信消息结构体
+        """
+        try:
+            xml = ET.fromstring(msg.content)
+            v3 = xml.attrib["encryptusername"]
+            v4 = xml.attrib["ticket"]
+            self.wcf.accept_new_friend(v3, v4)
+
+        except Exception as e:
+            self.LOG.error(f"同意好友出错：{e}")
+
+    def say_hi_to_new_friend(self, msg):
+        """
+        接收好友请求后自动发送消息
+        :param msg:
+        """
+        nick_name = re.findall(r"你已添加了(.*)，现在可以开始聊天了。", msg.content)
+        if nick_name:
+            # 添加了好友，更新好友列表
+            self.allPeople = nick_name[0]
+            self.send_text_msg(f"Hi {nick_name[0]}，我自动通过了你的好友请求。", msg.sender)
+
+    def keep_running_and_block_the_process(self):
+        """
+        保持机器人运行，不让进程退出
+        """
+        while True:
+            self.run_pending_jobs()
+            time.sleep(1)

wcfree.md

@@ -1,4 +1,25 @@
 # 使用方法
+
+
+
+1. `is_login` 检查登录状态
+2. `get_self_wxid` 获取登陆账号的wxid
+3. `get_msg_types` 获取消息类型
+4. `get_contacts` 获取所有联系人
+5. `get_friends` 获取所有好友
+6. `get_dbs` 获取数据库
+7. `get_tables` 获取数据库中的表
+8. `send_text` 发送文本消息，可以@
+9. `send_image` 发送图片消息
+10. `send_xml` 发送xml
+11. `send_file` 发送文件消息
+12. `enable_receiving_msg ` 允许接收消息
+13. `disable_recv_msg` 停止接收消息
+14. `query_sql` 执行sql查询
+15. `accept_new_friend` 接收好友申请
+16. `add_chatroom_members` 添加群成员
+
+---
 1. **enable_recv_msg()**
    - 首先判断是否已经在接收消息，如果是，则直接返回`True`。
    - 如果回调函数为空，则返回`False`。
@@ -63,4 +84,5 @@
                     self.LOG.error("退出失败！")
                     return
             self._is_running = False
-     ```
+     ```
+