Как реализовать изменение данных анкеты в БД через inline клавиатуру в телеграм боте?

от
prokoding.ru > Python > Как реализовать изменение данных анкеты в БД через inline клавиатуру в телеграм боте?

Отличный вопрос! Реализация изменения данных анкеты через inline-клавиатуру — это классический и очень удобный паттерн для Telegram ботов. Я подробно разберу весь процесс, от проектирования до кода, с использованием популярной библиотеки python-telegram-bot (v13.x или v20.x).

Общая концепция и логика работы

  1. Инициация: Пользователь нажимает кнопку "Редактировать анкету" или аналогичную.
  2. Отображение текущих данных: Бот присылает сообщение с текстом анкеты и inline-клавиатурой, где каждая кнопка соответствует полю для редактирования (например, "Имя", "Возраст", "Описание").
  3. Выбор поля: Пользователь нажимает на одну из кнопок.
  4. Запрос нового значения: Бот удаляет предыдущее сообщение с клавиатурой (для чистоты чата) и запрашивает ввести новое значение для выбранного поля.
  5. Сохранение: Бот получает новое значение, обновляет запись в базе данных и подтверждает успешное изменение. После этого можно снова показать обновленную анкету с клавиатурой для дальнейшего редактирования.

---

Шаг 1: Установка зависимостей

Убедитесь, что у вас установлены необходимые библиотеки.

pip install python-telegram-bot sqlalchemy
  • python-telegram-bot: для работы с Telegram Bot API.
  • sqlalchemy: ORM для работы с базой данных (в примере использую SQLite для простоты). Вы можете заменить на любую другую БД (PostgreSQL, MySQL) или драйвер (psycopg2, pymysql).

---

Шаг 2: Модель данных (SQLAlchemy)

Создадим простую модель пользователя для анкеты.

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

# Настройка подключения к БД (SQLite)
engine = create_engine('sqlite:///users.db', echo=True)
Base = declarative_base()
Session = sessionmaker(bind=engine)

# Модель пользователя
class User(Base):
    __tablename__ = 'users'
    
    id = Column(Integer, primary_key=True)
    telegram_id = Column(Integer, unique=True, nullable=False)
    name = Column(String(50), default='Не указано')
    age = Column(Integer, default=0)
    bio = Column(String(500), default='Не указано')
    photo = Column(String(500), default='')  # Может хранить file_id фото

# Создаем таблицы
Base.metadata.create_all(engine)

---

Шаг 3: Создание Inline-клавиатур

Ключевой компонент — это клавиатуры. Мы создадим две:

  1. Главную для выбора поля.
  2. Подтверждения (опционально, но хороший тон) для отмены действия.
from telegram import InlineKeyboardButton, InlineKeyboardMarkup

# Главная клавиатура для редактирования
def edit_profile_keyboard():
    keyboard = [
        [InlineKeyboardButton("Имя", callback_data='edit_name')],
        [InlineKeyboardButton("Возраст", callback_data='edit_age')],
        [InlineKeyboardButton("О себе", callback_data='edit_bio')],
        [InlineKeyboardButton("Фото", callback_data='edit_photo')],
        [InlineKeyboardButton("Готово", callback_data='edit_done')]  # Выход из режима редактирования
    ]
    return InlineKeyboardMarkup(keyboard)

# Клавиатура для отмены во время ввода данных
def cancel_keyboard():
    keyboard = [[InlineKeyboardButton("Отмена", callback_data='cancel_edit')]]
    return InlineKeyboardMarkup(keyboard)

---

Шаг 4: Вспомогательные функции

Нам понадобятся функции для получения/сохранения пользователя и форматирования текста анкеты.

def get_user(telegram_id):
    """Получает пользователя из БД по telegram_id. Создает нового, если не найден."""
    session = Session()
    user = session.query(User).filter_by(telegram_id=telegram_id).first()
    if not user:
        user = User(telegram_id=telegram_id)
        session.add(user)
        session.commit()
    session.close()
    return user

def update_user(telegram_id, **kwargs):
    """Обновляет поля пользователя."""
    session = Session()
    user = session.query(User).filter_by(telegram_id=telegram_id).first()
    for key, value in kwargs.items():
        setattr(user, key, value)
    session.commit()
    session.close()

def format_profile_text(user):
    """Форматирует текст анкеты для отправки."""
    text = (
        f"*Твоя анкета:*nn"
        f"*Имя:* {user.name}n"
        f"*Возраст:* {user.age}n"
        f"*О себе:* {user.bio}n"
    )
    return text

---

Шаг 5: Обработчики команд и сообщений (Ядро бота)

Теперь самое интересное — логика состояний. Мы будем использовать ConversationHandler и CallbackQueryHandler для управления диалогом.

Импорты и настройка состояний

from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackQueryHandler, ConversationHandler

# Определяем состояния для ConversationHandler
EDITING, WAITING_FOR_INPUT = range(2)
# Константы для данных callback
CALLBACK_EDIT_NAME, CALLBACK_EDIT_AGE, CALLBACK_EDIT_BIO, CALLBACK_EDIT_PHOTO = 'edit_name', 'edit_age', 'edit_bio', 'edit_photo'
CALLBACK_CANCEL, CALLBACK_DONE = 'cancel_edit', 'edit_done'

Обработчик команды /start или /edit

def start(update, context):
    """Обрабатывает команду /start и показывает анкету."""
    user = get_user(update.effective_user.id)
    text = format_profile_text(user)
    
    # Если у пользователя есть фото, отправляем фото с подписью и клавиатурой
    if user.photo:
        update.message.reply_photo(
            photo=user.photo,
            caption=text,
            parse_mode='Markdown',
            reply_markup=edit_profile_keyboard()
        )
    else:
        # Иначе просто текстовое сообщение
        update.message.reply_text(
            text,
            parse_mode='Markdown',
            reply_markup=edit_profile_keyboard()
        )
    # Переводим бота в состояние "редактирования"
    return EDITING

Обработчик нажатий на inline-кнопки

Это сердце функционала.

def button_handler(update, context):
    """Обрабатывает нажатия на кнопки inline-клавиатуры."""
    query = update.callback_query
    query.answer()  # Важно! Чтобы убрать "часики" на кнопке
    user_id = query.from_user.id
    data = query.data

    # Обрабатываем кнопку "Готово"
    if data == CALLBACK_DONE:
        query.edit_message_text("Редактирование завершено!")
        return ConversationHandler.END

    # Обрабатываем кнопку "Отмена"
    if data == CALLBACK_CANCEL:
        # Возвращаем пользователя к просмотру анкеты
        user = get_user(user_id)
        text = format_profile_text(user)
        query.edit_message_text(text, parse_mode='Markdown', reply_markup=edit_profile_keyboard())
        return EDITING

    # Обрабатываем выбор поля для редактирования
    # Сохраняем выбранное поле в context.user_data
    context.user_data['editing_field'] = data

    # Запрашиваем новое значение в зависимости от поля
    if data == CALLBACK_EDIT_NAME:
        prompt = "Введите новое имя:"
    elif data == CALLBACK_EDIT_AGE:
        prompt = "Введите новый возраст (число):"
    elif data == CALLBACK_EDIT_BIO:
        prompt = "Введите новое описание:"
    elif data == CALLBACK_EDIT_PHOTO:
        prompt = "Отправьте новое фото:"

    # Удаляем сообщение с кнопками и запрашиваем ввод
    query.edit_message_text(prompt, reply_markup=cancel_keyboard())

    # Переводим бота в состояние ожидания ввода
    return WAITING_FOR_INPUT

Обработчик ввода новых данных от пользователя

def save_input(update, context):
    """Сохраняет введенные пользователем данные."""
    user_id = update.message.from_user.id
    user_data = context.user_data
    editing_field = user_data.get('editing_field')

    if not editing_field:
        update.message.reply_text("Что-то пошло не так. Начните с /start.")
        return ConversationHandler.END

    new_value = update.message.text

    # Валидация и преобразование данных в зависимости от поля
    try:
        if editing_field == CALLBACK_EDIT_AGE:
            new_value = int(new_value)
            if new_value < 0 or new_value > 120:
                update.message.reply_text("Пожалуйста, введите корректный возраст (0-120).", reply_markup=cancel_keyboard())
                return WAITING_FOR_INPUT
        # Для имени и био можно добавить проверку на длину
        elif editing_field == CALLBACK_EDIT_NAME and len(new_value) > 50:
            update.message.reply_text("Имя слишком длинное. Макс. 50 символов.", reply_markup=cancel_keyboard())
            return WAITING_FOR_INPUT
        elif editing_field == CALLBACK_EDIT_BIO and len(new_value) > 500:
            update.message.reply_text("Описание слишком длинное. Макс. 500 символов.", reply_markup=cancel_keyboard())
            return WAITING_FOR_INPUT
    except ValueError:
        update.message.reply_text("Пожалуйста, введите число для возраста.", reply_markup=cancel_keyboard())
        return WAITING_FOR_INPUT

    # Маппинг callback_data на названия полей в БД
    field_map = {
        CALLBACK_EDIT_NAME: 'name',
        CALLBACK_EDIT_AGE: 'age',
        CALLBACK_EDIT_BIO: 'bio'
    }
    db_field = field_map[editing_field]

    # Обновляем данные в БД
    update_user(user_id, **{db_field: new_value})

    # Подтверждаем обновление и снова показываем анкету
    user = get_user(user_id)
    text = format_profile_text(user)
    # Удаляем сообщение с запросом ввода (опционально)
    # context.bot.delete_message(chat_id=update.message.chat_id, message_id=update.message.message_id)
    update.message.reply_text("Данные обновлены!", reply_markup=edit_profile_keyboard())
    
    # Чистим временные данные
    user_data.pop('editing_field', None)

    return EDITING

Обработчик для загрузки фото

def save_photo(update, context):
    """Обрабатывает загрузку фото."""
    user_id = update.message.from_user.id
    # Получаем file_id самого большого размера фото
    photo_file = update.message.photo[-1].file_id
    
    # Сохраняем file_id в БД
    update_user(user_id, photo=photo_file)
    
    # Подтверждаем и показываем анкету
    user = get_user(user_id)
    text = format_profile_text(user)
    # Отправляем фото с обновленной анкетой
    update.message.reply_photo(
        photo=photo_file,
        caption=text,
        parse_mode='Markdown',
        reply_markup=edit_profile_keyboard()
    )
    
    # Чистим временные данные
    context.user_data.pop('editing_field', None)
    return EDITING

---

Шаг 6: Настройка диалога (ConversationHandler) и запуск бота

Собираем все обработчики вместе.

def main():
    # Замените 'YOUR_BOT_TOKEN' на реальный токен
    updater = Updater("YOUR_BOT_TOKEN", use_context=True)
    dp = updater.dispatcher

    # Создаем ConversationHandler
    conv_handler = ConversationHandler(
        entry_points=[CommandHandler('start', start)],
        states={
            EDITING: [CallbackQueryHandler(button_handler)],
            WAITING_FOR_INPUT: [
                MessageHandler(Filters.text & ~Filters.command, save_input),
                MessageHandler(Filters.photo, save_photo),
                CallbackQueryHandler(button_handler, pattern=f'^{CALLBACK_CANCEL}$')
            ],
        },
        fallbacks=[CommandHandler('start', start)], # Если что-то пошло не так, перезапускаем
    )

    dp.add_handler(conv_handler)

    # Запускаем бота
    updater.start_polling()
    updater.idle()

if __name__ == '__main__':
    main()

---

Полный пример кода и ключевые моменты

Весь код из примеров выше, собранный в один файл (условно).

Важные замечания по улучшению и масштабированию:

  1. Безопасность: Добавьте проверки, что пользователь изменяет свою анкету, а не чужую (в простом примере это и так работает через user_id).
  2. Валидация: Расширьте валидацию введенных данных (например, проверка имени на наличие запрещенных символов).
  3. Отмена на любом этапе: Реализована через кнопку "Отмена", которая возвращает к анкете.
  4. Управление сообщениями: Для лучшего UX можно удалять сообщения с запросами ввода, чтобы не засорять чат. Используйте context.bot.delete_message.
  5. Кэширование: Для снижения нагрузки на БД можно кэшировать объекты пользователей в context.user_data на время сессии.
  6. Производительность БД: Для высоконагруженных ботов используйте пулы соединений и асинхронные драйверы (например, asyncpg для PostgreSQL с aiopg или sqlalchemy.ext.asyncio).
  7. Версия 20.x: Код написан для версии 13.x. Для 20.x (где используется асинхронный подход) потребуется изменить обработчики на асинхронные функции (async def) и использовать ApplicationBuilder вместо Updater.

Этот код представляет собой надежный каркас для реализации функционала редактирования анкеты через inline-клавиатуру. Вы можете легко адаптировать его под свои нужды, добавляя новые поля и сложную логику.