Ultron/ultron/followup/admin.py

from django.contrib import admin
from django.db.models import fields

from config.views import search
from .models import (
    Chrono,
    LearnedSkill,
    Point,
    Accident,
    MindState,
    GymnastHasRoutine,
    NumberOfRoutineDone,
    HeightWeight,
)
from django_extensions.admin import ForeignKeyAutocompleteAdmin


class ChronoAdmin(ForeignKeyAutocompleteAdmin):
    model = Chrono

    list_display = ("date", "gymnast", "tof", "chrono_type")
    list_filter = ("chrono_type",)
    # search_fields = ('gymnast', 'routine')
    autocomplete_fields = ["gymnast"]

    related_search_fields = {"gymnast": ("last_name", "first_name")}


class LearnedSkillAdmin(admin.ModelAdmin):
    model = LearnedSkill

    list_display = ("gymnast", "skill", "cando", "date")
    list_filter = ("gymnast", "skill", "cando")
    search_fields = ("gymnast", "skill")
    autocomplete_fields = ["gymnast", "skill"]


class PointAdmin(admin.ModelAdmin):
    model = Point

    list_display = (
        "gymnast",
        "point_execution",
        "point_difficulty",
        "point_time_of_flight",
        "total",
    )
    ordering = ("gymnast",)
    list_filter = ("gymnast", "event", "routine_type")


class AccidentAdmin(admin.ModelAdmin):
    model = Accident

    fields = ("date", "gymnast", "skill", "informations")  # educative
    list_display = ("date", "gymnast", "skill")  # educative
    list_filter = ("date",)
    search_fields = ("date", "gymnast")  # educative
    autocomplete_fields = ["gymnast"]  # educative


class MindStateAdmin(admin.ModelAdmin):
    model = MindState

    fields = ("gymnast", "date", "score", "informations")
    list_display = ("date", "gymnast", "score")
    list_filter = ("date", "gymnast")


class GymnastHasRoutineAdmin(ForeignKeyAutocompleteAdmin):
    model = GymnastHasRoutine

    list_display = ("gymnast", "routine", "routine_type", "datebegin", "dateend")
    list_filter = ("gymnast", "routine_type")
    search_fields = ("gymnast", "routine")
    autocomplete_fields = ("gymnast", "routine")


class NumberOfRoutineDoneAdmin(ForeignKeyAutocompleteAdmin):
    model = NumberOfRoutineDone

    list_display = (
        "gymnast",
        "routine_type",
        "date",
        "number_of_successes",
        "number_of_try",
    )
    list_filter = ("gymnast", "routine_type")


class HeightWeightAdmin(ForeignKeyAutocompleteAdmin):
    model = HeightWeight

    list_display = ("gymnast", "height", "hips_height", "weight", "date")
    list_filter = ("gymnast",)


admin.site.register(Accident, AccidentAdmin)
admin.site.register(Point, PointAdmin)
admin.site.register(Chrono, ChronoAdmin)
admin.site.register(LearnedSkill, LearnedSkillAdmin)
admin.site.register(MindState, MindStateAdmin)
admin.site.register(GymnastHasRoutine, GymnastHasRoutineAdmin)
admin.site.register(NumberOfRoutineDone, NumberOfRoutineDoneAdmin)
admin.site.register(HeightWeight, HeightWeightAdmin)