from django.contrib import admin
from apps.accounts.admin_utils import CompanyScopedAdminMixin
from .models import LoanAccount, LoanInstallment, LoanRecovery, ExitLoanSettlement
class LoanAdmin(CompanyScopedAdminMixin, admin.ModelAdmin): company_lookup = "employee__company"
class InstallmentAdmin(CompanyScopedAdminMixin, admin.ModelAdmin):
    company_lookup = "loan__employee__company"
    fk_company_lookups = {"loan": "employee__company"}
class RecoveryAdmin(CompanyScopedAdminMixin, admin.ModelAdmin):
    company_lookup = "loan__employee__company"
    fk_company_lookups = {
        "loan": "employee__company",
        "installment": "loan__employee__company",
        "payroll_record": "period__company",
        "final_settlement": "exit_case__employee__company",
    }
admin.site.register(LoanAccount, LoanAdmin)
admin.site.register(LoanInstallment, InstallmentAdmin)
admin.site.register(LoanRecovery, RecoveryAdmin)
admin.site.register(ExitLoanSettlement, RecoveryAdmin)