init commit

smartsoltech/comunication/admin.py

@@ -0,0 +1,16 @@
+# communication/admin.py
+from django.contrib import admin
+from .models import EmailSettings, TelegramSettings, UserCommunication
+
+@admin.register(EmailSettings)
+class EmailSettingsAdmin(admin.ModelAdmin):
+    list_display = ('smtp_server', 'sender_email', 'use_tls', 'use_ssl')
+
+@admin.register(TelegramSettings)
+class TelegramSettingsAdmin(admin.ModelAdmin):
+    list_display = ('bot_name', 'bot_token', 'use_polling')
+    
+@admin.register(UserCommunication)
+class UserCommunicationAdmin(admin.ModelAdmin):
+    list_display = ('client', 'email', 'phone', 'chat_id')
+    

smartsoltech/comunication/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ComunicationConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'comunication'

smartsoltech/comunication/management/commands/start_telegram_bot.py

@@ -0,0 +1,11 @@
+# comunication/management/commands/start_telegram_bot.py
+from django.core.management.base import BaseCommand
+from comunication.telegram_bot import TelegramBot
+
+class Command(BaseCommand):
+    help = 'Starts the Telegram bot'
+
+    def handle(self, *args, **kwargs):
+        bot = TelegramBot()
+        self.stdout.write('Starting Telegram bot polling...')
+        bot.start_bot_polling()

smartsoltech/comunication/migrations/0001_initial.py

@@ -0,0 +1,36 @@
+# Generated by Django 5.1.1 on 2024-10-08 12:20
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='CommunicationMethod',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(choices=[('email', 'Email'), ('telegram', 'Telegram')], max_length=50)),
+                ('settings', models.JSONField()),
+            ],
+        ),
+        migrations.CreateModel(
+            name='UserCommunication',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('chat_id', models.CharField(blank=True, max_length=100, null=True)),
+                ('email', models.EmailField(blank=True, max_length=254, null=True)),
+                ('phone', models.CharField(blank=True, max_length=20, null=True)),
+                ('preferred_method', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to='comunication.communicationmethod')),
+                ('user', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+    ]

smartsoltech/comunication/migrations/0002_emailsettings_telegramsettings.py

@@ -0,0 +1,36 @@
+# Generated by Django 5.1.1 on 2024-10-08 12:40
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('comunication', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='EmailSettings',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('smtp_server', models.CharField(max_length=255)),
+                ('smtp_port', models.PositiveIntegerField()),
+                ('sender_email', models.EmailField(max_length=254)),
+                ('password', models.CharField(max_length=255)),
+                ('use_tls', models.BooleanField(default=True)),
+                ('use_ssl', models.BooleanField(default=False)),
+                ('display_name', models.CharField(blank=True, max_length=255, null=True)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='TelegramSettings',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('bot_name', models.CharField(max_length=100)),
+                ('bot_token', models.CharField(max_length=255)),
+                ('webhook_url', models.URLField(blank=True, null=True)),
+                ('use_polling', models.BooleanField(default=True)),
+            ],
+        ),
+    ]

smartsoltech/comunication/migrations/0003_remove_usercommunication_preferred_method_and_more.py

@@ -0,0 +1,27 @@
+# Generated by Django 5.1.1 on 2024-10-08 12:49
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('comunication', '0002_emailsettings_telegramsettings'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='usercommunication',
+            name='preferred_method',
+        ),
+        migrations.RemoveField(
+            model_name='usercommunication',
+            name='user',
+        ),
+        migrations.DeleteModel(
+            name='CommunicationMethod',
+        ),
+        migrations.DeleteModel(
+            name='UserCommunication',
+        ),
+    ]

smartsoltech/comunication/migrations/0004_usercommunication.py

@@ -0,0 +1,22 @@
+# Generated by Django 5.1.1 on 2024-10-13 00:17
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('comunication', '0003_remove_usercommunication_preferred_method_and_more'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='UserCommunication',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('email', models.EmailField(max_length=254)),
+                ('phone', models.CharField(blank=True, max_length=15)),
+                ('chat_id', models.CharField(blank=True, max_length=50)),
+            ],
+        ),
+    ]

smartsoltech/comunication/migrations/0005_alter_emailsettings_options_and_more.py

@@ -0,0 +1,47 @@
+# Generated by Django 5.1.1 on 2024-10-13 04:18
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('comunication', '0004_usercommunication'),
+        ('web', '0005_alter_blogpost_options_alter_category_options_and_more'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='emailsettings',
+            options={'ordering': ['-display_name'], 'verbose_name': 'Параметры E-mail', 'verbose_name_plural': 'Параметры E-mail'},
+        ),
+        migrations.AlterModelOptions(
+            name='telegramsettings',
+            options={'ordering': ['-bot_name'], 'verbose_name': 'Параметры Telegram бота', 'verbose_name_plural': 'Параметры Telegram ботов'},
+        ),
+        migrations.AlterModelOptions(
+            name='usercommunication',
+            options={'ordering': ['-id'], 'verbose_name': 'Связь с клиентом', 'verbose_name_plural': 'Связи с клиентами'},
+        ),
+        migrations.AddField(
+            model_name='usercommunication',
+            name='client',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, related_name='communications', to='web.client', verbose_name='Клиент'),
+        ),
+        migrations.AlterField(
+            model_name='usercommunication',
+            name='chat_id',
+            field=models.CharField(blank=True, max_length=50, verbose_name='Telegram Chat ID'),
+        ),
+        migrations.AlterField(
+            model_name='usercommunication',
+            name='email',
+            field=models.EmailField(max_length=254, verbose_name='Электронная почта'),
+        ),
+        migrations.AlterField(
+            model_name='usercommunication',
+            name='phone',
+            field=models.CharField(blank=True, max_length=15, verbose_name='Телефон'),
+        ),
+    ]

smartsoltech/comunication/migrations/0006_alter_usercommunication_client.py

@@ -0,0 +1,20 @@
+# Generated by Django 5.1.1 on 2024-10-13 04:20
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('comunication', '0005_alter_emailsettings_options_and_more'),
+        ('web', '0005_alter_blogpost_options_alter_category_options_and_more'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='usercommunication',
+            name='client',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='communications', to='web.client', verbose_name='Клиент'),
+        ),
+    ]

smartsoltech/comunication/models.py

@@ -0,0 +1,50 @@
+# communication/models.py
+from django.db import models
+from django.contrib.auth.models import User
+from web.models import Client
+    
+class EmailSettings(models.Model):
+    smtp_server = models.CharField(max_length=255)
+    smtp_port = models.PositiveIntegerField()
+    sender_email = models.EmailField()
+    password = models.CharField(max_length=255)
+    use_tls = models.BooleanField(default=True)
+    use_ssl = models.BooleanField(default=False)
+    display_name = models.CharField(max_length=255, null=True, blank=True)
+
+    def __str__(self):
+        return f"SMTP: {self.smtp_server}, Email: {self.sender_email}"
+    class Meta:
+        verbose_name = 'Параметры E-mail'
+        verbose_name_plural = 'Параметры E-mail'
+        ordering = ['-display_name']
+        
+class TelegramSettings(models.Model):
+    bot_name = models.CharField(max_length=100)
+    bot_token = models.CharField(max_length=255)
+    webhook_url = models.URLField(null=True, blank=True)
+    use_polling = models.BooleanField(default=True)
+
+    def __str__(self):
+        return f"Telegram Bot: {self.bot_name}"
+    class Meta:
+        verbose_name = 'Параметры Telegram бота'
+        verbose_name_plural = 'Параметры Telegram ботов'
+        ordering = ['-bot_name']
+class UserCommunication(models.Model):
+    client = models.ForeignKey(
+        'web.Client', on_delete=models.CASCADE, related_name='communications', verbose_name='Клиент', null=True, blank=True
+    )
+    email = models.EmailField(verbose_name='Электронная почта')
+    phone = models.CharField(max_length=15, blank=True, verbose_name='Телефон')
+    chat_id = models.CharField(max_length=50, blank=True, verbose_name='Telegram Chat ID')
+
+    class Meta:
+        verbose_name = 'Связь с клиентом'
+        verbose_name_plural = 'Связи с клиентами'
+        ordering = ['-id']
+
+    def __str__(self):
+        if self.client:
+            return f"Связь с клиентом: {self.client.first_name} {self.client.last_name} ({self.email})"
+        return f"Связь без клиента ({self.email})"

smartsoltech/comunication/telegram_bot.py

@@ -0,0 +1,232 @@
+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 and bot_settings.bot_token:
+            TELEGRAM_BOT_TOKEN = bot_settings.bot_token.strip()
+            
+            # Проверяем валидность токена
+            if self._validate_token(TELEGRAM_BOT_TOKEN):
+                self.bot = telebot.TeleBot(TELEGRAM_BOT_TOKEN)
+                logging.info(f"[TelegramBot] Бот инициализирован с токеном для {bot_settings.bot_name}.")
+            else:
+                logging.error(f"[TelegramBot] Токен невалиден: {TELEGRAM_BOT_TOKEN[:10]}...")
+                raise Exception(f"Невалидный токен Telegram бота. Обновите токен в базе данных.")
+        else:
+            raise Exception("Telegram bot settings not found or token is empty")
+
+    def _validate_token(self, token):
+        """Проверяет валидность токена через Telegram API"""
+        url = f"https://api.telegram.org/bot{token}/getMe"
+        try:
+            response = requests.get(url, timeout=10)
+            result = response.json()
+            if result.get('ok'):
+                bot_info = result.get('result', {})
+                logging.info(f"[TelegramBot] Токен валиден. Бот: @{bot_info.get('username', 'unknown')}")
+                return True
+            else:
+                logging.error(f"[TelegramBot] Ошибка Telegram API: {result.get('description', 'Unknown error')}")
+                return False
+        except requests.RequestException as e:
+            logging.error(f"[TelegramBot] Ошибка при проверке токена: {e}")
+            return False
+
+    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)
+
+

smartsoltech/comunication/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

smartsoltech/comunication/views.py

@@ -0,0 +1,3 @@
+from django.shortcuts import render
+
+# Create your views here.