summaryrefslogtreecommitdiffstats
path: root/engagements/admin.py
blob: 5c582ee3183c85980ce6476bd9dd46a3d94a15dd (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from django.contrib import admin

from .models import (
    Engagement,
    EngagementEffort,
    EngagementType,
    Organisation,
    Person,
    RegulatedEntityType,
    RegulatoryCycle,
    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}"


class RegulatoryCycleAdmin(admin.ModelAdmin):
    @admin.display(description="year")
    def year(self, obj):
        return f"{obj.get_year()}"

    list_display = ("__str__", "year")
    ordering = ("-start_date",)


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