romaaaka/queuebot: Код бота для создания виртуальных очередей

src/services/chat.py

35 строк1 KB

Roman

dockerizing

14 июн 2024, 11:55

14 июн 2024, 11:55427bfd1

from vkwave.api import APIOptionsRequestContext
from vkwave.bots import SimpleBotEvent, ForWhat

import settings
from models import User, Chat
from resources import messages
from intances import fsm
from middlewares.types.states import AddGroupStates


class ChatService:

    def __init__(self, api_context: APIOptionsRequestContext):
        self.api_context = api_context

    async def create_chat(self, event: SimpleBotEvent, user: User, chat: Chat):
        if user.chats.count() > settings.MAX_USER_GROUPS:
            await self.api_context.messages.send(
                user_id=user.user_id,
                message=messages.MAX_GROUP_REACHED,
                random_id=0
            )
        chat.owner = user
        chat.save()
        await self.api_context.messages.send(
            user_id=user.user_id,
            message=messages.ADD_GROUP
        )
        await fsm.set_state(event=event, state=AddGroupStates.NAMING, for_what=ForWhat.FOR_USER)

    async def name_chat(self, event: SimpleBotEvent, user: User, chat: Chat):
        name = event.object.object.message.text
        chat.name = name
        chat.save()
        await fsm.finish(event=event, for_what=ForWhat.FOR_USER)