Django - Підручник 039. Додавання приватних повідомлень і чатів на сайт - Частина 2 (лічильник діалогів і чатів з непрочитаними повідомленнями)

django, сообщение, Python, чат

Видався вільний час, щоб скоригувати особисті повідомлення на сайті. Даний функціонал використовується не особливо часто, тому не докладаю великих зусиль для його поліпшення, хоча час вже привести цей функціонал до адекватної роботи.

Раніше була дуже велика недоробка, яка полягала в тому, що не показувався лічильник діалогів з непрочитаними повідомленнями, що призводило до того, що користувачів, котрому надсилали повідомлення, просто не звертав на нього уваги, оскільки не знав про це повідомлень.

Тепер я нарешті виправив цей недолік. І в рамках попереднього коду покажу, які були додані виправлення.

Я розмірковував над двома варіантами організації лічильників непрочитаних повідомлень. Точніше одним варіантів та її більш просунутої версією.

• При кожному запиті перевіряти всі чати, вибирати з них останні повідомлення та перевіряти, чи є автором авторизований користувач, для якого потрібно перевірити це повідомлення. Якщо він не є автором, то перевіряємо, чи прочитане це повідомлення, якщо ні, то цей діалог для даного користувача вважаємо непрочитаним. Кількість таких діалогів вважатимемо кількістю непрочитаних діалогів.
• А другий варіант, на якому я зупинився, передбачає ту ж саму логіку, тільки діалог або чат повинен мати зовнішній ключ на останнє повідомлення, яке було в нього додано. Цей ключ оновлюватиметься під час кожного нового повідомлення. Тоді відпадає необхідність вибірки повідомлень по чату та їх сортування, щоб отримати останнє повідомлення. Що, на мою думку, може вилитися у великі накладні витрати для бази даних, якщо діалогів буде дуже багато.

Реалізація

models.py

Додамо зовнішній ключ на останнє повідомлення, а також кастомний менеджер для чатів ChatManager.

# -*- coding: utf-8 -*-

#... Код из предыдущей части


class ChatManager(models.Manager):
    use_for_related_fields = True

    # Метод принимает пользователя, для которого должна производиться выборка
    # Если пользователь не добавлен, то будет возвращены все диалоги,
    # в которых хотя бы одно сообщение не прочитано
    def unreaded(self, user=None):
        qs = self.get_queryset().exclude(last_message__isnull=True).filter(last_message__is_readed=False)
        return qs.exclude(last_message__author=user) if user else qs


class Chat(models.Model):
    #... Код из предыдущей части

    # внешний ключ на последнее сообщение,
    # важный момент в том, что название класса Message пишем обычной строкой,
    # поскольку на момент чтения класса Chat интерпретатор Python ничего не знает о классе Message
    # Также необходимо добавить related_name, имя через которое будет ассоциироваться выборка данного сообщения из базы данных
    last_message = models.ForeignKey('Message', related_name='last_message', null=True, blank=True, on_delete=models.SET_NULL)

    objects = ChatManager()

    @models.permalink
    def get_absolute_url(self):
        return 'users:messages', (), {'chat_id': self.pk }


class Message(models.Model):
    #... Код из предыдущей части

receivers.py

Я вперше пишу про такий файл Python у проекті на Django. Суть його в тому, що там буде оголошено обробників сигналів від моделі. Справа в тому, що Django при збереженні об'єкта моделі випускаються деякі сигнали, які можна обробляти. Це дозволяє деяку однакову логіку помістити в окремий файл і зробити інший код проекту дещо чистіше. Мінусом і те, що цей код може бути неочевидним, оскільки жодних посилань на цей код в інших частинах проекту не буде.

# -*- coding: utf-8 -*-

from django.db.models.signals import post_save
from django.dispatch import receiver

from users.models import Message


# обработчик сохранения объекта сообщения
@receiver(post_save, sender=Message)
def post_save_comment(sender, instance, created, **kwargs):
    # если объект был создан
    if created:
        # указываем чату, в котором находится данное сообщение, что это последнее сообщение
        instance.chat.last_message = instance
        # и обновляем данный внешний ключ чата
        instance.chat.save(update_fields=['last_message'])

Але просто так цей код не почне працювати, оскільки цей файл теж потрібно завантажити в інтерпретатор.

Зробити це можна у файлі apps.py при ініціалізації програми

apps.py

Робиться це у методі ready

# -*- coding: utf-8 -*-

from django.apps import AppConfig
from django.utils.translation import ugettext_lazy as _


class UsersConfig(AppConfig):
    name = 'users'
    verbose_name = _('Users')

    def ready(self):
        import users.receivers

Застосування

Тепер, коли у вас є все необхідне для отримання кількості діалогів з непрочитаними повідомленнями, ви можете додати цю інформацію в контекст для візуалізації шаблону.

context['unreaded_dialogs_counter'] = user.chat_set.unreaded(user=user).count()

Як виправити старі діалоги

Залишилося лише виправити старі діалоги, у яких вже є повідомлення. Зробити це можна буде через адмінку, якщо додати відповідну дію.

admin.py

# -*- coding: utf-8 -*-

from django.contrib import admin

from users import models


class ChatAdmin(admin.ModelAdmin):
    autocomplete_fields = ['members']
    search_fields = ('members',)
    actions = ['fix_last_messages']

    def fix_last_messages(self, request, queryset):
        for chat in queryset.all():
            chat.last_message = chat.message_set.all().order_by('-pub_date').first()
            chat.save(update_fields=['last_message'])

    fix_last_messages.short_description = "Fix last messages"


class MessageAdmin(admin.ModelAdmin):
    autocomplete_fields = ['chat', 'author']
    list_display = ('chat', 'author', 'message', 'pub_date', 'is_readed')


admin.site.register(models.Chat, ChatAdmin)
admin.site.register(models.Message, MessageAdmin)