cmsmanage/paperwork/admin.py

from django.contrib import admin

from .models import (CmsRedRiverVeteransScholarship,
                     CertificationDefinition, Certification,
                     CertificationVersion,
                     InstructorOrVendor, SpecialProgram, Waiver)


class CertificationVersionInline(admin.TabularInline):
    model = CertificationVersion
    extra = 1


@admin.register(CertificationVersion)
class CertificationVersionAdmin(admin.ModelAdmin):
    search_fields = ['definition__certification_name', 'version']
    list_display = ['definition', 'version']
    list_filter = ['definition__department', 'definition__certification_name']


@admin.register(CertificationDefinition)
class CertificationDefinitionAdmin(admin.ModelAdmin):
    search_fields = ['certification_name']
    list_display = ['certification_name', 'department']
    list_filter = ['department']
    inlines = [CertificationVersionInline]


@admin.register(Certification)
class CertificationAdmin(admin.ModelAdmin):
    search_fields = ['name', 'certification__certification_name', 'certification__department']
    autocomplete_fields = ['member']
    exclude = ['shop_lead_notified']

    @admin.display(description='Certification Name',
                   ordering='certification_version__definition__certification_name')
    def certification_name(self, obj):
        return obj.certification_version.definition.certification_name

    @admin.display(description='Certification Version',
                   ordering='certification_version__version')
    def certification_version_version(self, obj):
        return obj.certification_version.version

    @admin.display(description='Department',
                   ordering='certification_version__definition__department')
    def certification_department(self, obj):
        return obj.certification_version.definition.department

    list_display = [
        'certification_name',
        'name',
        'certification_version_version',
        'certification_department',
        'date',
        'shop_lead_notified',
        'certified_by',
    ]
    list_display_links = [
        'certification_name',
        'name',
    ]
    list_filter = [
        'certification_version__definition__department',
        ('shop_lead_notified', admin.EmptyFieldListFilter),
    ]


@admin.register(InstructorOrVendor)
class InstructorOrVendorAdmin(admin.ModelAdmin):
    search_fields = ['name']


@admin.register(SpecialProgram)
class SpecialProgramAdmin(admin.ModelAdmin):
    search_fields = ['program_name']


@admin.register(Waiver)
class WaiverAdmin(admin.ModelAdmin):
    search_fields = ['name']


admin.site.register(CmsRedRiverVeteransScholarship)