init commit 2

bot/bot.py

@@ -0,0 +1,39 @@
+from telegram import Update
+from telegram.ext import Updater, CommandHandler, CallbackContext
+from users.models import User, UserConfirmation
+import uuid
+
+def start(update: Update, context: CallbackContext):
+    user_id = update.message.from_user.id
+    chat_id = update.message.chat_id
+
+    user, created = User.objects.get_or_create(telegram_id=user_id, defaults={'chat_id': chat_id})
+    if not user.confirmed:
+        confirmation_code = str(uuid.uuid4())
+        UserConfirmation.objects.create(user=user, confirmation_code=confirmation_code)
+        update.message.reply_text(f"Ваш код подтверждения: {confirmation_code}")
+    else:
+        update.message.reply_text("Вы уже зарегистрированы!")
+
+def confirm(update: Update, context: CallbackContext):
+    user_id = update.message.from_user.id
+    code = ' '.join(context.args)
+
+    try:
+        confirmation = UserConfirmation.objects.get(user__telegram_id=user_id, confirmation_code=code)
+        confirmation.user.confirmed = True
+        confirmation.user.save()
+        confirmation.delete()
+        update.message.reply_text("Регистрация подтверждена!")
+    except UserConfirmation.DoesNotExist:
+        update.message.reply_text("Неверный код подтверждения!")
+
+def main():
+    updater = Updater("YOUR_TELEGRAM_BOT_TOKEN")
+    dispatcher = updater.dispatcher
+
+    dispatcher.add_handler(CommandHandler("start", start))
+    dispatcher.add_handler(CommandHandler("confirm", confirm))
+
+    updater.start_polling()
+    updater.idle()