from django.contrib import admin
from apps.accounts.admin_utils import CompanyScopedAdminMixin
from .models import Employee, EmployeeDocument, CustomFieldDefinition, CustomFieldValue, EmployeePaymentAccount

class EmployeeAdmin(CompanyScopedAdminMixin, admin.ModelAdmin): company_lookup = "company"
class EmployeeDocumentAdmin(CompanyScopedAdminMixin, admin.ModelAdmin): company_lookup = "employee__company"
class CustomFieldDefinitionAdmin(CompanyScopedAdminMixin, admin.ModelAdmin): company_lookup = "company"
class CustomFieldValueAdmin(CompanyScopedAdminMixin, admin.ModelAdmin): company_lookup = "employee__company"
class EmployeePaymentAccountAdmin(CompanyScopedAdminMixin, admin.ModelAdmin): company_lookup = "employee__company"

admin.site.register(Employee, EmployeeAdmin)
admin.site.register(EmployeeDocument, EmployeeDocumentAdmin)
admin.site.register(CustomFieldDefinition, CustomFieldDefinitionAdmin)
admin.site.register(CustomFieldValue, CustomFieldValueAdmin)

admin.site.register(EmployeePaymentAccount, EmployeePaymentAccountAdmin)