Бот не считает формулу. Как исправить?

Question

[Gryandr]

Бот должен принять число от пользователя и сделать расчеты по формуле.
Бот работает до момента, где пользователь делает выбор и вводит число.

from aiogram import Bot, types
from aiogram.dispatcher import Dispatcher
from aiogram.utils import executor
from config import token
from math import sqrt

bot = Bot(token)
dp = Dispatcher(bot)

async def on_startup(_):
	print('Бот онлайн.')

'''****КНОПКИ БАССЕЙНОВ***'''

@dp.message_handler(commands=['start', 'help'])
async def command_start(message : types.Message):
    markup=types.ReplyKeyboardMarkup(resize_keyboard=True)
    item1=types.KeyboardButton("Бассейн №1")
    item2=types.KeyboardButton("Бассейн №2")
    markup.add(item1)
    markup.add(item2)
    await bot.send_message(message.from_user.id, 'Привет! Какой бассейн меряем?', reply_markup=markup)

@dp.message_handler()
async def lake(message):
    if message.text == 'Бассейн №1':
        await bot.send_message(message.from_user.id, 'Введи расстояние до воды №1.')
    if message.text == 'Бассейн №2':
        await bot.send_message(message.from_user.id, 'Введи расстояние до воды №2.')

'''****ФОРМУЛЫ***'''

@dp.message_handler(content_types=['text'])
async def lake1(message):
    if message.text == 'Бассейн №1':
        b1 = int(message.text)
        h = 8 - b1
        v = (h / 3) * (2500 + (sqrt(2500 * (95 - (b1 * 110) - (b1 / 2))))) + (95 - (b1 * 110) - (b1 / 2))
        level = v / 16000 * 100
        await bot.send_message(message.from_user.id, str("Количество:" + str(v) + "м3. Глубина:" + str(h) + "м. Наполнение: " + str(level) + "%"))
    else:
        await bot.send_message(message.from_user.id, "Попробуй еще...")

@dp.message_handler(content_types=['text'])
async def lake2(message):
    if message.text == 'Бассейн №2':
        b2 = int(message.text)
        h2 = 5 - ((b2 - 1) * float(0.7))
        s2 = (90 - (b2 * 0.52)) * (100 - (b2 * 0.52)) / 2
        n = sqrt(2500 * s2)
        g = s1 + n + s2
        v2 = (h2 / 3) * g
        level2 = v2 / 17000 * 100
        await bot.send_message(message.from_user.id, str("Количество:" + str(v2) + "м3. Глубина:" + str(h2) + "м. Наполнение: " + str(level2) + "%"))
    else:
        await bot.send_message(message.from_user.id, 'Чтото не так!')

executor.start_polling(dp, on_startup=on_startup)

>

Comments

[Модератор]

См.п.3.8
Необходимо вставить код в вопрос в виде текста, обернутого тегом code.

[Gryandr]

Модератор, Спасибо! Исправил.

Answer 1

[Алан Гибизов]

Как я понял, когда бот начинает работать, он обрабатывает зарегистрированные в нем хэндлеры, и ваш хэндлер lake перехватывает всё, кроме явно start и help, поэтому дальше него и не проходит. Я глубоко не вникал, возможно этот момент я неправильно понял.
Но, в принципе, можно обойтись и этим одним хндлером, просто правильно обрабатывать то, что в него приходит. А приходить в него должно либо выбор кнопки, либо число.
При выборе кнопки фиксируем номер кнопки и ждем число. Если пришло число - вычисляем, возвращаем, сбрасываем номер кнопки и всё сначала.

lake_number = 0


@dp.message_handler()
async def lake(message: types.Message):
    global lake_number
    if message.text == 'Бассейн №1':
        await bot.send_message(message.from_user.id, 'Введи расстояние до воды №1.')
        lake_number = 1
    if message.text == 'Бассейн №2':
        await bot.send_message(message.from_user.id, 'Введи расстояние до воды №2.')
        lake_number = 2
    if bool(lake_number) and message.text.isdigit():
        b = int(message.text)
        if lake_number == 1:
            h = 8 - b
            v = (h / 3)
            level = v / 16000 * 100  # тут в формуле где-то ошибка, поэтому я ее просто убрал
            await bot.send_message(message.from_user.id,
                                   str("Количество:" + str(v) + "м3. Глубина:" + str(h) + "м. Наполнение: " + str(
                                       level) + "%"))
        if lake_number == 2:
            h2 = 5 - ((b - 1) * float(0.7))
            s2 = (90 - (b * 0.52)) * (100 - (b * 0.52)) / 2
            n = sqrt(2500 * s2)
            s1 = 1
            g = s1 + n + s2
            v2 = (h2 / 3) * g
            level2 = v2 / 17000 * 100
            await bot.send_message(message.from_user.id,
                                   str("Количество:" + str(v2) + "м3. Глубина:" + str(h2) +
                                       "м. Наполнение: " + str(level2) + "%"))
        lake_number = 0
        await bot.send_message(message.from_user.id, "Попробуй еще...")

Comments

[Алан Гибизов]

Правда, всё это совершенно кривое безобразие и я сделал всё именно так потому, что во-первых не знаю вообще этих ваших ботов, а во-вторых, лень вдумываться и причесывать.
Если бы я делал всё это по уму, я бы наверно глубже разобрался в этих ваших хэндлерах и условиях, которые они принимают, там наверняка можно сделать правильные условия, чтобы отловить цифры и правильно обработать. Ну и надо хорошенько подумать, где и как хранить состояние (нажатые до того кнопки, например). Я тупо положил номер кнопки в глобальную переменную, но это не очень хорошо.

[Gryandr]

Спасибо за вашу помощь! Использовал ваше решение и бот заработал, считает. Но образовалась пролемка)))
Бот принимает только целые числа потому что задано isdigital
Пробовал заменить на isfloat из модуля fastnumbers ... не получилось

@dp.message_handler()
async def lake(message: types.Message):
    global lake_number
    if message.text == 'Бассейн №1':
        await bot.send_message(message.from_user.id, 'Введи растояние до воды №1.')
        lake_number = 1
    if message.text == 'Бассейн №2':
        await bot.send_message(message.from_user.id, 'Введи растояние до воды №2.')
        lake_number = 2
    if bool(lake_number) and message.text.isdigit():
        b = int(message.text)
        if lake_number == 1:
            c = b * 0.52
            square = (95 - c * 2) * (50 - c * 2)
            h = (6 - b) * 0.7
            y = sqrt(4000 * square)
            v = round(h / 3 * (4000 + square + y))
            level = round(v / 15000 * 100)
            await bot.send_message(message.from_user.id,
                                   str(f"БАССЕЙН №1 \U0001F536\n"
                                       f"Количество: " + str(v) + " м3\n"
                                       f"Глубина: " + str(h) + " м\n"
                                       f"Наполнение: " + str(level) + " %"))
        if lake_number == 2:
            h2 = (7 - b) * 0.7
            s2 = (90 - (b * 0.52)) * (100 - (b * 0.52)) / 2
            n = sqrt(2500 * s2)
            s1 = 2500
            g = s1 + n + s2
            v2 = round((h2 / 3) * g)
            level2 = round(v2 / 14000 * 100)
            await bot.send_message(message.from_user.id,
                                   str(f"БАССЕЙН №2 \U0001F53A\n"
                                       f"Количество:   " + str(v2) + " м3\n"
                                       f"Глубина:    " + str(h2) + " м\n"
                                       f"Наполнение: " + str(level2) + " %"))
	lake_number = 0
        await bot.send_message(message.from_user.id, "Попробуй еще...")

[Алан Гибизов]

Gryandr, вот к примеру (нагуглил быстренько)

def is_digit(string):
    if string.isdigit():
       return True
    else:
        try:
            float(string)
            return True
        except ValueError:
            return False

Вообще, если для вас это сложно найти или сделать самому, вам рано заниматься ботами.

[Gryandr]

Алан Гибизов, понял))) Спасибо за помощь и терпение!

Answer 2

[Zo0meeer]

Бот на aiogram регистрирует хендлеры сверху вниз. Сначала регистрируется хендлер lake, потому что по умолчанию в любом хендлере стоит content_type='text'. Если хотите перехватывать выбор с ReplyKeyboardMurkup, мой совет использовать FSM с одним хендлером.
И код

if message.text == 'Бассейн №1':
        b1 = int(message.text)

замените на

if message.text == 'Бассейн №1':
        b1 = int(message.text[-1])

Answer 3

[Александр Нестеров]

if message.text == 'Бассейн №1':
        b1 = int(message.text)

Я с ТГ-ботами "на вы", но вот этот кусок выглядит как лютая дичь.
Чисто логически, переводя на "человеческий", это значит:

Если текст сообщения выглядит как 'Бассейн №1', то пусть этот текст станет числом.

Не находите противоречия?

Comments

[Gryandr]

Не ругайте сильно. Пробовал разные варианты.
Можно упростить и забрать у пользователя право выбора, попросив ввести данные двух бассейнов через пробел и задать числа так:

def lakes(message):
    a, b = message.text.split()
    b1 = int(a)
    b2 = int(b)

Но хотелось бы сохранить кнопки и если пользователь выберет Бассейн №1, принять его значение для формулы.

[Александр Нестеров]

Gryandr, я бы советовал посмотреть в сторону FSM. Это позволит вводить данные последовательно, через несколько сообщений.