from django.contrib import admin
from .models import ActivityType, Activity


@admin.register(ActivityType)
class ActivityTypeAdmin(admin.ModelAdmin):
    list_display = (
        'name',
        'is_active',
        'created_at',
    )

    list_filter = (
        'is_active',
    )

    search_fields = (
        'name',
        'description',
    )

    ordering = (
        'name',
    )


@admin.register(Activity)
class ActivityAdmin(admin.ModelAdmin):
    list_display = (
        'date',
        'activity_type',
        'location',
        'asset',
        'status',
        'priority',
    )

    list_filter = (
        'activity_type',
        'location',
        'status',
        'priority',
        'date',
    )

    search_fields = (
        'problem_description',
        'action_taken',
        'result',
        'notes',
        'asset__asset_code',
        'asset__name',
    )

    ordering = (
        '-date',
        '-created_at',
    )

    date_hierarchy = 'date'