python-wechaty

bot.py

"""doc"""
# pylint: disable=R0801
import asyncio
import logging
from typing import Optional, Union

from wechaty_puppet import FileBox, ScanStatus  # type: ignore

import base64

from wechaty import Wechaty, Contact
from wechaty.user import Message, Room

logging.basicConfig(level=logging.INFO)
log = logging.getLogger(__name__)


class MyBot(Wechaty):
    """
    listen wechaty event with inherited functions, which is more friendly for
    oop developer
    """
    def __init__(self):
        super().__init__()

    async def on_message(self, msg: Message):
        """
        listen for message event
        """
        from_contact = msg.talker()
        text = msg.text()
        room = msg.room()
        if text == '#ding':
            conversation: Union[
                Room, Contact] = from_contact if room is None else room
            await conversation.ready()
            await conversation.say('dong')

            with open('./avatar.jpg', 'rb') as f:
                content = base64.b64encode(f.read())
                file_box = FileBox.from_base64(name='avatar.jpg', base64=content)
                await conversation.say(file_box)


bot: Optional[MyBot] = None


async def main():
    global bot
    bot = MyBot()
    await bot.start()


asyncio.run(main())