from django.contrib import admin

from .models import (
    Engagement,
    EngagementEffort,
    EngagementType,
    Organisation,
    Person,
    RegulatedEntityType,
    RegulatoryRole,
)

site = admin.site

site.site_header = "DefNucSyR Engagement Database (DED)"


class PersonAdmin(admin.ModelAdmin):
    list_display = ("__str__", "organisation")

    @admin.display(description="fullname")
    def fullname(self, obj):
        return f"{obj.first_name} {obj.last_name}"


site.register(Person, PersonAdmin)
site.register(Organisation)
site.register(RegulatedEntityType)
site.register(RegulatoryRole)
site.register(EngagementType)
site.register(Engagement)
site.register(EngagementEffort)