from django.contrib import admin
from ctrack.caf.models import CAF, EssentialService
from .models import (
Address,
AddressType,
IncidentReport,
Mode,
Organisation,
Person,
Role,
Stakeholder,
Submode,
)
# So we can get the organisation name - a reverse lookup
def get_organisation_name(person):
return Organisation.objects.filter(person__id=person.id).first().name
def get_first_caf(org):
return CAF.objects.filter(organisation__id=org.id).first().version
# We need this to ensure the column header in the admin does't read the func name
get_organisation_name.short_description = "Organisation"
class IncidentReportAdmin(admin.ModelAdmin):
model = IncidentReport
readonly_fields = ["date_time_incident_reported"]
class AddressTypeAdmin(admin.ModelAdmin):
pass
class StakeholderAdmin(admin.ModelAdmin):
model = Stakeholder
class AddressInLine(admin.StackedInline):
model = Address
max_num = 3
extra = 1
class EssentialServiceInline(admin.StackedInline):
model = EssentialService
max_num = 3
extra = 1
class OrganisationAdmin(admin.ModelAdmin):
inlines = [AddressInLine, EssentialServiceInline]
list_display = (
"name",
"submode",
"date_updated",
"lead_inspector",
"deputy_lead_inspector",
)
class PersonAdmin(admin.ModelAdmin):
model = Person
list_display = [
"first_name",
"last_name",
"job_title",
get_organisation_name,
"email",
"mobile",
]
class RoleAdmin(admin.ModelAdmin):
model = Role
class ModeAdmin(admin.ModelAdmin):
model = Mode
class SubmodeAdmin(admin.ModelAdmin):
model = Submode
# Register your models here.
admin.site.register(Organisation, OrganisationAdmin)
admin.site.register(AddressType, AddressTypeAdmin)
admin.site.register(Role, RoleAdmin)
admin.site.register(Person, PersonAdmin)
admin.site.register(Mode, ModeAdmin)
admin.site.register(Submode, SubmodeAdmin)
admin.site.register(Stakeholder, StakeholderAdmin)
admin.site.register(IncidentReport, IncidentReportAdmin)
