main funtions are ready

issue: ServiceRequest creates when QR code scanned

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,256 @@
+# import telebot
+# from decouple import config
+# from django.shortcuts import get_object_or_404
+# from web.models import Client, ServiceRequest, Order
+# from comunication.models import TelegramSettings
+# import re
+# import base64
+# import logging
+
+# class TelegramBot:
+#     def __init__(self):
+#         # Get bot settings from the database
+#         bot_settings = TelegramSettings.objects.first()
+#         if bot_settings:
+#             TELEGRAM_BOT_TOKEN = bot_settings.bot_token
+#             self.bot = telebot.TeleBot(TELEGRAM_BOT_TOKEN)
+#         else:
+#             raise Exception("Telegram bot settings not found")
+
+#     def start_bot_polling(self):
+#         @self.bot.message_handler(commands=['start'])
+#         def send_welcome(message):
+#             # Проверяем, содержатся ли параметры в команде /start
+#             match = re.match(r'/start request_(\d+)_token_(.*)', message.text)
+#             if match:
+#                 self.handle_confirm_command(message, match)
+#             elif message.text.strip() == '/start':
+#                 self.bot.reply_to(message, "Ошибка: Некорректная команда. Пожалуйста, используйте ссылку, предоставленную на сайте для регистрации.")
+#             else:
+#                 self.bot.reply_to(message, "Здравствуйте! Чем я могу помочь? Вы можете задать вопросы о статусе заявки или заказе.")
+
+#         @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_email=client.email)
+#                 if service_requests.exists():
+#                     response = "Ваши заявки:\n"
+#                     for req in service_requests:
+#                         response += f"Номер заявки: {req.id}, Услуга: {req.service.name}, Дата создания: {req.created_at.strftime('%d-%m-%Y')}\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}, Услуга: {order.service.name}, Статус: {order.get_status_display()}\n"
+#                 else:
+#                     response = "У вас нет активных заказов."
+#             else:
+#                 response = "Клиент не найден. Пожалуйста, зарегистрируйтесь."
+#             self.bot.reply_to(message, response)
+
+#         self.bot.polling(non_stop=True)
+
+#     def handle_confirm_command(self, message, match=None):
+#         chat_id = message.chat.id
+#         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)
+
+#             # Декодируем токен из base64
+#             try:
+#                 token = base64.urlsafe_b64decode(encoded_token + '==').decode('utf-8')
+#                 logging.info(f"Декодированный токен: {token}")
+#             except Exception as e:
+#                 logging.error(f"Ошибка при декодировании токена: {e}")
+#                 self.bot.send_message(chat_id, "Ошибка: Некорректный токен. Пожалуйста, повторите попытку позже.")
+#                 return
+
+#             # Получаем заявку по ID и токену
+#             service_request = ServiceRequest.objects.filter(id=request_id, token=token).first()
+#             if service_request:
+#                 # Обновляем chat_id клиента
+#                 service_request.chat_id = chat_id
+#                 service_request.client_name = message.from_user.first_name
+#                 service_request.save()
+
+#                 response_message = (
+#                     f"Здравствуйте, {message.from_user.first_name}!\n"
+#                     f"Ваша заявка на услугу успешно зарегистрирована. "
+#                     f"Пожалуйста, вернитесь на сайт для продолжения оформления."
+#                 )
+#             else:
+#                 response_message = "Ошибка: Неверная заявка или токен. Пожалуйста, проверьте ссылку."
+
+#             self.bot.send_message(chat_id, response_message)
+#         else:
+#             response_message = "Ошибка: Некорректная команда. Пожалуйста, используйте ссылку, предоставленную на сайте для регистрации."
+#             self.bot.send_message(chat_id, response_message)
+
+#     def send_telegram_message(self, client_id, service_request_id, custom_message, order_id=None):
+#         # Get the client and service request from the database
+#         client = get_object_or_404(Client, pk=client_id)
+#         service_request = get_object_or_404(ServiceRequest, pk=service_request_id)
+#         chat_id = client.chat_id
+
+#         # Build the message content
+#         message = f"Здравствуйте, {client.first_name} {client.last_name}!\n"
+#         message += custom_message
+
+#         if order_id:
+#             order = get_object_or_404(Order, pk=order_id)
+#             message += f"\n\nДетали заказа:\nУслуга: {order.service.name}\nСтатус: {order.get_status_display()}\n"
+        
+#         # Add service request details
+#         message += f"\nНомер заявки: {service_request.id}\nДата создания заявки: {service_request.created_at.strftime('%d-%m-%Y')}\n"
+
+#         # Send the message using the bot
+#         try:
+#             self.bot.send_message(chat_id, message)
+#         except Exception as e:
+#             logging.error(f"Ошибка при отправке сообщения в Telegram: {e}")
+
+
+import telebot
+from decouple import config
+from django.shortcuts import get_object_or_404
+from web.models import Client, ServiceRequest, Order
+from comunication.models import TelegramSettings
+import re
+import base64
+import logging
+
+class TelegramBot:
+    def __init__(self):
+        # Get bot settings from the database
+        bot_settings = TelegramSettings.objects.first()
+        if bot_settings:
+            TELEGRAM_BOT_TOKEN = bot_settings.bot_token
+            self.bot = telebot.TeleBot(TELEGRAM_BOT_TOKEN)
+        else:
+            raise Exception("Telegram bot settings not found")
+
+    def start_bot_polling(self):
+        @self.bot.message_handler(commands=['start'])
+        def send_welcome(message):
+            # Проверяем, содержатся ли параметры в команде /start
+            match = re.match(r'/start request_(\d+)_token_(.*)', message.text)
+            if match:
+                self.handle_confirm_command(message, match)
+            elif message.text.strip() == '/start':
+                self.bot.reply_to(message, "Ошибка: Некорректная команда. Пожалуйста, используйте ссылку, предоставленную на сайте для регистрации.")
+            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_email=client.email)
+                if service_requests.exists():
+                    response = "Ваши заявки:\n"
+                    for req in service_requests:
+                        response += f"Номер заявки: {req.id}, Услуга: {req.service.name}, Дата создания: {req.created_at.strftime('%d-%m-%Y')}\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}, Услуга: {order.service.name}, Статус: {order.get_status_display()}\n"
+                else:
+                    response = "У вас нет активных заказов."
+            else:
+                response = "Клиент не найден. Пожалуйста, зарегистрируйтесь."
+            self.bot.reply_to(message, response)
+
+        self.bot.polling(non_stop=True)
+
+    def handle_confirm_command(self, message, match=None):
+        chat_id = message.chat.id
+        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)
+
+            # Декодируем токен из base64
+            try:
+                token = base64.urlsafe_b64decode(encoded_token + '==').decode('utf-8')
+                logging.info(f"Декодированный токен: {token}")
+            except Exception as e:
+                logging.error(f"Ошибка при декодировании токена: {e}")
+                self.bot.send_message(chat_id, "Ошибка: Некорректный токен. Пожалуйста, повторите попытку позже.")
+                return
+
+            # Получаем заявку по ID и токену
+            service_request = ServiceRequest.objects.filter(id=request_id, token=token).first()
+            if service_request:
+                # Обновляем chat_id клиента
+                service_request.chat_id = chat_id
+                service_request.client_name = message.from_user.first_name
+                service_request.save()
+
+                response_message = (
+                    f"Здравствуйте, {message.from_user.first_name}!\n"
+                    f"Ваш Telegram аккаунт успешно подтвержден. Пожалуйста, вернитесь на сайт для заполнения остальных данных."
+                )
+            else:
+                response_message = "Ошибка: Неверная заявка или токен. Пожалуйста, проверьте ссылку."
+
+            self.bot.send_message(chat_id, response_message)
+        else:
+            response_message = "Ошибка: Некорректная команда. Пожалуйста, используйте ссылку, предоставленную на сайте для регистрации."
+            self.bot.send_message(chat_id, response_message)
+
+    def send_telegram_message(self, client_id, service_request_id, custom_message, order_id=None):
+        # Get the client and service request from the database
+        client = get_object_or_404(Client, pk=client_id)
+        service_request = get_object_or_404(ServiceRequest, pk=service_request_id)
+        chat_id = client.chat_id
+
+        # Build the message content
+        message = f"Здравствуйте, {client.first_name} {client.last_name}!\n"
+        message += custom_message
+
+        if order_id:
+            order = get_object_or_404(Order, pk=order_id)
+            message += f"\n\nДетали заказа:\nУслуга: {order.service.name}\nСтатус: {order.get_status_display()}\n"
+        
+        # Add service request details
+        message += f"\nНомер заявки: {service_request.id}\nДата создания заявки: {service_request.created_at.strftime('%d-%m-%Y')}\n"
+
+        # Send the message using the bot
+        try:
+            self.bot.send_message(chat_id, message)
+        except Exception as e:
+            logging.error(f"Ошибка при отправке сообщения в Telegram: {e}")
+
+# Example usage:
+# bot = TelegramBot()
+# bot.start_bot_polling()
+# bot.send_telegram_message(client_id=1, service_request_id=1, custom_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.