finbot.py

# -*- coding: utf-8 -*-

import logging
import time
from queue import Empty
from threading import Thread

import schedule

from wcferry import Wcf, WxMsg

from config import WX_ID


class FinBot:
    """财务机器人类"""
    _instance = None
    _initialized = False

    def __new__(cls, wcf=None):
        if cls._instance is None:
            cls._instance = super().__new__(cls)
        return cls._instance

    def __init__(self) -> None:
        if not self._initialized:
            print('FinBot.__init__()')
            self.wcf = Wcf(debug=True)
            self.LOG = logging.getLogger("Robot")
            self.wxid = self.wcf.get_self_wxid() if self.wcf else None
            self._initialized = True

    def processMsg(self, msg: WxMsg) -> None:
        from message_parser import MessageParser

        msg_parser = MessageParser()
        """当接收到消息的时候，会调用本方法。如果不实现本方法，则打印原始消息。
        此处可进行自定义发送的内容,如通过 msg.content 关键字自动获取当前天气信息，并发送到对应的群组@发送者
        群号：msg.roomid  微信ID：msg.sender  消息内容：msg.content
        content = "xx天气信息为："
        receivers = msg.roomid
        self.sendTextMsg(content, receivers, msg.sender)
        """
        if msg.type == 1 and msg.sender == WX_ID:
            msg_parser.parse_msg_self(msg.content)
        if msg.type == 49:
            msg_parser.parse_msg_xml(msg.content)

    def enableReceivingMsg(self) -> None:
        def innerProcessMsg(wcf: Wcf):
            while wcf.is_receiving_msg():
                try:
                    msg = wcf.get_msg()
                    # self.LOG.info(msg)
                    self.processMsg(msg)
                except Empty:
                    continue  # Empty message
                except Exception as e:
                    self.LOG.error(f"Receiving message error: {e}")

        # print(f'消息类型: ${self.wcf.get_msg_types()}')
        self.wcf.enable_receiving_msg()


        Thread(target=innerProcessMsg, name="GetMessage", args=(self.wcf,), daemon=True).start()

    def getAllContacts(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 keepRunningAndBlockProcess(self) -> None:
        """
        保持机器人运行，不让进程退出
        """
        while True:
            schedule.run_pending()
            time.sleep(1)

    def send_text_msg(self, content: str, retry: int=3) -> None:
        if self.wcf:
            cnt = 0
            while cnt < retry:
                result = self.wcf.send_text(content, WX_ID)
                if result == 0:
                    return
                time.sleep(1)
                cnt += 1
        else:
            print('wcf没有初始化')

    def clean(self):
        self.wcf.cleanup()