from django.contrib import admin

from .models import DocumentRecord, EmployeeDocument


@admin.register(EmployeeDocument)
class EmployeeDocumentAdmin(admin.ModelAdmin):
    list_display = ('name', 'company', 'user', 'sort_id', 'is_required', 'date_added')
    list_filter = ('company', 'is_required')
    search_fields = ('name', 'company__company_name', 'user__username', 'user__email')


@admin.register(DocumentRecord)
class DocumentRecordAdmin(admin.ModelAdmin):
    list_display = ('id', 'employee_document', 'status', 'deadline', 'uploaded_date', 'last_update')
    list_filter = ('status',)
    search_fields = (
        'employee_document__name',
        'employee_document__company__company_name',
        'employee_document__user__username',
        'employee_document__user__email',
    )