from django.contrib import admin
from ctrack.register.models import EngagementEvent, SingleDateTimeEvent, NoteEvent, CAFSingleDateEvent
from ctrack.register.models import EngagementType
class EngagementEventAdmin(admin.ModelAdmin):
model = EngagementEvent
list_display = ("type", "user", "date", "short_description", "response_date_requested")
class EngagementEventTypeAdmin(admin.ModelAdmin):
model = EngagementEventAdmin
list_display = ("descriptor", "enforcement_instrument", "single_date_type", "regulation_reference")
class SingleDateTimeEventAdmin(admin.ModelAdmin):
model = SingleDateTimeEvent
list_display = ("type_descriptor", "short_description", "date", "user", "created_date")
class CAFSingleDateEventAdmin(admin.ModelAdmin):
model = CAFSingleDateEvent
list_display = ("type_descriptor", "date")
class NoteEventAdmin(admin.ModelAdmin):
model = NoteEvent
list_display = ("short_description", "organisation", "user")
admin.site.register(EngagementEvent, EngagementEventAdmin)
admin.site.register(EngagementType, EngagementEventTypeAdmin)
admin.site.register(SingleDateTimeEvent, SingleDateTimeEventAdmin)
admin.site.register(CAFSingleDateEvent, CAFSingleDateEventAdmin)
admin.site.register(NoteEvent, NoteEventAdmin)
