130 lines
5.0 KiB
Python
130 lines
5.0 KiB
Python
from django.db import models
|
|
|
|
from . import interfaces
|
|
from .api import make_api
|
|
|
|
|
|
class Bot(models.Model):
|
|
"""
|
|
Пользователь заводит ботов указывая токен. Имя бота можно получить из API.
|
|
Из админки нужен интерфейс публикации и отключения бота, а также отображение актуального состояния.
|
|
"""
|
|
|
|
token = models.CharField(max_length=128, unique=True)
|
|
username = models.CharField(max_length=64, blank=True, editable=False)
|
|
name = models.CharField(max_length=128, blank=True, editable=False)
|
|
interface = models.CharField(
|
|
max_length=255, blank=True,
|
|
help_text='Класс, определяющий поведение бота. Меняется на лету.',
|
|
)
|
|
is_active = models.BooleanField('опубликован', default=False, editable=False)
|
|
update_offset = models.BigIntegerField(default=0, editable=False)
|
|
heartbeat_at = models.DateTimeField(null=True, blank=True, editable=False)
|
|
last_error = models.TextField(blank=True, editable=False)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
verbose_name = 'бот'
|
|
verbose_name_plural = 'боты'
|
|
|
|
def __str__(self):
|
|
return f'@{self.username}' if self.username else f'bot #{self.pk}'
|
|
|
|
def api(self):
|
|
return make_api(self.token)
|
|
|
|
def get_interface(self) -> interfaces.Interface | None:
|
|
cls = interfaces.get(self.interface)
|
|
return cls() if cls else None
|
|
|
|
def publish(self):
|
|
"""Проверяет токен, подтягивает имя бота и включает его обслуживание раннером."""
|
|
api = self.api()
|
|
me = api.get_me()
|
|
api.delete_webhook()
|
|
self.username = me.username or ''
|
|
self.name = me.first_name or ''
|
|
self.is_active = True
|
|
self.last_error = ''
|
|
self.save()
|
|
|
|
def revoke(self):
|
|
self.is_active = False
|
|
self.save(update_fields=['is_active'])
|
|
|
|
|
|
class User(models.Model):
|
|
"""
|
|
Когда клиент из тг пишет боту личное сообщение, то пользователя нужно завести в базу.
|
|
"""
|
|
|
|
tg_id = models.BigIntegerField(unique=True)
|
|
username = models.CharField(max_length=64, blank=True)
|
|
first_name = models.CharField(max_length=128, blank=True)
|
|
last_name = models.CharField(max_length=128, blank=True)
|
|
language_code = models.CharField(max_length=16, blank=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
verbose_name = 'пользователь telegram'
|
|
verbose_name_plural = 'пользователи telegram'
|
|
|
|
def __str__(self):
|
|
return f'@{self.username}' if self.username else str(self.tg_id)
|
|
|
|
|
|
class Chat(models.Model):
|
|
"""
|
|
При добавлении бота в группу, нужно добавить запись об этом в базу.
|
|
При удалении, нужно пометить запись как удаленную, но не удалять.
|
|
|
|
Чат принадлежит боту: у одного пользователя с разными ботами разные чаты и разное состояние диалога.
|
|
"""
|
|
|
|
bot = models.ForeignKey(Bot, on_delete=models.CASCADE, related_name='chats')
|
|
chat_id = models.BigIntegerField()
|
|
type = models.CharField(max_length=16)
|
|
title = models.CharField(max_length=255, blank=True)
|
|
user = models.ForeignKey(
|
|
User, on_delete=models.SET_NULL, null=True, blank=True, related_name='chats',
|
|
help_text='Собеседник в личном чате',
|
|
)
|
|
state = models.CharField(max_length=128, blank=True, help_text='Состояние диалога, задается интерфейсом бота')
|
|
data = models.JSONField(default=dict, blank=True, help_text='Данные диалога, задаются интерфейсом бота')
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
deleted_at = models.DateTimeField(null=True, blank=True)
|
|
|
|
class Meta:
|
|
verbose_name = 'чат'
|
|
verbose_name_plural = 'чаты'
|
|
constraints = [models.UniqueConstraint(fields=['bot', 'chat_id'], name='telegram_chat_unique')]
|
|
|
|
def __str__(self):
|
|
return self.title or str(self.user or self.chat_id)
|
|
|
|
|
|
class Message(models.Model):
|
|
"""
|
|
Исходящие и входящие сообщения между администратором и пользователем.
|
|
"""
|
|
|
|
class Direction(models.TextChoices):
|
|
IN = 'in', 'входящее'
|
|
OUT = 'out', 'исходящее'
|
|
|
|
chat = models.ForeignKey(Chat, on_delete=models.CASCADE, related_name='messages')
|
|
user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True, related_name='messages')
|
|
direction = models.CharField(max_length=3, choices=Direction.choices)
|
|
message_id = models.BigIntegerField(null=True, blank=True)
|
|
text = models.TextField(blank=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
verbose_name = 'сообщение'
|
|
verbose_name_plural = 'сообщения'
|
|
ordering = ['-created_at']
|
|
|
|
def __str__(self):
|
|
return f'{self.get_direction_display()}: {self.text[:50]}'
|