Как получить inline модель в инстанс post_save (Django Signals)?

Question

[Vic Shóstak]

Привет всем!

Использую Django Signals (1.11.7) для некоторых действий над только что сохранённой моделью (отправка письма на почту с инфой из модели, в основном). Всё отлично, работает (ещё бы не работало :D ). Но только я добавляю к этой модели ещё одну, связанную (ForeignKey) с основной (как inlines=[...] в admin.py) — она, как будто бы, не участвует в сохранении инстанса основной модели.

То есть, для доступа к inline модели и какого-то действия при post_save сигнале — нужно редактировать только что созданную запись + ещё раз сохранять.

Вот мой код:

# /tours/models.py

class Tours(models.Model):
    country = models.CharField(...)
    ...


class ToursHotels(models.Model):
    tour = models.ForeignKey(Tours, ...)
    cost = models.IntegerField(...)
    ...


# Signal
@receiver(post_save, sender=Tours)
def do_something(sender, **kwargs):
    tour = Tours.objects.get(id=kwargs.get('instance').id)
    hotels = ToursHotels.objects.filter(tour_id=kwargs.get('instance').id).order_by('cost')
    ...

Так вот, hotels будет пустым, пока я снова не отредактирую эту запись..

И тут сразу же вопрос — как лучше сделать? Мне приходит в голову только: попробовать сделать через Celery задание, которое будет происходить уже после post_save (что логично, ведь вначале произойдёт post_save, а потом только task из очереди), когда и Tours и ToursHotels уже есть в БД. Но это ещё одна надстройка, ради которой нужно Redis/RMQ поднимать..

Может быть есть какие-либо best practice на эту тему? На SO встретил много вопросов без ответа на похожие темы с post_save, но все они 2010-2012 года и свежее нет — может быть уже всё поменялось, а я и не знаю? ;)

Буду рад толковым комментариям! Заранее спасибо.

Comments

[sim3x]

Мой бест практис - не использовать сигналы
Есть def save
Если логика сохранения требует редактировать несколько моделей, то деаем все в форме

[Vic Shóstak]

sim3x, а можно по-подробнее про это, если не сложно:

Если логика сохранения требует редактировать несколько моделей, то деаем все в форме

[sim3x]

Vic Shóstak,

Хм
У нас есть форма с кучей полей

В def save(self): суем из cleaned_data по тем моделям куда им положено

[Vic Shóstak]

sim3x, спасибо. Кстати, а почему использование Signals — это анти-паттерн в Django? Уже не первый раз встречаю такие высказывания..

[javedimka]

Vic Shóstak, https://lincolnloop.com/blog/django-anti-patterns-...

[sim3x]

Vic Shóstak, если прочитали тут - то скорее всего ето были мои высказывания

Сигналы можно всунуть в любом месте и их не сразу заметишь
Те нарушение простоты и прозрачности

[Vic Shóstak]

javedimka, sim3x, хм. спасибо за ссылку и объяснение ;)

Я так понимаю, что в моём случае проще всего делать через переопределение save() метода модели Tours? Если да, то вопрос: каким образом получить внутри Tours.save() все данные из инлайновой модели ToursHotels (для сохранённой записи)?

Ведь когда сохраняется модель Tours, то inline модель ToursHotels ещё не сохранена... или я что-то путаю?

[Vic Shóstak]

sim3x, javedimka, Pavel Denisov, перефразирую вопрос: можно ли в методе save() модели Tours определить вначале сохранение всей пачки ToursHotels, а уже потом сохранять основную? При этом, используя pk в инстансе модели Tours?

[javedimka]

Vic Shóstak, можешь объяснить подробнее какая цель и что вообще хочешь получить? Потому что, например, я - плохо понимаю что ты имеешь ввиду под «инлайновой» моделью, толи в админке инлайны сделаны, толи ещё что, в общем попонятнее бы. Просто добавить доступные отели для созданного тура?

[Vic Shóstak]

javedimka, Pavel Denisov, Ок. Поясню.

Есть модель Tours и ToursHotels (назовём её просто Hotels для лучшего понимания). Модель Tours определена в admin.py (и доступна для CRUD операций в админке), а модель Hotels числится в её inlines=[...] параметре.

Цель: при сохранении формы в админке — отправлять сохранённые данные из этой записи на почту администратору.

Выглядит всё вот так: https://ibb.co/i1O2Hb (иными словами, модель Hotels может быть очень много раз добавлена к модели Tours — поэтому и нужно именно при сохранении Tours выполнять отправку на почту, а не при Hotels — ведь на каждый тик Hotels — будет улетать отдельное письмо).

Состояние: переопределён метод save() для модели Tours.

Проблема: после сохранения новой записи — отправляются только данные из модели Tours, а всё то, что в Hotels — нет. Почему? Я так понял, потому что родительская модель Tours сохраняется раньше, чем все прикреплённые к ней по ForeignKey потомки.

В идеале: каким-то образом переопределить сохранение модели Tours, чтобы вначале сохранялись её потомки (модель Hotels), используя её pk, а только потом сам родитель (Tours). Далее — у нас уже есть в БД и Tours и Hotels — можно отправлять письмо админу с полными данными по этому заказу тура.

[javedimka]

Vic Shóstak, возможно save() криво переопределен, отправлять письмо надо после вызова super()
Джанго должна сама сохранить связи, надо только письмо отправить.

[Vic Shóstak]

javedimka, так отправка и вызывается после super():

def save(self, *args, **kwargs):
    # Tours model save
    super(Tours, self).save(*args, **kwargs)
    # Define checker of new object
    is_new_object = self.pk is None
    # Send mail
    self._send_mail_to_admin(is_new_object)

[javedimka]

Vic Shóstak, непонятно зачем «чекер», он всегда будет false потому что проверка идёт после super(), ну и код самого метода отправки тоже можно скинуть.

[Vic Shóstak]

javedimka, да, конечно. Вот полный код:

def _send_mail_to_admin(self, is_new_object=False):
    if is_new_object:

        tour = Tours.objects.get(id=self.id)
        hotels = ToursHotels.objects.filter(tour_id=tour.id).order_by('cost')

        message = 'Тур ID ' + tour.id + '\n'
        message += 'Страна: ' + tour.country_name + ' Город: ' + tour.country_name + '\n'
        message += 'Отели: \n'
        for hotel in hotels:
            message += hotel.name + ' ' + hotel.star + ' ' + hotel.cost + ' руб. \n'

        send_mail(
            'From Admin',
            message,
            '[e-mail]',
            ['[e-mail]'],
            fail_silently=False,
        )
        
def save(self, *args, **kwargs):
    # Define checker of new object
    is_new_object = self.pk is None
    # Tours model save
    super(Tours, self).save(*args, **kwargs)
    # Send mail
    self._send_mail_to_admin(is_new_object)

Всё то, что в tour — приходит на почту, а hotels — пустой при сохранении нового заказа в админке (только после повторного сохранения отправляется).

[javedimka]

Vic Shóstak,
Замечания по методу:

hotels = ToursHotels.objects.filter(tour_id=tour.id).order_by('cost')

это вообще какая-то грязь, у джанги есть backwards relations, если бы метод работал, то для того, чтобы получить связанные отели для тура можно было бы написать так:

hotels = self.tourshotels_set.all().order_by('cost')
# или если полю ForeignKey передан аргумент related_name:
# class ToursHotels(models.Model):
#    tour = models.ForeignKey(Tours, related_name='hotels')
# так:
hotels = self.hotels.all().order_by('cost')

По решению проблемы:
igorsobreira.com/2011/02/12/change-object-after-sa...
Статья старая, но рабочая, если вкратце:
Модель сохраняется после валидации самой модели и инлайнов, но до того как сохраняются инлайны, т.е. переписанный save() вызывается до сохранения связанных объектов, чтобы это исправить, следуя статье, нужно пепреписать метод response_add на ModelAdmin, заходим в доку и ищем что там написано:

response_add is called after the admin form is submitted and just after the object and all the related instances have been created and saved. You can override it to change the default behavior after the object has been created.

Подходит, получается что-то типо этого:

class HotelInline(admin.TabularInline):
    model = Hotel


class TourAdmin(admin.ModelAdmin):
    inlines = (HotelInline,)

    def response_add(self, request, new_object):
        print(new_object.pk)
        print(new_object.country)
        print(new_object.hotels.all())
        return super().response_add(request, new_object)


admin.site.register(Tour, TourAdmin)

[Vic Shóstak]

javedimka, спасибо за ссылку и подробное пояснение!
Исправил косячный метод на self.hotels.all().order_by('cost') (плюс ещё во многих местах эту грязь убрал).

Почитал доки и сделал вот так:

@admin.register(Tours)  # Register `ToursAdmin` as Admin model
class ToursAdmin(admin.ModelAdmin):
    exclude = ('created_at',)
    list_display = ('country',)
    ordering = ('created_at',)
    inlines = (HotelsInline,)

    def response_add(self, request, obj, post_url_continue=None):
        print(obj.pk)
        print(obj.country)
        print(obj.hotels.all())
        return super().response_add(request, obj, post_url_continue)

Но то ли лыжи не едут, то ли я долбанутый — ничего не поменялось.

[javedimka]

Vic Shóstak, ну тык print()`ы то надо было поменять на функцию с отправкой письма) Это я просто для примера написал доступ к атрибутам созданного объекта, save() из модели вообще удалить надо и делать отправку в response_add, т.е. что-то типо этого:

#yourapp/utils.py

def send_mail_to_admin(obj):
    hotels = obj.hotels.all().order_by('cost')

    message = 'Тур ID ' + obj.pk + '\n'
    message += 'Страна: ' + obj.country_name + ' Город: ' + obj.country_name + '\n'
    message += 'Отели: \n'
    for hotel in hotels:
        message += hotel.name + ' ' + hotel.star + ' ' + hotel.cost + ' руб. \n'

    send_mail(
        'From Admin',
        message,
        '[e-mail]',
        ['[e-mail]'],
        fail_silently=False,
    )

#yourapp/admin.py
from .utils import send_mail_to_admin


class ToursAdmin(admin.ModelAdmin):
    exclude = ('created_at',)
    list_display = ('country',)
    ordering = ('created_at',)
    inlines = (HotelsInline,)

    def response_add(self, request, obj, post_url_continue=None):
        send_mail_to_admin(obj)
        return super().response_add(request, obj, post_url_continue)

[Vic Shóstak]

javedimka, вот уж воистину, тупанул так тупанул! o_O

Спасибо огромное, теперь работает, как задумывалось! По аналогии, добавил и response_change() (с отправкой письма о правках в записи).

Перенесите коммент выше в ответы — отмечу решением (для потомков).

P.S. Кстати, а почему в методе возвращается пустой super(), а не super(ToursAdmin, self), например? Это чисто для себя на будущее, чтобы знать.

[javedimka]

Vic Shóstak, в данном случае просто сокращение доступное в python3 +, вообще super() не простая тема и про нее есть хорошие статьи, в контексте джанги понимание как работает super() гарантирует хорошее понимание mixins в джанговских cbv и вообще django generic cbv в целом.
А ответ - я с телефона, может попозже, или сам можешь написать, мне не жалко.

Answer 1

[Vic Shóstak]

Выражаю благодарность пользователю javedimka за участие и помощь в пояснении многих моментов. Так как я не люблю вопросы без ответов в интернете, то напишу его тут (благо, что javedimka разрешил).

Итак, лучшая практика решения такой проблемы — не использовать Django Signals совсем. Особенно, когда есть встроенные методы, типа ModelAdmin.response_add, и уйти из модели в admin.py:

# ./app/utils.py

def send_mail_to_admin(obj):
    hotels = obj.hotels.all().order_by('cost')

    message = 'Тур ID ' + obj.pk + '\n'
    message += 'Страна: ' + obj.country_name + ' Город: ' + obj.country_name + '\n'
    message += 'Отели: \n'
    for hotel in hotels:
        message += hotel.name + ' ' + hotel.star + ' ' + hotel.cost + ' руб. \n'

    send_mail(
        'From Admin',
        message,
        '[e-mail]',
        ['[e-mail]'],
        fail_silently=False,
    )

# ./app/admin.py

from .utils import send_mail_to_admin


class ToursAdmin(admin.ModelAdmin):
    exclude = ('created_at',)
    list_display = ('country',)
    ordering = ('created_at',)
    inlines = (HotelsInline,)

    def response_add(self, request, obj, post_url_continue=None):
        send_mail_to_admin(obj)
        return super().response_add(request, obj, post_url_continue)

Answer 2

[Владимир Х]

Помоему нужно добавить 1 строку
# Signal
@receiver(post_save, sender=Tours)
@receiver(post_save, sender=ToursHotels) # отправляем сигнал при правке дочерней модели
def do_something(sender, **kwargs):
tour = Tours.objects.get(id=kwargs.get('instance').id)
hotels = ToursHotels.objects.filter(tour_id=kwargs.get('instance').id).order_by('cost')
...