主页 / user-505252

Zlets's questions

Martin Hope
Zlets
Asked: 2022-08-25 00:32:35 +0800 CST
  • 2
from vkbottle import Bot, Message
from vkbottle.api.keyboard import Keyboard, Text
from vkbottle.branch import Branch, ExitBranch


bot=Bot('')

dialogs = {}
wait = []
emoji = '' # Спасибо моему IDLE :)

start_keyboard = Keyboard(one_time=False)
start_keyboard.add_row()
start_keyboard.add_button(Text(label="Поиск собеседника"), color="negative")

wait_keyboard = Keyboard(one_time=True)
wait_keyboard.add_row()
wait_keyboard.add_button(Text(label="Отменить поиск"), color="positive")

stop_keyboard = Keyboard(one_time=False)
stop_keyboard.add_row()
stop_keyboard.add_button(Text(label="Отключиться от диалога"), color="primary")


@bot.on.message(text='Поиск собеседника', lower = True)
async def start(ans: Message):
    if ans.from_id not in wait and ans.from_id not in dialogs:
        if len(wait) == 0:
            await ans(f'{emoji} Вы добавлены в очередь поиска собеседника.', keyboard=wait_keyboard)
            wait.append(ans.from_id)
            await bot.branch.add(ans.peer_id, "wait")
        else:
            dialogs[ans.from_id] = wait[0]
            dialogs[wait[0]] = ans.from_id
            await bot.api.messages.send(peer_id=ans.from_id, random_id=0, message=f'{emoji} Мы нашли вам собеседника!', keyboard=stop_keyboard)
            await bot.api.messages.send(peer_id=wait[0], random_id=0, message=f'{emoji} Мы нашли вам собеседника!', keyboard=stop_keyboard)
            await bot.branch.add(ans.from_id, "dialog")
            await bot.branch.add(wait[0], "dialog")
            del wait[0]

@bot.branch.simple_branch("wait")
async def branch(ans: Message):
    if ans.text == "Отменить поиск":
        await ans(f"{emoji} Поиск собеседника остановлен.", keyboard=start_keyboard)
        await bot.branch.exit(ans.peer_id)
        del wait[0]
    else:
        await ans(f'{emoji} Вы находитесь в поиске собеседника.', keyboard=wait_keyboard)

@bot.branch.simple_branch("dialog")
async def branch(ans: Message):
    if ans.text == "Отключиться от диалога":
        await bot.api.messages.send(peer_id=ans.from_id, random_id=0, message=f'{emoji} Диалог был остановлен.', keyboard=start_keyboard)
        await bot.api.messages.send(peer_id=dialogs[ans.from_id], random_id=0, message=f'{emoji} Собеседник остановил диалог.', keyboard=start_keyboard)
        await bot.branch.exit(dialogs[ans.from_id])
        await bot.branch.exit(ans.from_id)
        del dialogs[dialogs[ans.from_id]]
        del dialogs[ans.from_id]
    else:
        await bot.api.messages.send(peer_id=dialogs[ans.from_id], random_id=0, message='Собеседник: ' + ans.text)

@bot.on.message()
async def all(ans: Message):
    await ans('Привет! Я анонимный чат-бот. Чтобы начать поиск собеседника, воспользуйтесь кнопками.', keyboard=start_keyboard)

bot.run_polling()
python вконтакте