from django.contrib import admin
from apps.accounts.admin_utils import CompanyScopedAdminMixin
from .models import LeaveType, LeaveBalance, LeaveRequest
class LeaveTypeAdmin(CompanyScopedAdminMixin, admin.ModelAdmin): company_lookup = "company"
class EmployeeLeaveAdmin(CompanyScopedAdminMixin, admin.ModelAdmin): company_lookup = "employee__company"
admin.site.register(LeaveType, LeaveTypeAdmin)
admin.site.register(LeaveBalance, EmployeeLeaveAdmin)
admin.site.register(LeaveRequest, EmployeeLeaveAdmin)