Не переходит на следующую функцию

# handlers.py
 
from telegram import Update, InlineKeyboardMarkup, InlineKeyboardButton
from telegram.ext import ContextTypes, ConversationHandler
from models import add_user_request
import logging
 
# Логгирование
logging.basicConfig(format='%(asctime)s - %(levelname)s - %(message)s', level=logging.INFO)
logger = logging.getLogger(__name__)
 
# Константы состояний для ConversationHandler
NAME, PHONE_NUMBER, COMPANY_NAME, CONTACT_TIME, ISSUE_DESC, CONTACT_METHOD = range(6)
 
# Обработчик для команды /start
async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
    logger.info(f"Start command received from {update.effective_message.from_user.id}")
 
    # Короткое описание компании
    welcome_text = """
    Привет! Мы - Arassa NUSGA. Мы стараемся поддерживать наших клиентов.
    Вот наши контактные данные:
    Телефон: +1 800 123 4567
    Сайт: https://arassanusga.com
    Instagram: @arassanusga
    Почта: support@arassanusga.com
    """
    await update.message.reply_text(welcome_text)
 
    # Запрашиваем имя клиента
    await update.message.reply_text("Как к вам обращаться?")
    return NAME # Переход к состоянию NAME
 
async def button(update: Update, context: ContextTypes.DEFAULT_TYPE):
    query = update.callback_query
    data = query.data  # Получаем callback data
    logger.info(f"Button '{data}' pressed by {update.effective_message.from_user.id}")
 
    # Обрабатываем разные кнопки
    if data == 'about':
        await query.answer()  # Подтверждаем обработку
        await query.message.reply_text("Пожалуйста, укажите название вашей компании.")
        return COMPANY_NAME
 
    # elif data == 'support':
    #     # Здесь сразу обрабатываем запрос на техподдержку
    #     await query.answer()  # Подтверждаем обработку
    #     # await query.message.reply_text("Пожалуйста, опишите вашу проблему.")
 
    #     # Устанавливаем состояние для сбора описания проблемы
    #     # context.user_data['stage'] = 'ISSUE_DESC'
    #     logger.info(f"Updated state in user_data: {context.user_data.get('stage')}")
 
    elif data == 'support':
        await query.answer()  # Подтверждаем обработку
        context.user_data['stage'] = 'ISSUE_DESC'  # Обновляем состояние
        logger.info(f"Stage updated to: {context.user_data.get('stage')}")
 
        # Переход к состоянию для сбора описания проблемы
        return ISSUE_DESC
 
    elif data == 'call':
        await call_request(update, context)
 
 
 
# Обработчик для ввода имени
async def name(update: Update, context: ContextTypes.DEFAULT_TYPE):
    user_name = update.effective_message.text
    telegram_id = update.effective_message.from_user.id
    logger.info(f"Received name from {telegram_id}: {user_name}")
 
    # Сохраняем имя в БД
    add_user_request(telegram_id=telegram_id, name=user_name)
 
    # Запрос кнопок
    keyboard = [
        [InlineKeyboardButton("Хочу узнать об Arassa NUSGA", callback_data='about')],
        [InlineKeyboardButton("Я уже работаю в Arassa NUSGA", callback_data='working')]
    ]
    reply_markup = InlineKeyboardMarkup(keyboard)
 
    await update.message.reply_text(f"Спасибо, {user_name}! Выберите опцию:", reply_markup=reply_markup)
    return COMPANY_NAME
 
 
# async def support_request(update: Update, context: ContextTypes.DEFAULT_TYPE):
#     issue = update.effective_message.text
 
#     user_id = update.effective_message.from_user.id
#     logger.info(f"Button 'Запросить тех. поддержку' pressed by {user_id}")
 
#     # Логируем состояние
#     logger.info(f"Current state in user_data: {context.user_data.get('stage')}")
 
#     # Переход на этап ввода описания проблемы
#     await update.callback_query.answer()
#     await update.callback_query.message.reply_text("Пожалуйста, опишите вашу проблему.")
 
#     # Устанавливаем новое состояние
#     context.user_data['stage'] = 'ISSUE_DESC'
#     logger.info(f"Updated state in user_data: {context.user_data.get('stage')}")
 
#     # Переходим в состояние ISSUE_DESC
#     return ISSUE_DESC
 
 
 
# Обработчик для кнопки "Запросить звонок"
async def call_request(update: Update, context: ContextTypes.DEFAULT_TYPE):
    logger.info(f"Button 'Запросить звонок' pressed by {update.effective_message.from_user.id}")
 
    # Переход на этап ввода контактного времени
    await update.callback_query.answer()  # Подтверждаем обработку коллбека
    await update.callback_query.message.reply_text("Пожалуйста, укажите ваш номер телефона и время, когда с вами можно связаться.")
 
    # Возвращаем состояние CONTACT_TIME
    return CONTACT_TIME  # Переходим на стадию CONTACT_TIME
 
# Обработчик для ввода номера телефона
async def phone_number(update: Update, context: ContextTypes.DEFAULT_TYPE):
    phone = update.effective_message.text
    telegram_id = update.effective_message.from_user.id
    logger.info(f"Received phone number from {telegram_id}: {phone}")
 
    # Сохраняем номер телефона в БД
    add_user_request(telegram_id=telegram_id, phone_number=phone)
 
    await update.message.reply_text(f"Спасибо! Ваш номер телефона {phone} был принят.")
    return ConversationHandler.END
 
# Обработчик для ввода названия компании
async def company_name(update: Update, context: ContextTypes.DEFAULT_TYPE):
    company = update.effective_message.text
    telegram_id = update.effective_message.from_user.id
    logger.info(f"Received company name from {telegram_id}: {company}")
 
    # Сохраняем название компании в БД
    add_user_request(telegram_id=telegram_id, company_name=company)
 
    # Выводим текст о поддержке
    await update.message.reply_text("Мы стараемся поддерживать наших клиентов, нам очень важно быть с вами на связи.")
 
    # Кнопки для дальнейших действий
    keyboard = [
        [InlineKeyboardButton("Запросить тех. поддержку", callback_data='support')],
        [InlineKeyboardButton("Запросить звонок", callback_data='call')]
    ]
    reply_markup = InlineKeyboardMarkup(keyboard)
 
    # Сообщаем пользователю и предлагаем дальнейшие действия
    await update.message.reply_text(f"Спасибо! Название вашей компании: {company}.")
    await update.message.reply_text("Что бы вы хотели сделать?", reply_markup=reply_markup)
 
    return ConversationHandler.END
 
# Обработчик для ввода описания проблемы
async def issue_desc(update: Update, context: ContextTypes.DEFAULT_TYPE):
    issue = update.effective_message.text
    telegram_id = update.effective_message.from_user.id
    logger.info(f"Received issue description from {telegram_id}: {issue}")
 
    # Сохраняем описание проблемы в БД
    add_user_request(telegram_id=telegram_id, issue_desc=issue)
 
    # Запрашиваем способ связи
    await update.message.reply_text("Как с вами связаться? Укажите способ связи (например, телефон, email, Telegram).")
 
    # Переход к состоянию CONTACT_METHOD
    context.user_data['stage'] = 'CONTACT_METHOD'
    return CONTACT_METHOD  # Переход к следующему состоянию
 
# Обработчик для ввода времени и контакта
async def contact_time(update: Update, context: ContextTypes.DEFAULT_TYPE):
    contact_info = update.effective_message.text
    telegram_id = update.effective_message.from_user.id
    logger.info(f"Received contact time and info from {telegram_id}: {contact_info}")
 
    # Сохраняем контактное время в БД
    add_user_request(telegram_id=telegram_id, contact_time=contact_info)
 
    await update.message.reply_text(f"Спасибо! Мы свяжемся с вами в указанное время: {contact_info}.")
    return ConversationHandler.END
 
    # Обработчик для ввода способа связи
async def contact_method(update: Update, context: ContextTypes.DEFAULT_TYPE):
    contact_method = update.effective_message.text
    telegram_id = update.effective_message.from_user.id
    logger.info(f"Received contact method from {telegram_id}: {contact_method}")
 
    # Сохраняем способ связи в поле phone_number (или можно использовать другое поле)
    add_user_request(telegram_id=telegram_id, phone_number=contact_method)
 
    # Подтверждаем получение способа связи
    await update.message.reply_text(f"Спасибо! Мы будем связываться с вами через: {contact_method}")
 
    # Завершаем разговор
    return ConversationHandler.END

# main.py
import logging
from telegram import Update
from telegram.ext import Application, CommandHandler, MessageHandler, filters, ConversationHandler, CallbackQueryHandler
from handlers import start, name, button, call_request, issue_desc, contact_time, phone_number, company_name, contact_method
from config import TOKEN
 
# Настройка логгирования
logging.basicConfig(format='%(asctime)s - %(levelname)s - %(message)s', level=logging.INFO)
logger = logging.getLogger(__name__)
 
# Константы состояний для ConversationHandler
NAME, PHONE_NUMBER, COMPANY_NAME, CONTACT_TIME, ISSUE_DESC, CONTACT_METHOD = range(6)
 
# Функция для добавления обработчиков
def main():
    application = Application.builder().token(TOKEN).build()  # Используем TOKEN из config.py
 
    # Обработчики команд и состояний
    application.add_handler(CommandHandler("start", start))
 
    # ConversationHandler для работы с состояниями
    conv_handler = ConversationHandler(
        entry_points=[MessageHandler(filters.TEXT & ~filters.COMMAND, name)],
 
        states={
            NAME: [MessageHandler(filters.TEXT & ~filters.COMMAND, name)],
            PHONE_NUMBER: [MessageHandler(filters.TEXT & ~filters.COMMAND, phone_number)],
            COMPANY_NAME: [MessageHandler(filters.TEXT & ~filters.COMMAND, company_name)],
            CONTACT_TIME: [MessageHandler(filters.TEXT & ~filters.COMMAND, contact_time)],
            ISSUE_DESC: [MessageHandler(filters.TEXT & ~filters.COMMAND, issue_desc)],
            CONTACT_METHOD: [MessageHandler(filters.TEXT & ~filters.COMMAND, contact_method)],
        },
        fallbacks=[CommandHandler("cancel", start)],  # Можно использовать /cancel для выхода из беседы
    )
 
    application.add_handler(conv_handler)
    application.add_handler(CallbackQueryHandler(button, pattern="^about$|^working$|^support$|^call$"))
 
    application.run_polling()
 
if __name__ == '__main__':
    main()

# models.py
 
from sqlalchemy import create_engine, Column, Integer, BigInteger, String, Text, DateTime
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
import datetime
from config import DATABASE_URL
 
# Создаем подключение к базе данных
engine = create_engine(DATABASE_URL)
Base = declarative_base()
 
# Сессия для работы с БД
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
# Модель для хранения запросов клиентов
class UserRequest(Base):
    __tablename__ = 'users'  # Убедитесь, что имя таблицы совпадает с тем, что у вас в БД
 
    id = Column(Integer, primary_key=True, index=True)  # Поле для уникального идентификатора
    telegram_id = Column(BigInteger, index=True)  # Убираем уникальность для telegram_id
    name = Column(String, index=True)
    company_name = Column(String, nullable=True)
    phone_number = Column(String, nullable=True)
    contact_time = Column(String, nullable=True)
    issue_desc = Column(Text, nullable=True)
    created_at = Column(DateTime, default=datetime.datetime.utcnow)  # Дата создания запроса
    updated_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow)  # Дата обновления запроса
 
# Функция для добавления запроса в БД
def add_user_request(telegram_id, name=None, company_name=None, phone_number=None, contact_time=None, issue_desc=None):
    session = SessionLocal()  # Создание сессии для работы с БД
 
    # Создаем новый запрос от пользователя (он может повторяться)
    db_user = UserRequest(
        telegram_id=telegram_id,
        name=name,
        company_name=company_name,
        phone_number=phone_number,
        contact_time=contact_time,
        issue_desc=issue_desc,
        created_at=datetime.datetime.utcnow(),
        updated_at=datetime.datetime.utcnow()
    )
 
    # Добавляем пользователя в сессию и сохраняем изменения в БД
    session.add(db_user)
    session.commit()
    session.refresh(db_user)  # Обновляем объект с добавленными данными
    session.close()  # Закрываем сессию
 
    return db_user  # Возвращаем объект пользователя с данными
 
Base.metadata.create_all(bind=engine)