Add files via upload

Initial project upload

handlers/registration.py

@@ -0,0 +1,106 @@
+from handlers.menu import show_main_menu
+from keyboards.reply_keyboards import get_phone_keyboard, get_business_types_keyboard, BUSINESS_TYPES
+from database import get_user, save_user
+
+# Хранилище для временных данных пользователей
+user_data = {}
+
+def process_registration(message, bot):
+    """Обработка регистрации"""
+    user_id = message.from_user.id
+    text = message.text
+    step = user_data[user_id].get('step', 1)
+    
+    if step == 1:
+        # Сохраняем имя
+        user_data[user_id]['name'] = text
+        user_data[user_id]['step'] = 2
+        
+        bot.send_message(
+            user_id,
+            f"👋 Приятно познакомиться, {text}!\n\n"
+            "📱 Теперь отправь мне свой номер телефона.\n"
+            "Нажми на кнопку ниже 👇",
+            reply_markup=get_phone_keyboard()
+        )
+    
+    elif step == 2:
+        # Сохраняем телефон
+        if message.contact:
+            phone = message.contact.phone_number
+        else:
+            phone = text
+        
+        user_data[user_id]['phone'] = phone
+        user_data[user_id]['step'] = 3
+        
+        bot.send_message(
+            user_id,
+            "📋 Выбери свой вид предпринимательства:",
+            reply_markup=get_business_types_keyboard()
+        )
+    
+    elif step == 3:
+        # Проверяем выбор вида бизнеса
+        selected_type = None
+        for key, value in BUSINESS_TYPES.items():
+            if text == f"{value['emoji']} {key}. {value['name']}":
+                selected_type = int(key)
+                break
+        
+        if not selected_type:
+            bot.send_message(
+                user_id,
+                "❌ Пожалуйста, выбери вид бизнеса из кнопок ниже:",
+                reply_markup=get_business_types_keyboard()
+            )
+            return
+        
+        user_data[user_id]['business_type'] = selected_type
+        user_data[user_id]['step'] = 4
+        
+        bot.send_message(
+            user_id,
+            f"✅ Отлично! Ты выбрал: {BUSINESS_TYPES[str(selected_type)]['name']}\n\n"
+            "📝 Теперь расскажи немного о своем бизнесе:\n"
+            "(Чем занимаешься? Сколько лет на рынке? Какие цели?)",
+            reply_markup=None
+        )
+    
+    elif step == 4:
+        # Сохраняем описание
+        description = text
+        
+        save_user(
+            user_id,
+            user_data[user_id]['name'],
+            user_data[user_id]['phone'],
+            user_data[user_id]['business_type'],
+            description
+        )
+        
+        # Сохраняем вид бизнеса для дальнейшего использования
+        business_type_name = BUSINESS_TYPES[str(user_data[user_id]['business_type'])]['name']
+        
+        # Очищаем временные данные
+        del user_data[user_id]
+        
+        welcome_text = f"""
+🎉 ПОЗДРАВЛЯЮ! РЕГИСТРАЦИЯ ЗАВЕРШЕНА!
+
+✅ Ты зарегистрирован как предприниматель!
+📋 Тип бизнеса: {business_type_name}
+
+💡 Теперь я буду помогать тебе:
+• Генерировать контент для соцсетей
+• Давать советы по бизнесу
+• Анализировать твою нишу
+
+Нажми «💡 Предложить контент» в меню! 👇
+        """
+        bot.send_message(user_id, welcome_text)
+        show_main_menu(user_id, bot)
+
+def is_in_registration(user_id):
+    """Проверяет, находится ли пользователь в процессе регистрации"""
+    return user_id in user_data