from django.contrib import admin
from .models import Worker, Project, Team, WorkLog, UserProfile

@admin.register(UserProfile)
class UserProfileAdmin(admin.ModelAdmin):
    list_display = ('user', 'pin', 'is_admin')

@admin.register(Worker)
class WorkerAdmin(admin.ModelAdmin):
    list_display = ('name', 'id_no', 'phone_no', 'monthly_salary', 'date_of_employment', 'projects_worked_on_count')
    search_fields = ('name', 'id_no')
    readonly_fields = ('projects_worked_on_count',) # Calculated field should be readonly in edit form

@admin.register(Project)
class ProjectAdmin(admin.ModelAdmin):
    list_display = ('name', 'created_at')
    filter_horizontal = ('supervisors',)

@admin.register(Team)
class TeamAdmin(admin.ModelAdmin):
    list_display = ('name', 'supervisor', 'created_at')
    filter_horizontal = ('workers',)

@admin.register(WorkLog)
class WorkLogAdmin(admin.ModelAdmin):
    list_display = ('date', 'project', 'supervisor')
    list_filter = ('date', 'project', 'supervisor')
    filter_horizontal = ('workers',)