from django.contrib import admin

from .models import SOP, Instrument, SubInstrument


class SubInstrumentAdmin(admin.ModelAdmin):
    list_display = ("__str__", "itype", "parent")


class SOPInstrumentAdmin(admin.ModelAdmin):
    list_display = ("__str__", "designation", "sp_link", "draft_status", "author_lead")
    ordering = ("name",)


admin.site.register(Instrument)
admin.site.register(SubInstrument, SubInstrumentAdmin)
admin.site.register(SOP, SOPInstrumentAdmin)