from django.contrib import admin

from gym.models import Room, Equipment, TheoreticalMax, Session, Round, Unit, WorkForm


@admin.register(Room)
class RoomAdmin(admin.ModelAdmin):
    list_display = ["name", "latitude", "longitude"]


@admin.register(Equipment)
class EquipmentAdmin(admin.ModelAdmin):
    list_display = [
        "name",
        "room",
        "unit",
        "last_theoretical_max",
        "default_work_form",
        "default_repetition_number",
    ]
    list_editable = ["default_work_form", "default_repetition_number"]
    list_filter = ["room", "unit", "default_work_form"]


@admin.register(TheoreticalMax)
class TheoreticalMaxAdmin(admin.ModelAdmin):
    list_display = ["equipment", "date", "value"]
    list_display_links = ["equipment", "date"]
    list_filter = ["equipment"]
    date_hierarchy = "date"


@admin.register(Session)
class SessionAdmin(admin.ModelAdmin):
    list_display = ["start", "room", "default_theoretical_max_percentage"]
    list_filter = ["room"]
    date_hierarchy = "start"


@admin.register(Round)
class RoundAdmin(admin.ModelAdmin):
    list_display = [
        "equipment",
        "session",
        "repetition_number",
        "theoretical_max_percentage",
        "chosen_weight",
        "work_form",
    ]
    list_display_links = ["equipment", "session"]
    list_filter = ["equipment", "work_form"]
    date_hierarchy = "session__start"


@admin.register(Unit)
class UnitAdmin(admin.ModelAdmin):
    pass


@admin.register(WorkForm)
class WorkFormAdmin(admin.ModelAdmin):
    pass