admin.py

__copyright__ = "Copyright © Stichting SciPost (SciPost Foundation)"
__license__ = "AGPL v3"


from django.contrib import admin

from .models import (
    Subsidy,
    SubsidyPayment,
    SubsidyAttachment,
    PubFrac,
    PubFracCompensation,
    PublicationExpenditureCoverage,
    WorkLog,
    PeriodicReportType,
    PeriodicReport,
)


class SubsidyPaymentInline(admin.TabularInline):
    model = SubsidyPayment


class SubsidyAttachmentInline(admin.TabularInline):
    model = SubsidyAttachment


@admin.register(Subsidy)
class SubsidyAdmin(admin.ModelAdmin):
    inlines = [
        SubsidyPaymentInline,
        SubsidyAttachmentInline,
    ]
    autocomplete_fields = [
        "organization",
        "renewal_of",
    ]
    search_fields = [
        "organization__name",
        "organization__name_original",
        "organization__acronym",
    ]


@admin.register(SubsidyAttachment)
class SubsidyAttachmentAdmin(admin.ModelAdmin):
    list_display = [
        "kind",
        "date",
        "subsidy",
    ]
    list_filter = [
        "kind",
        "date",
    ]
    autocomplete_fields = [
        "subsidy",
    ]
    search_fields = [
        "description",
        "subsidy__organization__name",
        "subsidy__organization__name_original",
        "subsidy__organization__acronym",
    ]


@admin.register(PubFrac)
class PubFracAdmin(admin.ModelAdmin):
    list_display = [
        "organization",
        "doi_label_display",
        "fraction",
        "cf_value",
    ]
    autocomplete_fields = [
        "organization",
        "publication",
    ]
    search_fields = [
        "publication__doi_label",
        "organization__name",
        "organization__name_original",
        "organization__acronym",
    ]

    @admin.display(description='doi label')
    def doi_label_display(self, obj):
        return (obj.publication.doi_label)


@admin.register(PubFracCompensation)
class PubFracCompensationAdmin(admin.ModelAdmin):
    list_display = [
        "subsidy",
        "doi_label_display",
        "amount",
    ]
    autocomplete_fields = [
        "subsidy",
        "pubfrac",
    ]
    search_fields = [
        "subsidy",
        "pubfrac__organization__name",
        "pubfrac__organization__name_original",
        "pubfrac__organization__acronym",
        "pubfrac__publication__doi_label",
    ]

    @admin.display(description='doi label')
    def doi_label_display(self, obj):
        return (obj.pubfrac.publication.doi_label)


@admin.register(PublicationExpenditureCoverage)
class PublicationExpenditureCoverageAdmin(admin.ModelAdmin):
    list_display = [
        "subsidy",
        "doi_label_display",
        "amount",
    ]
    autocomplete_fields = [
        "subsidy",
        "publication",
    ]
    search_fields = [
        "subsidy",
        "subsidy__organization__name",
        "subsidy__organization__name_original",
        "subsidy__organization__acronym",
        "publication__doi_label",
    ]

    @admin.display(description='doi label')
    def doi_label_display(self, obj):
        return (obj.publication.doi_label)


@admin.register(WorkLog)
class WorkLogAdmin(admin.ModelAdmin):
    autocomplete_fields = ["user"]



admin.site.register(PeriodicReportType)

admin.site.register(PeriodicReport)