Django many-to-many-field в отображении списка для элемента типа count
Я хочу отображать общее количество лайков против каждого элемента в Django admin list_display. Поле many to many связано с моделью пользователя. Ниже приведен мой код
Модель:
class Item(models.Model):
title = models.CharField(max_length=100)
description= RichTextField(blank=True, null=True)
main_image= models.ImageField(null=False, blank=False,upload_to='images/')
upload = models.FileField(upload_to ='uploads/')
date = models.DateTimeField(auto_now_add=True)
item_category = models.ForeignKey(Categories, default='Coding', on_delete=SET_DEFAULT)
item_tool = models.ForeignKey(Tools, default='XD', on_delete=SET_DEFAULT)
slug = models.SlugField(unique=True, blank=True, null=True) # new
author = models.ForeignKey(User, on_delete=models.CASCADE)
likes = models.ManyToManyField(User, related_name='post_likes')
def total_likes(self):
return self.likes.count()
Admin.py
from django.contrib import admin
from .models import Categories, Item, Tools
class ItemAdmin(admin.ModelAdmin):
list_display = ('title','author','item_category','date')
list_filter = ('item_category',)
admin.site.register(Categories)
admin.site.register(Item, ItemAdmin)
admin.site.register(Tools)
Вы можете добавить пользовательский дисплей к вашему list_display
, чтобы вернуть количество лайков, как это:
class ItemAdmin(admin.ModelAdmin):
list_display = ('title','author','item_category','date','likes')
list_filter = ('item_category',)
def likes(self, obj):
return len(obj.likes.all())
def get_queryset(self, *args, **kwargs):
return super().get_queryset(*args, **kwargs).prefetch_related('likes')
Обратите внимание на дополнительное изменение для get_queryset
, чтобы уменьшить количество обращений к базе данных при использовании prefetch_related
. Без него каждая строка экземпляров Item
будет обращаться к базе данных для получения количества лайков.