56 lines
2.5 KiB
Python
56 lines
2.5 KiB
Python
import telebot
|
||
from django.contrib.auth.models import User
|
||
from django.core.management.base import BaseCommand
|
||
from telebot.types import Message
|
||
|
||
from Main.models import UserInfo
|
||
|
||
bot = telebot.TeleBot("1994460106:AAGrGsCZjF6DVG_T-zycELuVfxnWw8x7UyU")
|
||
|
||
|
||
@bot.message_handler(commands=["start"])
|
||
def do_action(message: Message):
|
||
bot.send_message(message.chat.id, "Привет! Я тут чтобы помогать!\n/register - зарегистрироваться в сервисе\nБольше команд нет:(")
|
||
|
||
|
||
@bot.message_handler(commands=["register"])
|
||
def register(message: Message):
|
||
username = message.from_user.username
|
||
if username == "":
|
||
bot.send_message(message.chat.id, "Добавть имя пользователя к своему телеграм аккаунту")
|
||
return
|
||
ui = UserInfo.objects.filter(telegram_chat_id=message.chat.id).first()
|
||
if ui:
|
||
bot.send_message(message.chat.id, "Ты уже зарегистрировался")
|
||
return
|
||
user = User.objects.create(username=username)
|
||
UserInfo.objects.create(user=user, telegram_chat_id=message.chat.id)
|
||
bot.send_message(message.chat.id, 'Приветствую в Sprint! Сейчас я помогу тебе создать аккаунт.\nДля начала отправь мне свою фамилию')
|
||
|
||
|
||
@bot.message_handler(content_types=["text"])
|
||
def do_action(message: Message):
|
||
user = User.objects.filter(userinfo__telegram_chat_id=message.chat.id).first()
|
||
if not user:
|
||
bot.send_message(message.chat.id, "Зарегистрируйся в сервисе, чтобы взаимодействовать со мной")
|
||
return
|
||
if user.userinfo.surname is None:
|
||
user.userinfo.surname = message.text
|
||
user.userinfo.save()
|
||
bot.send_message(message.chat.id, "Отлично! Теперь отправь мне свое имя")
|
||
elif user.userinfo.name is None:
|
||
user.userinfo.name = message.text
|
||
user.userinfo.save()
|
||
bot.send_message(message.chat.id, f"Регистрация завершена! Теперь можешь ты можешь войти в сервис под именем пользователя: {user.username}")
|
||
else:
|
||
bot.send_message(message.chat.id, "Я пока больше ничего не умею")
|
||
|
||
|
||
class Command(BaseCommand):
|
||
help = 'starts bot'
|
||
|
||
def handle(self, *args, **options):
|
||
print('bot is starting')
|
||
bot.polling()
|
||
print('bot failed')
|