breccia-mapper/people/admin.py

"""
Admin site panels for models in the People app.
"""

from django.contrib import admin
from django.contrib.auth.admin import UserAdmin

from . import models


@admin.register(models.User)
class CustomUserAdmin(UserAdmin):
    """Add email address field to new user form."""
    add_fieldsets = UserAdmin.add_fieldsets + (
        ('Details', {'fields': ('email', )}),
    )  # yapf: disable


class OrganisationQuestionChoiceInline(admin.TabularInline):
    model = models.OrganisationQuestionChoice


@admin.register(models.OrganisationQuestion)
class OrganisationQuestionAdmin(admin.ModelAdmin):
    inlines = [
        OrganisationQuestionChoiceInline,
    ]


class OrganisationAnswerSetInline(admin.TabularInline):
    model = models.OrganisationAnswerSet
    readonly_fields = [
        'question_answers',
    ]


@admin.register(models.Organisation)
class OrganisationAdmin(admin.ModelAdmin):
    inlines = [
        OrganisationAnswerSetInline,
    ]


class PersonQuestionChoiceInline(admin.TabularInline):
    model = models.PersonQuestionChoice


@admin.register(models.PersonQuestion)
class PersonQuestionAdmin(admin.ModelAdmin):
    inlines = [
        PersonQuestionChoiceInline,
    ]


class PersonAnswerSetInline(admin.TabularInline):
    model = models.PersonAnswerSet
    readonly_fields = [
        'question_answers',
    ]


@admin.register(models.Person)
class PersonAdmin(admin.ModelAdmin):
    inlines = [
        PersonAnswerSetInline,
    ]


class RelationshipQuestionChoiceInline(admin.TabularInline):
    model = models.RelationshipQuestionChoice


@admin.register(models.RelationshipQuestion)
class RelationshipQuestionAdmin(admin.ModelAdmin):
    inlines = [
        RelationshipQuestionChoiceInline,
    ]


@admin.register(models.Relationship)
class RelationshipAdmin(admin.ModelAdmin):
    pass


class OrganisationRelationshipQuestionChoiceInline(admin.TabularInline):
    model = models.OrganisationRelationshipQuestionChoice


@admin.register(models.OrganisationRelationshipQuestion)
class OrganisationRelationshipQuestionAdmin(admin.ModelAdmin):
    inlines = [
        OrganisationRelationshipQuestionChoiceInline,
    ]


@admin.register(models.OrganisationRelationship)
class OrganisationRelationshipAdmin(admin.ModelAdmin):
    pass