chore: база

src/Telegram/models.py

@@ -1,31 +1,129 @@
 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]}'