django-ledger 0.5.5.5__py3-none-any.whl → 0.5.6.0__py3-none-any.whl

django_ledger/migrations/0014_ledgermodel_ledger_xid_and_more.py

@@ -0,0 +1,22 @@
+# Generated by Django 4.2.6 on 2024-01-15 13:11
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('django_ledger', '0013_stagedtransactionmodel_bundle_split'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='ledgermodel',
+            name='ledger_xid',
+            field=models.SlugField(allow_unicode=True, blank=True, help_text='User Defined Ledger ID', max_length=150, null=True, verbose_name='Ledger Slug'),
+        ),
+        migrations.AlterUniqueTogether(
+            name='ledgermodel',
+            unique_together={('entity', 'ledger_xid')},
+        ),
+    ]

django_ledger/models/accounts.py

@@ -44,8 +44,8 @@ from django_ledger.io.roles import (ACCOUNT_ROLE_CHOICES, BS_ROLES, GROUP_INVOIC
                                     GROUP_LIABILITIES, GROUP_CAPITAL, GROUP_INCOME, GROUP_EXPENSES, GROUP_COGS,
                                     ROOT_GROUP, BS_BUCKETS, ROOT_ASSETS, ROOT_LIABILITIES,
                                     ROOT_CAPITAL, ROOT_INCOME, ROOT_EXPENSES, ROOT_COA)
-from django_ledger.models import lazy_loader
 from django_ledger.models.mixins import CreateUpdateMixIn
+from django_ledger.models.utils import lazy_loader
 from django_ledger.settings import DJANGO_LEDGER_ACCOUNT_CODE_GENERATE, DJANGO_LEDGER_ACCOUNT_CODE_USE_PREFIX
 
 DEBIT = 'debit'
@@ -160,6 +160,11 @@ class AccountModelManager(MP_NodeManager):
         """
         return AccountModelQuerySet(self.model).order_by('path')
 
+    def for_user(self, user_model):
+        qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
+
     # todo: search for uses and pass EntityModel whenever possible.
     def for_entity(self,
                    user_model,
@@ -188,7 +193,7 @@ class AccountModelManager(MP_NodeManager):
         AccountModelQuerySet
             A QuerySet of all requested EntityModel Chart of Accounts.
         """
-        qs = self.get_queryset()
+        qs = self.for_user(user_model)
         if select_coa_model:
             qs = qs.select_related('coa_model')
 
@@ -203,11 +208,7 @@ class AccountModelManager(MP_NodeManager):
 
         if coa_slug:
             qs = qs.filter(coa_model__slug__exact=coa_slug)
-
-        return qs.filter(
-            Q(coa_model__entity__admin=user_model) |
-            Q(coa_model__entity__managers__in=[user_model])
-        ).order_by('coa_model')
+        return qs.order_by('coa_model')
 
     def for_entity_available(self, user_model, entity_slug, coa_slug: Optional[str] = None) -> AccountModelQuerySet:
         """

django_ledger/models/bank_account.py

@@ -64,6 +64,15 @@ class BankAccountModelManager(models.Manager):
     Custom defined Model Manager for the BankAccountModel.
     """
 
+    def for_user(self, user_model):
+        qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
+        return qs.filter(
+            Q(entity_model__admin=user_model) |
+            Q(entity_model__managers__in=[user_model])
+        )
+
     def for_entity(self, entity_slug, user_model) -> BankAccountModelQuerySet:
         """
         Allows only the authorized user to query the BankAccountModel for a given EntityModel.
@@ -76,21 +85,13 @@ class BankAccountModelManager(models.Manager):
         user_model
             Logged in and authenticated django UserModel instance.
         """
-        qs = self.get_queryset()
+        qs = self.for_user(user_model)
         if isinstance(entity_slug, lazy_loader.get_entity_model()):
             return qs.filter(
-                Q(entity_model=entity_slug) &
-                (
-                        Q(entity_model__admin=user_model) |
-                        Q(entity_model__managers__in=[user_model])
-                )
+                Q(entity_model=entity_slug)
             )
         return qs.filter(
-            Q(entity_model__slug__exact=entity_slug) &
-            (
-                    Q(entity_model__admin=user_model) |
-                    Q(entity_model__managers__in=[user_model])
-            )
+            Q(entity_model__slug__exact=entity_slug)
         )
 
 

django_ledger/models/bill.py

@@ -205,6 +205,8 @@ class BillModelManager(models.Manager):
             Returns a BillModelQuerySet with applied filters.
         """
         qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
         return qs.filter(
             Q(ledger__entity__admin=user_model) |
             Q(ledger__entity__managers__in=[user_model])
@@ -233,20 +235,14 @@ class BillModelManager(models.Manager):
         BillModelQuerySet
             Returns a BillModelQuerySet with applied filters.
         """
-        qs = self.get_queryset()
+        qs = self.for_user(user_model)
         if isinstance(entity_slug, EntityModel):
             return qs.filter(
-                Q(ledger__entity=entity_slug) & (
-                        Q(ledger__entity__admin=user_model) |
-                        Q(ledger__entity__managers__in=[user_model])
-                )
+                Q(ledger__entity=entity_slug)
             )
         elif isinstance(entity_slug, str):
             return qs.filter(
-                Q(ledger__entity__slug__exact=entity_slug) & (
-                        Q(ledger__entity__admin=user_model) |
-                        Q(ledger__entity__managers__in=[user_model])
-                )
+                Q(ledger__entity__slug__exact=entity_slug)
             )
 
 

django_ledger/models/closing_entry.py

@@ -35,23 +35,23 @@ class ClosingEntryModelQuerySet(models.QuerySet):
 
 class ClosingEntryModelManager(models.Manager):
 
+    def for_user(self, user_model):
+        qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
+        return qs.filter(
+            Q(entity_model__admin=user_model) |
+            Q(entity_model__managers__in=[user_model])
+        )
+
     def for_entity(self, entity_slug, user_model):
+        qs = self.for_user(user_model)
         if isinstance(entity_slug, lazy_loader.get_entity_model()):
-            return self.get_queryset().filter(
-                Q(entity_model=entity_slug) &
-                (
-                        Q(entity_model__admin=user_model) |
-                        Q(entity_model__managers__in=[user_model])
-                )
-
-            )
-        return self.get_queryset().filter(
-            Q(entity_model__slug__exact=entity_slug) &
-            (
-                    Q(entity_model__admin=user_model) |
-                    Q(entity_model__managers__in=[user_model])
+            return qs.filter(
+                Q(entity_model=entity_slug)
             )
-
+        return qs.filter(
+            Q(entity_model__slug__exact=entity_slug)
         )
 
 

django_ledger/models/coa.py

@@ -233,7 +233,7 @@ class ChartOfAccountModelAbstract(SlugNameMixIn, CreateUpdateMixIn):
                     message=_(f'CoA {self.uuid} already has a slug')
                 )
             return
-        self.slug = ''.join(choices(SLUG_SUFFIX, k=5))
+        self.slug = f'coa-{self.entity.slug[-5:]}-' + ''.join(choices(SLUG_SUFFIX, k=15))
 
     def configure(self, raise_exception: bool = True):
 

django_ledger/models/customer.py

@@ -107,6 +107,8 @@ class CustomerModelManager(models.Manager):
             >>> customer_model_qs = CustomerModel.objects.for_user(user_model=request_user)
         """
         qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
         return qs.filter(
             Q(entity__admin=user_model) |
             Q(entity__managers__in=[user_model])
@@ -135,22 +137,14 @@ class CustomerModelManager(models.Manager):
         CustomerModelQueryset
             A filtered CustomerModel QuerySet.
         """
-        qs = self.get_queryset()
+        qs = self.for_user(user_model)
 
         if isinstance(entity_slug, lazy_loader.get_entity_model()):
             return qs.filter(
-                Q(entity_model=entity_slug) &
-                (
-                        Q(entity_model__admin=user_model) |
-                        Q(entity_model__managers__in=[user_model])
-                )
+                Q(entity_model=entity_slug)
             )
         return qs.filter(
-            Q(entity_model__slug__exact=entity_slug) &
-            (
-                    Q(entity_model__admin=user_model) |
-                    Q(entity_model__managers__in=[user_model])
-            )
+            Q(entity_model__slug__exact=entity_slug)
         )
 
 

django_ledger/models/data_import.py

@@ -42,14 +42,20 @@ class ImportJobModelManager(models.Manager):
             'ledger_model'
         )
 
-    def for_entity(self, entity_slug: str, user_model):
+    def for_user(self, user_model):
         qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
         return qs.filter(
-            Q(bank_account_model__entity_model__slug__exact=entity_slug) &
-            (
-                    Q(bank_account_model__entity_model__admin=user_model) |
-                    Q(bank_account_model__entity_model__managers__in=[user_model])
-            )
+            Q(bank_account_model__entity_model__admin=user_model) |
+            Q(bank_account_model__entity_model__managers__in=[user_model])
+
+        )
+
+    def for_entity(self, entity_slug: str, user_model):
+        qs = self.for_user(user_model)
+        return qs.filter(
+            Q(bank_account_model__entity_model__slug__exact=entity_slug)
         )
 
 

django_ledger/models/entity.py

@@ -27,7 +27,7 @@ from decimal import Decimal
 from itertools import zip_longest
 from random import choices
 from string import ascii_lowercase, digits
-from typing import Tuple, Union, Optional, List, Dict
+from typing import Tuple, Union, Optional, List, Dict, Set
 from uuid import uuid4, UUID
 
 from django.contrib.auth import get_user_model
@@ -45,7 +45,7 @@ from django.utils.translation import gettext_lazy as _
 from treebeard.mp_tree import MP_Node, MP_NodeManager, MP_NodeQuerySet
 
 from django_ledger.io import roles as roles_module, validate_roles, IODigestContextManager
-from django_ledger.io.io_mixin import IOMixIn
+from django_ledger.io.io_core import IOMixIn
 from django_ledger.models.accounts import AccountModel, AccountModelQuerySet, DEBIT, CREDIT
 from django_ledger.models.bank_account import BankAccountModelQuerySet, BankAccountModel
 from django_ledger.models.coa import ChartOfAccountModel, ChartOfAccountModelQuerySet
@@ -114,7 +114,7 @@ class EntityModelManager(MP_NodeManager):
 
     def get_queryset(self):
         """Sets the custom queryset as the default."""
-        qs = super().get_queryset()
+        qs = EntityModelQuerySet(self.model).order_by('path')
         return qs.order_by('path').select_related('admin', 'default_coa')
 
     def for_user(self, user_model):
@@ -135,6 +135,8 @@ class EntityModelManager(MP_NodeManager):
                 2. Is a manager.
         """
         qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
         return qs.filter(
             Q(admin=user_model) |
             Q(managers__in=[user_model])
@@ -890,8 +892,15 @@ class EntityModelAbstract(MP_Node,
         return user_model.id == self.admin_id
 
     # #### LEDGER MANAGEMENT....
-    def create_ledger(self, name):
-        return self.ledgermodel_set.create(name=name)
+    def create_ledger(self, name: str, ledger_xid: Optional[str] = None, posted: bool = False, commit: bool = True):
+        if commit:
+            return self.ledgermodel_set.create(name=name, ledger_xid=ledger_xid, posted=posted)
+        return LedgerModel(
+            entity=self,
+            posted=posted,
+            name=name,
+            ledger_xid=ledger_xid
+        )
 
     # #### SLUG GENERATION ###
     @staticmethod
@@ -1293,7 +1302,7 @@ class EntityModelAbstract(MP_Node,
         return self.get_coa_accounts(active=active, order_by=order_by)
 
     def get_accounts_with_codes(self,
-                                code_list: Union[str, List[str]],
+                                code_list: Union[str, List[str], Set[str]],
                                 coa_model: Optional[Union[ChartOfAccountModel, UUID, str]] = None
                                 ) -> AccountModelQuerySet:
         """
@@ -1351,12 +1360,72 @@ class EntityModelAbstract(MP_Node,
         return account_model_qs.get(role__exact=role)
 
     def create_account(self,
-                       account_model_kwargs: Dict,
+                       code: str,
+                       role: str,
+                       name: str,
+                       balance_type: str,
+                       active: bool,
                        coa_model: Optional[Union[ChartOfAccountModel, UUID, str]] = None,
-                       raise_exception: bool = True) -> Tuple[ChartOfAccountModel, AccountModel]:
+                       raise_exception: bool = True) -> AccountModel:
         """
         Creates a new AccountModel for the EntityModel.
 
+        Parameters
+        ----------
+        code: str
+            The AccountModel code of the new Account.
+        role: str
+            The choice of role that the new Account belongs to.
+        name: str
+            The name of the new Account.
+        balance_type: str
+            The balance type of the new account. Possible values are 'debit' or 'credit'.
+        active: bool
+            Active status of the new account.
+        coa_model: ChartOfAccountModel, UUID, str
+            The ChartOfAccountsModel UUID, model instance or slug to pull accounts from. Uses default Coa if not
+            provided.
+        raise_exception: bool
+            Raises EntityModelValidationError if ChartOfAccountsModel is not valid for the EntityModel instance.
+
+        Returns
+        -------
+        A tuple of ChartOfAccountModel, AccountModel
+            The ChartOfAccountModel and AccountModel instance just created.
+        """
+        if coa_model:
+            if isinstance(coa_model, UUID):
+                coa_model = self.chartofaccountsmodel_set.get(uuid__exact=coa_model)
+            elif isinstance(coa_model, str):
+                coa_model = self.chartofaccountsmodel_set.get(slug__exact=coa_model)
+            elif isinstance(coa_model, ChartOfAccountModel):
+                self.validate_chart_of_accounts_for_entity(
+                    coa_model=coa_model,
+                    raise_exception=raise_exception
+                )
+        else:
+            coa_model = self.default_coa
+
+        account_model = AccountModel(
+            code=code,
+            name=name,
+            role=role,
+            active=active,
+            balance_type=balance_type
+        )
+
+        account_model.clean()
+        return coa_model.create_account(account_model=account_model)
+
+    def create_account_by_kwargs(self,
+                                 account_model_kwargs: Dict,
+                                 coa_model: Optional[Union[ChartOfAccountModel, UUID, str]] = None,
+                                 raise_exception: bool = True) -> Tuple[ChartOfAccountModel, AccountModel]:
+        """
+        Creates a new AccountModel for the EntityModel by passing AccountModel KWARGS.
+        This is a legacy method for creating a new AccountModel for the EntityModel.
+        Will be deprecated in favor of create_account().
+
         Parameters
         ----------
         coa_model: ChartOfAccountModel, UUID, str
@@ -1378,7 +1447,10 @@ class EntityModelAbstract(MP_Node,
             elif isinstance(coa_model, str):
                 coa_model = self.chartofaccountsmodel_set.get(slug__exact=coa_model)
             elif isinstance(coa_model, ChartOfAccountModel):
-                self.validate_chart_of_accounts_for_entity(coa_model=coa_model, raise_exception=raise_exception)
+                self.validate_chart_of_accounts_for_entity(
+                    coa_model=coa_model,
+                    raise_exception=raise_exception
+                )
         else:
             coa_model = self.default_coa
 
@@ -2692,6 +2764,12 @@ class EntityModelAbstract(MP_Node,
         data_generator.populate_entity()
 
     # URLS ----
+    def get_absolute_url(self):
+        return reverse(viewname='django_ledger:entity-dashboard',
+                       kwargs={
+                           'entity_slug': self.slug
+                       })
+
     def get_dashboard_url(self) -> str:
         """
         The EntityModel Dashboard URL.
@@ -2954,7 +3032,7 @@ class EntityStateModelAbstract(models.Model):
         ]
 
     def __str__(self):
-        return f'{self.__class__.__name__} {self.entity_id}: FY: {self.fiscal_year}, KEY: {self.get_key_display()}'
+        return f'{self.__class__.__name__} {self.entity_model_id}: FY: {self.fiscal_year}, KEY: {self.get_key_display()}'
 
 
 class EntityStateModel(EntityStateModelAbstract):

django_ledger/models/estimate.py

@@ -111,6 +111,15 @@ class EstimateModelManager(models.Manager):
     A custom defined EstimateModelManager that that implements custom QuerySet methods related to the EstimateModel.
     """
 
+    def for_user(self, user_model):
+        qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
+        return qs.filter(
+            Q(entity__admin=user_model) |
+            Q(entity__managers__in=[user_model])
+        )
+
     def for_entity(self, entity_slug: Union[EntityModel, str], user_model):
         """
         Fetches a QuerySet of EstimateModels associated with a specific EntityModel & UserModel.
@@ -134,19 +143,13 @@ class EstimateModelManager(models.Manager):
         EstimateModelQuerySet
             Returns a EstimateModelQuerySet with applied filters.
         """
-        qs = self.get_queryset()
+        qs = self.for_user(user_model)
         if isinstance(entity_slug, EntityModel):
             return qs.filter(
-                Q(entity=entity_slug) & (
-                        Q(entity__admin=user_model) |
-                        Q(entity__managers__in=[user_model])
-                )
+                Q(entity=entity_slug)
             )
         return qs.filter(
-            Q(entity__slug__exact=entity_slug) & (
-                    Q(entity__admin=user_model) |
-                    Q(entity__managers__in=[user_model])
-            )
+            Q(entity__slug__exact=entity_slug)
         )
 
 

django_ledger/models/invoice.py

@@ -179,6 +179,15 @@ class InvoiceModelManager(models.Manager):
             'ledger__entity'
         )
 
+    def for_user(self, user_model):
+        qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
+        return qs.filter(
+            Q(ledger__entity__admin=user_model) |
+            Q(ledger__entity__managers__in=[user_model])
+        )
+
     def for_entity(self, entity_slug, user_model) -> InvoiceModelQuerySet:
         """
         Returns a QuerySet of InvoiceModels associated with a specific EntityModel & UserModel.
@@ -196,10 +205,7 @@ class InvoiceModelManager(models.Manager):
         InvoiceModelQuerySet
             A Filtered InvoiceModelQuerySet.
         """
-        qs = self.get_queryset().filter(
-            Q(ledger__entity__admin=user_model) |
-            Q(ledger__entity__managers__in=[user_model])
-        )
+        qs = self.for_user(user_model)
         if isinstance(entity_slug, EntityModel):
             return qs.filter(ledger__entity=entity_slug)
         elif isinstance(entity_slug, str):

django_ledger/models/items.py

@@ -879,14 +879,19 @@ class ItemTransactionModelQuerySet(models.QuerySet):
 
 class ItemTransactionModelManager(models.Manager):
 
-    def for_entity(self, user_model, entity_slug):
+    def for_user(self, user_model):
         qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
         return qs.filter(
-            Q(item_model__entity__slug__exact=entity_slug) &
-            (
-                    Q(item_model__entity__admin=user_model) |
-                    Q(item_model__entity__managers__in=[user_model])
-            )
+            Q(item_model__entity__admin=user_model) |
+            Q(item_model__entity__managers__in=[user_model])
+        )
+
+    def for_entity(self, user_model, entity_slug):
+        qs = self.for_user(user_model)
+        return qs.filter(
+            Q(item_model__entity__slug__exact=entity_slug)
         )
 
     def for_bill(self, user_model, entity_slug, bill_pk):

django_ledger/models/journal_entry.py

@@ -140,6 +140,8 @@ class JournalEntryModelManager(models.Manager):
 
     def for_user(self, user_model):
         qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
         return qs.filter(
             Q(ledger__entity__admin=user_model) |
             Q(ledger__entity__managers__in=[user_model])
@@ -168,22 +170,13 @@ class JournalEntryModelManager(models.Manager):
         JournalEntryModelQuerySet
             Returns a JournalEntryModelQuerySet with applied filters.
         """
+        qs = self.for_user(user_model)
         if isinstance(entity_slug, lazy_loader.get_entity_model()):
-            return self.get_queryset().filter(
-                Q(ledger__entity=entity_slug) &
-                (
-                        Q(ledger__entity__admin=user_model) |
-                        Q(ledger__entity__managers__in=[user_model])
-                )
-
+            return qs.filter(
+                Q(ledger__entity=entity_slug)
             )
         return self.get_queryset().filter(
-            Q(ledger__entity__slug__iexact=entity_slug) &
-            (
-                    Q(ledger__entity__admin=user_model) |
-                    Q(ledger__entity__managers__in=[user_model])
-            )
-
+            Q(ledger__entity__slug__iexact=entity_slug)
         )
 
     def for_ledger(self, ledger_pk: Union[str, UUID], entity_slug, user_model):