PyInventory/inventory/admin/item.py

import logging

import tagulous
from adminsortable2.admin import SortableAdminMixin, SortableInlineAdminMixin
from django.conf import settings
from django.contrib import admin
from django.template.loader import render_to_string
from django.urls import reverse
from django.utils.html import format_html
from django.utils.translation import gettext_lazy as _
from import_export.admin import ImportExportMixin
from import_export.resources import ModelResource

from inventory.admin.base import (
    BaseFileModelInline,
    BaseImageModelInline,
    BaseUserAdmin,
    LimitTreeDepthListFilter,
    UserInlineMixin,
)
from inventory.admin.tagulous_fix import TagulousModelAdminFix
from inventory.models import ItemLinkModel, ItemModel
from inventory.models.item import ItemFileModel, ItemImageModel
from inventory.string_utils import ltruncatechars


logger = logging.getLogger(__name__)


class ItemLinkModelInline(UserInlineMixin, SortableInlineAdminMixin, admin.TabularInline):
    model = ItemLinkModel
    extra = 0


class ItemImageModelInline(BaseImageModelInline):
    model = ItemImageModel


class ItemFileModelInline(BaseFileModelInline):
    model = ItemFileModel


class ItemModelResource(ModelResource):
    class Meta:
        model = ItemModel


@admin.register(ItemModel)
class ItemModelAdmin(TagulousModelAdminFix, ImportExportMixin, SortableAdminMixin, BaseUserAdmin):
    @admin.display(description=_('Related items'))
    def related_items(self, obj):
        if obj.pk is None:
            # Add a new item -> there are no related items ;)
            return '-'

        related_qs = ItemModel.tree_objects.related_objects(instance=obj)
        context = {
            'items': related_qs,
            'opts': self.opts,
        }
        return render_to_string('admin/item/related_items.html', context)

    @admin.display(ordering='path_str', description=_('ItemModel.verbose_name'))
    def item(self, obj):
        path = obj.path
        if len(path) > 1:
            prefixes = ' › '.join(path[:-1] + [''])
            prefixes = ltruncatechars(prefixes, max_length=settings.TREE_PATH_STR_MAX_LENGTH)
        else:
            prefixes = ''
        item = path[-1]
        url = reverse('admin:inventory_itemmodel_change', args=[obj.pk])
        return format_html(
            format_string='<a href="{url}">{prefixes}<strong>{item}</strong></a>',
            url=url,
            prefixes=prefixes,
            item=item,
        )

    def get_queryset(self, request):
        qs = super().get_queryset(request)
        qs = qs.prefetch_related(
            'location',
            'kind',
            'producer',
        )
        return qs

    def get_max_order(self, request, obj=None):
        # Work-a-round for: https://github.com/jrief/django-admin-sortable2/issues/341
        return 0

    date_hierarchy = 'create_dt'
    list_display = ('producer', 'item', 'kind', 'location', 'received_date', 'update_dt')
    ordering = ('path_str',)
    list_display_links = ()
    list_filter = (
        LimitTreeDepthListFilter,
        ('kind', admin.RelatedOnlyFieldListFilter),
        ('location', admin.RelatedOnlyFieldListFilter),
        ('producer', admin.RelatedOnlyFieldListFilter),
        ('tags', admin.RelatedOnlyFieldListFilter),
    )
    search_fields = ('name', 'description', 'kind__name', 'tags__name')
    fieldsets = (
        (
            _('Internals'),
            {
                'classes': ('collapse',),
                'fields': (
                    ('id', 'version'),
                    'user',
                ),
            },
        ),
        (_('Meta'), {'classes': ('collapse',), 'fields': ('create_dt', 'update_dt')}),
        (
            _('Basic'),
            {
                'fields': (
                    'kind',
                    ('producer', 'name'),
                    'description',
                    'tags',
                    'fcc_id',
                    'parent',
                    'location',
                )
            },
        ),
        (_('Related items'), {'classes': ('collapse',), 'fields': ('related_items',)}),
        (
            _('Lent'),
            {
                'classes': ('collapse',),
                'fields': (
                    'lent_to',
                    (
                        'lent_from_date',
                        'lent_until_date',
                    ),
                ),
            },
        ),
        (
            _('Received'),
            {
                'classes': ('collapse',),
                'fields': (('received_from', 'received_date', 'received_price'),),
            },
        ),
        (
            _('Handed over'),
            {
                'classes': ('collapse',),
                'fields': (('handed_over_to', 'handed_over_date', 'handed_over_price'),),
            },
        ),
    )
    autocomplete_fields = ('parent', 'location')
    readonly_fields = ('id', 'create_dt', 'update_dt', 'user', 'related_items')
    inlines = (ItemImageModelInline, ItemFileModelInline, ItemLinkModelInline)

    def get_list_display(self, request):
        list_display = list(super().get_list_display(request))

        # FIXME: SortableAdminMixin.get_list_display() adds this, we didn't need here:
        # See: https://github.com/jrief/django-admin-sortable2/issues/363
        if '_reorder_' in list_display:
            list_display.remove('_reorder_')

        return list_display


tagulous.admin.enhance(ItemModel, ItemModelAdmin)