210 lines
12 KiB
Python
210 lines
12 KiB
Python
import logging
|
||
import json
|
||
import requests
|
||
import os
|
||
import base64
|
||
import re
|
||
from comunication.models import TelegramSettings
|
||
from web.models import ServiceRequest, Order, Project, Client, User
|
||
import telebot
|
||
from django.shortcuts import get_object_or_404
|
||
from django.utils.crypto import get_random_string
|
||
|
||
class TelegramBot:
|
||
def __init__(self):
|
||
# Получение настроек бота из базы данных
|
||
bot_settings = TelegramSettings.objects.first()
|
||
if bot_settings:
|
||
TELEGRAM_BOT_TOKEN = bot_settings.bot_token
|
||
self.bot = telebot.TeleBot(TELEGRAM_BOT_TOKEN)
|
||
logging.info("[TelegramBot] Бот инициализирован с токеном.")
|
||
else:
|
||
raise Exception("Telegram bot settings not found")
|
||
|
||
def start_bot_polling(self):
|
||
logging.info("[TelegramBot] Бот начал работу в режиме polling.")
|
||
|
||
@self.bot.message_handler(commands=['start'])
|
||
def send_welcome(message):
|
||
# Проверяем, содержатся ли параметры в команде /start
|
||
match = re.match(r'/start request_(\d+)_token_(.*)', message.text)
|
||
logging.info(f"[TelegramBot] Получена команда /start: {message.text}")
|
||
|
||
if match:
|
||
self.handle_confirm_command(message, match)
|
||
elif message.text.strip() == '/start':
|
||
# Ответ на просто команду /start без параметров
|
||
self.bot.reply_to(message, "Здравствуйте! Пожалуйста, используйте команду /start с корректными параметрами для подтверждения регистрации.")
|
||
else:
|
||
self.bot.reply_to(message, "Здравствуйте! Пожалуйста, используйте команду /start с корректными параметрами для подтверждения регистрации.")
|
||
|
||
@self.bot.message_handler(func=lambda message: 'статус заявки' in message.text.lower())
|
||
def handle_service_request_status(message):
|
||
chat_id = message.chat.id
|
||
client = Client.objects.filter(chat_id=chat_id).first()
|
||
if client:
|
||
service_requests = ServiceRequest.objects.filter(client=client)
|
||
if service_requests.exists():
|
||
response = "Ваши заявки:\n"
|
||
for req in service_requests:
|
||
response += (
|
||
f"Номер заявки: {req.id}\n"
|
||
f"Услуга: {req.service.name}\n"
|
||
f"Дата создания: {req.created_at.strftime('%Y-%m-%d')}\n"
|
||
f"UID заявки: {req.token}\n"
|
||
f"Подтверждена: {'Да' if req.is_verified else 'Нет'}\n\n"
|
||
)
|
||
else:
|
||
response = "У вас нет активных заявок."
|
||
else:
|
||
response = "Клиент не найден. Пожалуйста, зарегистрируйтесь."
|
||
self.bot.reply_to(message, response)
|
||
|
||
@self.bot.message_handler(func=lambda message: 'статус заказа' in message.text.lower())
|
||
def handle_order_status(message):
|
||
chat_id = message.chat.id
|
||
client = Client.objects.filter(chat_id=chat_id).first()
|
||
if client:
|
||
orders = Order.objects.filter(client=client)
|
||
if orders.exists():
|
||
response = "Ваши заказы:\n"
|
||
for order in orders:
|
||
response += (
|
||
f"Номер заказа: {order.id}\n"
|
||
f"Услуга: {order.service.name}\n"
|
||
f"Статус: {order.get_status_display()}\n\n"
|
||
)
|
||
else:
|
||
response = "У вас нет активных заказов."
|
||
else:
|
||
response = "Клиент не найден. Пожалуйста, зарегистрируйтесь."
|
||
self.bot.reply_to(message, response)
|
||
|
||
@self.bot.message_handler(func=lambda message: 'статус проекта' in message.text.lower())
|
||
def handle_project_status(message):
|
||
chat_id = message.chat.id
|
||
client = Client.objects.filter(chat_id=chat_id).first()
|
||
if client:
|
||
projects = Project.objects.filter(order__client=client)
|
||
if projects.exists():
|
||
response = "Ваши проекты:\n"
|
||
for project in projects:
|
||
response += (
|
||
f"Номер проекта: {project.id}\n"
|
||
f"Название проекта: {project.name}\n"
|
||
f"Статус: {project.get_status_display()}\n"
|
||
f"Дата завершения: {project.completion_date.strftime('%Y-%m-%d') if project.completion_date else 'В процессе'}\n\n"
|
||
)
|
||
else:
|
||
response = "У вас нет активных проектов."
|
||
else:
|
||
response = "Клиент не найден. Пожалуйста, зарегистрируйтесь."
|
||
self.bot.reply_to(message, response)
|
||
|
||
|
||
# Запуск бота
|
||
try:
|
||
self.bot.polling(non_stop=True)
|
||
except Exception as e:
|
||
logging.error(f"[TelegramBot] Ошибка при запуске polling: {e}")
|
||
|
||
def handle_confirm_command(self, message, match=None):
|
||
chat_id = message.chat.id
|
||
logging.info(f"[TelegramBot] Получено сообщение для подтверждения: {message.text}")
|
||
|
||
if not match:
|
||
match = re.match(r'/start request_(\d+)_token_(.*)', message.text)
|
||
|
||
if match:
|
||
request_id = match.group(1)
|
||
encoded_token = match.group(2)
|
||
|
||
# Декодируем токен
|
||
try:
|
||
token = base64.urlsafe_b64decode(encoded_token + '==').decode('utf-8')
|
||
logging.info(f"[TelegramBot] Декодированный токен: {token}")
|
||
except Exception as e:
|
||
logging.error(f"[TelegramBot] Ошибка при декодировании токена: {e}")
|
||
self.bot.send_message(chat_id, "Ошибка: Некорректный токен. Пожалуйста, повторите попытку позже.")
|
||
return
|
||
|
||
# Получаем заявку
|
||
try:
|
||
service_request = ServiceRequest.objects.get(id=request_id, token=token)
|
||
logging.info(f"[TelegramBot] Заявка найдена: {service_request}")
|
||
except ServiceRequest.DoesNotExist:
|
||
logging.error(f"[TelegramBot] Заявка с id {request_id} и токеном {token} не найдена.")
|
||
self.bot.send_message(chat_id, "Ошибка: Неверная заявка или токен. Пожалуйста, проверьте ссылку.")
|
||
return
|
||
|
||
# Если заявка найдена, обновляем и подтверждаем клиента
|
||
if service_request:
|
||
service_request.chat_id = chat_id
|
||
service_request.is_verified = True # Обновляем статус на подтвержденный
|
||
service_request.save()
|
||
logging.info(f"[TelegramBot] Заявка {service_request.id} подтверждена и обновлена.")
|
||
|
||
# Обновляем или создаем клиента, связанного с заявкой
|
||
client = service_request.client
|
||
|
||
# Проверяем, существует ли связанный пользователь, если нет — создаем его
|
||
if not client.user:
|
||
user, created = User.objects.get_or_create(
|
||
email=client.email,
|
||
defaults={
|
||
'username': f"{client.email.split('@')[0]}_{get_random_string(5)}",
|
||
'first_name': message.from_user.first_name,
|
||
'last_name': message.from_user.last_name if message.from_user.last_name else ''
|
||
}
|
||
)
|
||
|
||
if not created:
|
||
# Если пользователь уже существовал, обновляем его данные
|
||
user.first_name = message.from_user.first_name
|
||
if message.from_user.last_name:
|
||
user.last_name = message.from_user.last_name
|
||
user.save()
|
||
logging.info(f"[TelegramBot] Обновлен пользователь {user.username} с данными из Телеграм.")
|
||
|
||
# Связываем клиента с пользователем
|
||
client.user = user
|
||
client.save()
|
||
|
||
else:
|
||
# Обновляем данные существующего пользователя
|
||
user = client.user
|
||
user.first_name = message.from_user.first_name
|
||
if message.from_user.last_name:
|
||
user.last_name = message.from_user.last_name
|
||
user.save()
|
||
logging.info(f"[TelegramBot] Пользователь {user.username} обновлен с данными из Телеграм.")
|
||
|
||
# Обновляем chat_id клиента
|
||
client.chat_id = chat_id
|
||
client.save()
|
||
logging.info(f"[TelegramBot] Клиент {client.id} обновлен с chat_id {chat_id}")
|
||
|
||
# Отправляем сообщение пользователю в Telegram с подтверждением и информацией о заявке
|
||
confirmation_message = (
|
||
f"Здравствуйте, {client.first_name}!\n\n"
|
||
f"Ваш аккаунт успешно подтвержден! 🎉\n\n"
|
||
f"Детали вашей заявки:\n"
|
||
f"Номер заявки: {service_request.id}\n"
|
||
f"Услуга: {service_request.service.name}\n"
|
||
f"Статус заявки: {'Подтверждена' if service_request.is_verified else 'Не подтверждена'}\n"
|
||
f"Дата создания: {service_request.created_at.strftime('%Y-%m-%d %H:%M:%S')}\n"
|
||
f"Спасибо, что выбрали наши услуги! Если у вас возникнут вопросы, вы всегда можете обратиться к нам."
|
||
)
|
||
self.bot.send_message(chat_id, confirmation_message)
|
||
|
||
# Вместо дополнительного POST-запроса — сообщаем о подтверждении через сообщение
|
||
self.bot.send_message(chat_id, "Ваш аккаунт успешно подтвержден на сервере! Продолжайте на сайте.")
|
||
|
||
else:
|
||
self.bot.send_message(chat_id, "Ошибка: Неверная заявка или токен. Пожалуйста, проверьте ссылку.")
|
||
else:
|
||
response_message = "Ошибка: Некорректная команда. Пожалуйста, используйте ссылку, предоставленную на сайте для регистрации."
|
||
self.bot.send_message(chat_id, response_message)
|
||
|
||
|