thatandromeda/hamlet

from django.contrib import admin

from .models import Thesis, Person, Department, Contribution


class ContributionInline(admin.TabularInline):
    model = Contribution
    # This is important! There are enough Persons in our data set that the
    # admin page for individual theses will hang unless we use raw_id_fields.
    raw_id_fields = ('person',)


class ThesisAdmin(admin.ModelAdmin):
    inlines = (ContributionInline,)
    list_filter = ('department', 'degree')
    search_fields = ('contribution__person__name',
                     'department__course',
                     'identifier',
                     'title')


class PersonAdmin(admin.ModelAdmin):
    def theses_written(self, obj):
        return Thesis.objects.filter(
            contribution__role=Contribution.AUTHOR,
            contribution__person=obj).count()

    def theses_advised(self, obj):
        return Thesis.objects.filter(
            contribution__role=Contribution.ADVISOR,
            contribution__person=obj).count()

    list_display = ('name', 'theses_written', 'theses_advised')
    search_fields = ('name',)


admin.site.register(Thesis, ThesisAdmin)
admin.site.register(Person, PersonAdmin)
admin.site.register(Department, admin.ModelAdmin)