med/media/admin.py

# -*- mode: python; coding: utf-8 -*-
# Copyright (C) 2017-2019 by BDE ENS Paris-Saclay
# SPDX-License-Identifier: GPL-3.0-or-later

from django.urls import reverse
from django.utils.html import format_html
from django.utils.translation import ugettext_lazy as _
from polymorphic.admin import PolymorphicChildModelAdmin, \
    PolymorphicParentModelAdmin
from med.admin import admin_site
from reversion.admin import VersionAdmin

from .forms import MediaAdminForm
from .models import Author, Borrowable, CD, Comic, Emprunt, FutureMedium, \
    Game, Manga, Novel, Review, Vinyl


class AuthorAdmin(VersionAdmin):
    list_display = ('name',)
    search_fields = ('name',)


class BorrowableAdmin(PolymorphicParentModelAdmin):
    search_fields = ('title',)
    child_models = (CD, Comic, Manga, Novel, Review, Vinyl,)
    show_in_index = False


class MediumAdmin(VersionAdmin, PolymorphicChildModelAdmin):
    list_display = ('__str__', 'authors_list', 'side_identifier', 'isbn',
                    'external_link')
    search_fields = ('title', 'authors__name', 'side_identifier', 'subtitle',
                     'isbn')
    autocomplete_fields = ('authors',)
    date_hierarchy = 'publish_date'
    form = MediaAdminForm
    show_in_index = True

    def authors_list(self, obj):
        return ", ".join([a.name for a in obj.authors.all()])

    authors_list.short_description = _('authors')

    def external_link(self, obj):
        if obj.external_url:
            return format_html('<a href="{}" target="about:blank">{}</a>',
                               obj.external_url, obj.external_url)
        else:
            return "-"

    external_link.allow_tags = True
    external_link.short_description = _('external url')

    def get_form(self, request, obj=None, **kwargs):
        """
        Pass request to form (for ISBN magic)
        """
        form = super().get_form(request, obj=obj, **kwargs)
        form.request = request
        return form

    def changeform_view(self, request, object_id=None, form_url='',
                        extra_context=None):
        """
        We use _continue for ISBN fetching, so remove continue button
        """
        extra_context = extra_context or {}
        extra_context['show_save_and_continue'] = False
        return super().changeform_view(request, object_id, form_url,
                                       extra_context=extra_context)


class FutureMediumAdmin(VersionAdmin):
    list_display = ('isbn',)
    search_fields = ('isbn',)

    def changeform_view(self, request, object_id=None, form_url='',
                        extra_context=None):
        """
        We use _continue for ISBN fetching, so remove continue button
        """
        extra_context = extra_context or {}
        extra_context['show_save_and_continue'] = False
        extra_context['show_save'] = False
        return super().changeform_view(request, object_id, form_url,
                                       extra_context=extra_context)


class CDAdmin(VersionAdmin, PolymorphicChildModelAdmin):
    list_display = ('title', 'authors_list', 'side_identifier',)
    search_fields = ('title', 'authors__name', 'side_identifier',)
    autocomplete_fields = ('authors',)
    show_in_index = True

    def authors_list(self, obj):
        return ", ".join([a.name for a in obj.authors.all()])

    authors_list.short_description = _('authors')


class VinylAdmin(VersionAdmin, PolymorphicChildModelAdmin):
    list_display = ('title', 'authors_list', 'side_identifier', 'rpm',)
    search_fields = ('title', 'authors__name', 'side_identifier', 'rpm',)
    autocomplete_fields = ('authors',)
    show_in_index = True

    def authors_list(self, obj):
        return ", ".join([a.name for a in obj.authors.all()])

    authors_list.short_description = _('authors')


class ReviewAdmin(VersionAdmin, PolymorphicChildModelAdmin):
    list_display = ('__str__', 'number', 'year', 'month', 'day', 'double',)
    search_fields = ('title', 'number', 'year',)
    show_in_index = True


class EmpruntAdmin(VersionAdmin):
    list_display = ('media', 'user', 'date_emprunt', 'date_rendu',
                    'permanencier_emprunt', 'permanencier_rendu_custom')
    search_fields = ('media__title', 'media__side_identifier',
                     'user__username', 'date_emprunt', 'date_rendu')
    date_hierarchy = 'date_emprunt'
    autocomplete_fields = ('media', 'user', 'permanencier_emprunt',
                           'permanencier_rendu')

    def permanencier_rendu_custom(self, obj):
        """
        Show a button if item has not been returned yet
        """
        if obj.permanencier_rendu:
            return obj.permanencier_rendu
        else:
            return format_html(
                '<a class="button" href="{}">{}</a>',
                reverse('media:retour-emprunt', args=[obj.pk]),
                _('Turn back')
            )

    permanencier_rendu_custom.short_description = _('given back to')
    permanencier_rendu_custom.allow_tags = True

    def add_view(self, request, form_url='', extra_context=None):
        """
        Autoselect keyholder registering a new borrowed item
        """
        # Make GET data mutable
        data = request.GET.copy()
        data['permanencier_emprunt'] = request.user
        request.GET = data
        return super().add_view(request, form_url, extra_context)


class GameAdmin(VersionAdmin, PolymorphicChildModelAdmin):
    list_display = ('title', 'owner', 'duration', 'players_min',
                    'players_max', 'comment')
    search_fields = ('name', 'owner__username', 'duration', 'comment')
    autocomplete_fields = ('owner',)


admin_site.register(Author, AuthorAdmin)
admin_site.register(Borrowable, BorrowableAdmin)
admin_site.register(Comic, MediumAdmin)
admin_site.register(Manga, MediumAdmin)
admin_site.register(Novel, MediumAdmin)
admin_site.register(CD, CDAdmin)
admin_site.register(Vinyl, VinylAdmin)
admin_site.register(Review, ReviewAdmin)
admin_site.register(FutureMedium, FutureMediumAdmin)
admin_site.register(Emprunt, EmpruntAdmin)
admin_site.register(Game, GameAdmin)