diff --git a/payroll/models/tax_models.py b/payroll/models/tax_models.py index 034de1aa7..d423ce182 100644 --- a/payroll/models/tax_models.py +++ b/payroll/models/tax_models.py @@ -11,6 +11,7 @@ from django.db import models from django.utils.text import format_lazy from django.utils.translation import gettext_lazy as _ +from base.horilla_company_manager import HorillaCompanyManager from base.models import Company from horilla.models import HorillaModel from payroll.models.models import FilingStatus @@ -31,20 +32,12 @@ class PayrollSettings(HorillaModel): max_length=15, null=True, choices=choices, default="postfix" ) - company_id = models.ForeignKey( - Company, null=True, editable=False, on_delete=models.PROTECT - ) - objects = models.Manager() + company_id = models.ForeignKey(Company, null=True, on_delete=models.PROTECT) + objects = HorillaCompanyManager("company_id") def __str__(self): return f"Payroll Settings {self.currency_symbol}" - def save(self, *args, **kwargs): - if 1 < PayrollSettings.objects.count(): - raise ValidationError("You cannot add more conditions.") - - return super().save(*args, **kwargs) - class TaxBracket(HorillaModel): """ diff --git a/payroll/templates/payroll/settings/payroll_settings.html b/payroll/templates/payroll/settings/payroll_settings.html index 8a3d1a3df..ae0e9a776 100644 --- a/payroll/templates/payroll/settings/payroll_settings.html +++ b/payroll/templates/payroll/settings/payroll_settings.html @@ -44,6 +44,20 @@ {{ currency_form.position }} +