django-ledger 0.8.2.1__py3-none-any.whl → 0.8.2.2__py3-none-any.whl

django_ledger/migrations/0027_alter_accountmodel_role_alter_receiptmodel_amount_and_more.py

@@ -0,0 +1,159 @@
+# Generated by Django 5.2.6 on 2025-10-03 01:36
+
+import django.core.validators
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ('django_ledger', '0026_stagedtransactionmodel_customer_model_and_more'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='accountmodel',
+            name='role',
+            field=models.CharField(
+                choices=[
+                    (
+                        'Assets',
+                        [
+                            ('asset_ca_cash', 'Current Asset'),
+                            ('asset_ca_mkt_sec', 'Marketable Securities'),
+                            ('asset_ca_recv', 'Receivables'),
+                            ('asset_ca_inv', 'Inventory'),
+                            ('asset_ca_uncoll', 'Uncollectibles'),
+                            ('asset_ca_prepaid', 'Prepaid'),
+                            ('asset_ca_other', 'Other Liquid Assets'),
+                            ('asset_lti_notes', 'Notes Receivable'),
+                            ('asset_lti_land', 'Land'),
+                            ('asset_lti_sec', 'Securities'),
+                            ('asset_ppe_build', 'Buildings'),
+                            ('asset_ppe_build_accum_depr', 'Buildings - Accum. Depreciation'),
+                            ('asset_ppe_plant', 'Plant'),
+                            ('asset_ppe_plant_depr', 'Plant - Accum. Depreciation'),
+                            ('asset_ppe_equip', 'Equipment'),
+                            ('asset_ppe_equip_accum_depr', 'Equipment - Accum. Depreciation'),
+                            ('asset_ia', 'Intangible Assets'),
+                            ('asset_ia_accum_amort', 'Intangible Assets - Accum. Amortization'),
+                            ('asset_adjustment', 'Other Assets'),
+                        ],
+                    ),
+                    (
+                        'Liabilities',
+                        [
+                            ('lia_cl_acc_payable', 'Accounts Payable'),
+                            ('lia_cl_credit_line', 'Credit Line'),
+                            ('lia_cl_wages_payable', 'Wages Payable'),
+                            ('lia_cl_int_payable', 'Interest Payable'),
+                            ('lia_cl_taxes_payable', 'Taxes Payable'),
+                            ('lia_cl_st_notes_payable', 'Short Term Notes Payable'),
+                            ('lia_cl_ltd_mat', 'Current Maturities of Long Tern Debt'),
+                            ('lia_cl_def_rev', 'Deferred Revenue'),
+                            ('lia_cl_other', 'Other Liabilities'),
+                            ('lia_ltl_notes', 'Long Term Notes Payable'),
+                            ('lia_ltl_bonds', 'Bonds Payable'),
+                            ('lia_ltl_mortgage', 'Mortgage Payable'),
+                        ],
+                    ),
+                    (
+                        'Equity',
+                        [
+                            ('eq_capital', 'Capital'),
+                            ('eq_stock_common', 'Common Stock'),
+                            ('eq_stock_preferred', 'Preferred Stock'),
+                            ('eq_adjustment', 'Other Equity Adjustments'),
+                            ('eq_dividends', 'Dividends & Distributions to Shareholders'),
+                            ('in_operational', 'Operational Income'),
+                            ('in_passive', 'Investing/Passive Income'),
+                            ('in_interest', 'Interest Income'),
+                            ('in_gain_loss', 'Capital Gain/Loss Income'),
+                            ('in_other', 'Other Income'),
+                            ('cogs_regular', 'Cost of Goods Sold'),
+                            ('ex_regular', 'Regular Expense'),
+                            ('ex_interest_st', 'Interest Expense - Short Term Debt'),
+                            ('ex_interest', 'Interest Expense - Long Term Debt'),
+                            ('ex_taxes', 'Tax Expense'),
+                            ('ex_capital', 'Capital Expense'),
+                            ('ex_depreciation', 'Depreciation Expense'),
+                            ('ex_amortization', 'Amortization Expense'),
+                            ('ex_other', 'Other Expense'),
+                        ],
+                    ),
+                    (
+                        'Root',
+                        [
+                            ('root_coa', 'CoA Root Account'),
+                            ('root_assets', 'Assets Root Account'),
+                            ('root_liabilities', 'Liabilities Root Account'),
+                            ('root_capital', 'Capital Root Account'),
+                            ('root_income', 'Income Root Account'),
+                            ('root_cogs', 'COGS Root Account'),
+                            ('root_expenses', 'Expenses Root Account'),
+                        ],
+                    ),
+                ],
+                max_length=30,
+                verbose_name='Account Role',
+            ),
+        ),
+        migrations.AlterField(
+            model_name='receiptmodel',
+            name='amount',
+            field=models.DecimalField(
+                decimal_places=2,
+                help_text='Amount of the receipt.',
+                max_digits=20,
+                validators=[django.core.validators.MinValueValidator(limit_value=0)],
+                verbose_name='Receipt Amount',
+            ),
+        ),
+        migrations.AlterField(
+            model_name='receiptmodel',
+            name='receipt_account',
+            field=models.ForeignKey(
+                blank=True,
+                help_text='The income or expense account where this transaction will be reflected',
+                null=True,
+                on_delete=django.db.models.deletion.PROTECT,
+                to='django_ledger.accountmodel',
+                verbose_name='PnL Account',
+            ),
+        ),
+        migrations.AlterField(
+            model_name='receiptmodel',
+            name='receipt_type',
+            field=models.CharField(
+                choices=[
+                    ('sales', 'Sales Receipt'),
+                    ('customer_refund', 'Sales Refund'),
+                    ('expense', 'Expense Receipt'),
+                    ('expense_refund', 'Expense Refund'),
+                    ('transfer', 'Transfer Receipt'),
+                    ('debt_paydown', 'Debt Paydown Receipt'),
+                ],
+                max_length=15,
+                verbose_name='Receipt Type',
+            ),
+        ),
+        migrations.AlterField(
+            model_name='stagedtransactionmodel',
+            name='receipt_type',
+            field=models.CharField(
+                blank=True,
+                choices=[
+                    ('sales', 'Sales Receipt'),
+                    ('customer_refund', 'Sales Refund'),
+                    ('expense', 'Expense Receipt'),
+                    ('expense_refund', 'Expense Refund'),
+                    ('transfer', 'Transfer Receipt'),
+                    ('debt_paydown', 'Debt Paydown Receipt'),
+                ],
+                help_text='The receipt type of the transaction.',
+                max_length=20,
+                null=True,
+                verbose_name='Receipt Type',
+            ),
+        ),
+    ]

django_ledger/models/chart_of_accounts.py

@@ -116,9 +116,7 @@ class ChartOfAccountModelQuerySet(QuerySet):
         if user_model.is_superuser:
             return self
 
-        return self.filter(
-            (Q(entity__admin=user_model) | Q(entity__managers__in=[user_model]))
-        )
+        return self.filter((Q(entity__admin=user_model) | Q(entity__managers__in=[user_model])))
 
 
 class ChartOfAccountModelManager(Manager):
@@ -148,9 +146,7 @@ class ChartOfAccountModelManager(Manager):
                     filter=Q(accountmodel__depth__gt=2) & Q(accountmodel__active=True),
                 ),
                 # Root-group presence and uniqueness checks:
-                accountmodel_rootgroup__count=Count(
-                    'accountmodel', filter=Q(accountmodel__role__in=ROOT_GROUP)
-                ),
+                accountmodel_rootgroup__count=Count('accountmodel', filter=Q(accountmodel__role__in=ROOT_GROUP)),
                 accountmodel_rootgroup_roles__distinct_count=Count(
                     'accountmodel__role',
                     filter=Q(accountmodel__role__in=ROOT_GROUP_META),
@@ -161,11 +157,7 @@ class ChartOfAccountModelManager(Manager):
                 configured=models.Case(
                     models.When(
                         Q(accountmodel_rootgroup__count__gte=1)
-                        & Q(
-                            accountmodel_rootgroup__count=F(
-                                'accountmodel_rootgroup_roles__distinct_count'
-                            )
-                        ),
+                        & Q(accountmodel_rootgroup__count=F('accountmodel_rootgroup_roles__distinct_count')),
                         then=Value(True, output_field=BooleanField()),
                     ),
                     default=Value(False, output_field=BooleanField()),
@@ -177,7 +169,9 @@ class ChartOfAccountModelManager(Manager):
 
     @deprecated_entity_slug_behavior
     def for_entity(
-        self, entity_model: 'Union[EntityModel | str | UUID]' = None, **kwargs
+        self,
+        entity_model: 'Union[EntityModel | str | UUID]' = None,  # noqa: F821
+        **kwargs,  # noqa: F821
     ) -> ChartOfAccountModelQuerySet:
         """
         Fetches a QuerySet of ChartOfAccountsModel associated with a specific EntityModel & UserModel.
@@ -240,16 +234,10 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
     """
 
     uuid = models.UUIDField(default=uuid4, editable=False, primary_key=True)
-    entity = models.ForeignKey(
-        'django_ledger.EntityModel', verbose_name=_('Entity'), on_delete=models.CASCADE
-    )
+    entity = models.ForeignKey('django_ledger.EntityModel', verbose_name=_('Entity'), on_delete=models.CASCADE)
     active = models.BooleanField(default=True, verbose_name=_('Is Active'))
-    description = models.TextField(
-        verbose_name=_('CoA Description'), null=True, blank=True
-    )
-    objects = ChartOfAccountModelManager.from_queryset(
-        queryset_class=ChartOfAccountModelQuerySet
-    )()
+    description = models.TextField(verbose_name=_('CoA Description'), null=True, blank=True)
+    objects = ChartOfAccountModelManager.from_queryset(queryset_class=ChartOfAccountModelQuerySet)()
 
     class Meta:
         abstract = True
@@ -299,9 +287,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
 
             if len(existing_root_roles) > 0:
                 if raise_exception:
-                    raise ChartOfAccountsModelValidationError(
-                        message=f'Root Nodes already Exist in CoA {self.uuid}...'
-                    )
+                    raise ChartOfAccountsModelValidationError(message=f'Root Nodes already Exist in CoA {self.uuid}...')
                 return
 
             if ROOT_COA not in existing_root_roles:
@@ -322,9 +308,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
                 AccountModel.add_root(instance=root_account)
 
                 # must retrieve root model after added pero django-treebeard documentation...
-                coa_root_account_model = AccountModel.objects.get(
-                    uuid__exact=account_pk
-                )
+                coa_root_account_model = AccountModel.objects.get(uuid__exact=account_pk)
 
                 for root_role in ROOT_GROUP_LEVEL_2:
                     if root_role not in existing_root_roles:
@@ -398,9 +382,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
 
         if account_model.coa_model_id != self.uuid:
             raise ChartOfAccountsModelValidationError(
-                message=_(
-                    f'The account model {account_model} is not part of the chart of accounts {self.name}.'
-                ),
+                message=_(f'The account model {account_model} is not part of the chart of accounts {self.name}.'),
             )
 
         # Chart of Accounts hasn't been configured...
@@ -412,29 +394,17 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
                 root_account_qs = self.get_coa_root_accounts_qs()
 
             if account_model.is_asset():
-                qs = root_account_qs.filter(
-                    code__exact=ROOT_GROUP_META[ROOT_ASSETS]['code']
-                )
+                qs = root_account_qs.filter(code__exact=ROOT_GROUP_META[ROOT_ASSETS]['code'])
             elif account_model.is_liability():
-                qs = root_account_qs.filter(
-                    code__exact=ROOT_GROUP_META[ROOT_LIABILITIES]['code']
-                )
+                qs = root_account_qs.filter(code__exact=ROOT_GROUP_META[ROOT_LIABILITIES]['code'])
             elif account_model.is_capital():
-                qs = root_account_qs.filter(
-                    code__exact=ROOT_GROUP_META[ROOT_CAPITAL]['code']
-                )
+                qs = root_account_qs.filter(code__exact=ROOT_GROUP_META[ROOT_CAPITAL]['code'])
             elif account_model.is_income():
-                qs = root_account_qs.filter(
-                    code__exact=ROOT_GROUP_META[ROOT_INCOME]['code']
-                )
+                qs = root_account_qs.filter(code__exact=ROOT_GROUP_META[ROOT_INCOME]['code'])
             elif account_model.is_cogs():
-                qs = root_account_qs.filter(
-                    code__exact=ROOT_GROUP_META[ROOT_COGS]['code']
-                )
+                qs = root_account_qs.filter(code__exact=ROOT_GROUP_META[ROOT_COGS]['code'])
             elif account_model.is_expense():
-                qs = root_account_qs.filter(
-                    code__exact=ROOT_GROUP_META[ROOT_EXPENSES]['code']
-                )
+                qs = root_account_qs.filter(code__exact=ROOT_GROUP_META[ROOT_EXPENSES]['code'])
             else:
                 raise ChartOfAccountsModelValidationError(
                     message=f'Unable to locate Balance Sheet'
@@ -445,9 +415,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
                 return qs
             return qs.get()
 
-        raise ChartOfAccountsModelValidationError(
-            message='Adding Root account to Chart of Accounts is not allowed.'
-        )
+        raise ChartOfAccountsModelValidationError(message='Adding Root account to Chart of Accounts is not allowed.')
 
     def get_non_root_coa_accounts_qs(self) -> AccountModelQuerySet:
         """
@@ -518,13 +486,9 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
         """
         if self.slug:
             if raise_exception:
-                raise ChartOfAccountsModelValidationError(
-                    message=_(f'CoA {self.uuid} already has a slug')
-                )
+                raise ChartOfAccountsModelValidationError(message=_(f'CoA {self.uuid} already has a slug'))
             return
-        self.slug = f'coa-{self.entity.slug[-5:]}-' + ''.join(
-            choices(SLUG_SUFFIX, k=15)
-        )
+        self.slug = f'coa-{self.entity.slug[-5:]}-' + ''.join(choices(SLUG_SUFFIX, k=15))
 
         if commit:
             self.save(update_fields=['slug', 'updated'])
@@ -569,14 +533,10 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
 
         """
         if not isinstance(account_model_qs, AccountModelQuerySet):
-            raise ChartOfAccountsModelValidationError(
-                message='Must pass an instance of AccountModelQuerySet'
-            )
+            raise ChartOfAccountsModelValidationError(message='Must pass an instance of AccountModelQuerySet')
         for acc_model in account_model_qs:
             if not acc_model.coa_model_id == self.uuid:
-                raise ChartOfAccountsModelValidationError(
-                    message=f'Invalid root queryset for CoA {self.name}'
-                )
+                raise ChartOfAccountsModelValidationError(message=f'Invalid root queryset for CoA {self.name}')
 
     def insert_account(
         self,
@@ -680,9 +640,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
         )
         account_model.clean()
 
-        account_model = self.insert_account(
-            account_model=account_model, root_account_qs=root_account_qs
-        )
+        account_model = self.insert_account(account_model=account_model, root_account_qs=root_account_qs)
         return account_model
 
     # ACTIONS -----
@@ -697,9 +655,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
         account_qs.update(locked=False)
         return account_qs
 
-    def mark_as_default(
-        self, commit: bool = False, raise_exception: bool = False, **kwargs
-    ):
+    def mark_as_default(self, commit: bool = False, raise_exception: bool = False, **kwargs):
         """
         Marks the current Chart of Accounts instances as default for the EntityModel.
 
@@ -719,9 +675,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
         if not self.can_mark_as_default():
             if raise_exception:
                 raise ChartOfAccountsModelValidationError(
-                    message=_(
-                        f'The Chart of Accounts {self.slug} cannot be marked as default'
-                    )
+                    message=_(f'The Chart of Accounts {self.slug} cannot be marked as default')
                 )
             return
         self.entity.default_coa_id = self.uuid
@@ -752,9 +706,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
         """
         return all([self.is_active(), not self.is_default()])
 
-    def mark_as_active(
-        self, commit: bool = False, raise_exception: bool = False, **kwargs
-    ):
+    def mark_as_active(self, commit: bool = False, raise_exception: bool = False, **kwargs):
         """
         Marks the current Chart of Accounts as Active.
 
@@ -767,9 +719,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
         """
         if self.is_active():
             if raise_exception:
-                raise ChartOfAccountsModelValidationError(
-                    message=_('The Chart of Accounts is currently active.')
-                )
+                raise ChartOfAccountsModelValidationError(message=_('The Chart of Accounts is currently active.'))
             return
 
         self.active = True
@@ -777,9 +727,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
         if commit:
             self.save(update_fields=['active', 'updated'])
 
-    def mark_as_inactive(
-        self, commit: bool = False, raise_exception: bool = False, **kwargs
-    ):
+    def mark_as_inactive(self, commit: bool = False, raise_exception: bool = False, **kwargs):
         """
         Marks the current Chart of Accounts as Active.
 
@@ -792,9 +740,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
         """
         if not self.is_active():
             if raise_exception:
-                raise ChartOfAccountsModelValidationError(
-                    message=_('The Chart of Accounts is currently not active.')
-                )
+                raise ChartOfAccountsModelValidationError(message=_('The Chart of Accounts is currently not active.'))
             return
 
         self.active = False
@@ -846,9 +792,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
         )
 
     def get_coa_list_url(self):
-        return reverse(
-            viewname='django_ledger:coa-list', kwargs={'entity_slug': self.entity_slug}
-        )
+        return reverse(viewname='django_ledger:coa-list', kwargs={'entity_slug': self.entity_slug})
 
     def get_coa_list_inactive_url(self):
         return reverse(
@@ -906,9 +850,7 @@ class ChartOfAccountModel(ChartOfAccountModelAbstract):
 def chartofaccountsmodel_presave(instance: ChartOfAccountModelAbstract, **kwargs):
     instance.generate_slug()
     if instance.is_default() and not instance.active:
-        raise ChartOfAccountsModelValidationError(
-            _('Default Chart of Accounts cannot be deactivated.')
-        )
+        raise ChartOfAccountsModelValidationError(_('Default Chart of Accounts cannot be deactivated.'))
 
 
 @receiver(post_save, sender=ChartOfAccountModel)

django_ledger/models/coa_default.py

@@ -109,6 +109,7 @@ Default Chart of Accounts Table
   4050  in_other                    credit          Other Income                                                   root_income
   4020  in_passive                  credit          Investing Income                                               root_income
   2010  lia_cl_acc_payable          credit          Accounts Payable                                               root_liabilities
+  2015  lia_cl_credit_line          credit          Accounts Payable                                               root_liabilities
   2060  lia_cl_def_rev              credit          Deferred Revenues                                              root_liabilities
   2030  lia_cl_int_payable          credit          Interest Payable                                               root_liabilities
   2050  lia_cl_ltd_mat              credit          Current Maturities LT Debt                                     root_liabilities

django_ledger/models/customer.py

@@ -54,7 +54,7 @@ class CustomerModelQueryset(QuerySet):
     reports.
     """
 
-    def for_user(self, user_model):
+    def for_user(self, user_model) -> 'CustomerModelQueryset':
         """
         Fetches a QuerySet of BillModels that the UserModel as access to.
         May include BillModels from multiple Entities.
@@ -70,12 +70,9 @@ class CustomerModelQueryset(QuerySet):
         """
         if user_model.is_superuser:
             return self
-        return self.filter(
-            Q(entity_model__admin=user_model)
-            | Q(entity_model__managers__in=[user_model])
-        )
+        return self.filter(Q(entity_model__admin=user_model) | Q(entity_model__managers__in=[user_model]))
 
-    def active(self) -> QuerySet:
+    def active(self) -> 'CustomerModelQueryset':
         """
         Active customers can be assigned to new Invoices and show on dropdown menus and views.
 
@@ -86,7 +83,7 @@ class CustomerModelQueryset(QuerySet):
         """
         return self.filter(active=True)
 
-    def inactive(self) -> QuerySet:
+    def inactive(self) -> 'CustomerModelQueryset':
         """
         Active customers can be assigned to new Invoices and show on dropdown menus and views.
         Marking CustomerModels as inactive can help reduce Database load to populate select inputs and also inactivate
@@ -100,7 +97,7 @@ class CustomerModelQueryset(QuerySet):
         """
         return self.filter(active=False)
 
-    def hidden(self) -> QuerySet:
+    def hidden(self) -> 'CustomerModelQueryset':
         """
         Hidden customers do not show on dropdown menus, but may be used via APIs or any other method that does not
         involve the UI.
@@ -112,7 +109,7 @@ class CustomerModelQueryset(QuerySet):
         """
         return self.filter(hidden=True)
 
-    def visible(self) -> QuerySet:
+    def visible(self) -> 'CustomerModelQueryset':
         """
         Visible customers show on dropdown menus and views. Visible customers are active and not hidden.
 
@@ -130,10 +127,11 @@ class CustomerModelManager(Manager):
     CustomerModel.
     """
 
+    def get_queryset(self) -> CustomerModelQueryset:
+        return CustomerModelQueryset(self.model, using=self._db)
+
     @deprecated_entity_slug_behavior
-    def for_entity(
-        self, entity_model: 'EntityModel | str | UUID' = None, **kwargs
-    ) -> CustomerModelQueryset:
+    def for_entity(self, entity_model: 'EntityModel | str | UUID', **kwargs) -> CustomerModelQueryset:  # noqa: F821
         """
         Fetches a QuerySet of CustomerModel associated with a specific EntityModel & UserModel.
         May pass an instance of EntityModel or a String representing the EntityModel slug.
@@ -161,16 +159,15 @@ class CustomerModelManager(Manager):
                 qs = qs.for_user(kwargs['user_model'])
 
         if isinstance(entity_model, EntityModel):
-            qs = qs.filter(entity_model=entity_model)
+            return qs.filter(entity_model=entity_model)
         elif isinstance(entity_model, str):
-            qs = qs.filter(entity_model__slug__exact=entity_model)
+            return qs.filter(entity_model__slug__exact=entity_model)
         elif isinstance(entity_model, UUID):
-            qs = qs.filter(entity_model_id=entity_model)
+            return qs.filter(entity_model_id=entity_model)
         else:
             raise CustomerModelValidationError(
                 message='Must pass EntityModel, slug or EntityModel UUID',
             )
-        return qs
 
 
 class CustomerModelAbstract(ContactInfoMixIn, TaxCollectionMixIn, CreateUpdateMixIn):
@@ -232,9 +229,7 @@ class CustomerModelAbstract(ContactInfoMixIn, TaxCollectionMixIn, CreateUpdateMi
     description = models.TextField()
     active = models.BooleanField(default=True)
     hidden = models.BooleanField(default=False)
-    picture = models.ImageField(
-        upload_to=customer_picture_upload_to, null=True, blank=True
-    )
+    picture = models.ImageField(upload_to=customer_picture_upload_to, null=True, blank=True)
 
     additional_info = models.JSONField(null=True, blank=True, default=dict)
 
@@ -292,9 +287,7 @@ class CustomerModelAbstract(ContactInfoMixIn, TaxCollectionMixIn, CreateUpdateMi
                 'key__exact': EntityStateModel.KEY_CUSTOMER,
             }
 
-            state_model_qs = EntityStateModel.objects.filter(
-                **LOOKUP
-            ).select_for_update()
+            state_model_qs = EntityStateModel.objects.filter(**LOOKUP).select_for_update()
             state_model = state_model_qs.get()
             state_model.sequence = F('sequence') + 1
             state_model.save()
@@ -344,11 +337,9 @@ class CustomerModelAbstract(ContactInfoMixIn, TaxCollectionMixIn, CreateUpdateMi
 
         return self.customer_number
 
-    def validate_for_entity(self, entity_model: 'EntityModel'):
+    def validate_for_entity(self, entity_model: 'EntityModel'):  # noqa: F821
         if entity_model.uuid != self.entity_model_id:
-            raise CustomerModelValidationError(
-                'EntityModel does not belong to this Vendor'
-            )
+            raise CustomerModelValidationError('EntityModel does not belong to this Vendor')
 
     def clean(self):
         """