django-ledger 0.5.6.4__py3-none-any.whl → 0.6.0__py3-none-any.whl

django_ledger/models/utils.py

@@ -6,152 +6,165 @@ Contributions to this module:
     * Miguel Sanda <msanda@arrobalytics.com>
 """
 
+from django.apps import apps
+
 
 class LazyLoader:
     """
-    This class eliminates the circle dependency between models.
+    A class that provides lazy-loading functionality for various models in the django_ledger app.
+
+    Attributes:
+        app_config (AppConfig): The AppConfig object for the django_ledger app.
+
+        ENTITY_MODEL (str): The name of the entity model.
+        LEDGER_MODEL (str): The name of the ledger model.
+        JE_MODEL (str): The name of the journal entry model.
+        TRANSACTION_MODEL (str): The name of the transaction model.
+        ACCOUNT_MODEL (str): The name of the account model.
+        COA_MODEL (str): The name of the chart of account model.
+
+        ENTITY_STATE_MODEL (str): The name of the entity state model.
+        ENTITY_UNIT_MODEL (str): The name of the entity unit model.
+        CLOSING_ENTRY_MODEL (str): The name of the closing entry model.
+        CLOSING_ENTRY_TRANSACTION_MODEL (str): The name of the closing entry transaction model.
+
+        BANK_ACCOUNT_MODEL (str): The name of the bank account model.
+        PURCHASE_ORDER_MODEL (str): The name of the purchase order model.
+
+        CUSTOMER_MODEL (str): The name of the customer model.
+        INVOICE_MODEL (str): The name of the invoice model.
+        BILL_MODEL (str): The name of the bill model.
+        UOM_MODEL (str): The name of the unit of measure model.
+        VENDOR_MODEL (str): The name of the vendor model.
+        ESTIMATE_MODEL (str): The name of the estimate model.
+        ITEM_MODEL (str): The name of the item model.
+        ITEM_TRANSACTION_MODEL (str): The name of the item transaction model.
+
+        ENTITY_DATA_GENERATOR (EntityDataGenerator): The EntityDataGenerator class used for generating entity data.
+        BALANCE_SHEET_REPORT_CLASS (BalanceSheetReport): The BalanceSheetReport class used for generating balance sheet reports.
+        INCOME_STATEMENT_REPORT_CLASS (IncomeStatementReport): The IncomeStatementReport class used for generating income statement reports.
+        CASH_FLOW_STATEMENT_REPORT_CLASS (CashFlowStatementReport): The CashFlowStatementReport class used for generating cash flow statement reports.
+
+    Methods:
+        get_entity_model() -> Model: Returns the entity model.
+        get_entity_unit_model() -> Model: Returns the entity unit model.
+        get_entity_state_model() -> Model: Returns the entity state model.
+        get_bank_account_model() -> Model: Returns the bank account model.
+        get_account_model() -> Model: Returns the account model.
+        get_coa_model() -> Model: Returns the chart of account model.
+        get_txs_model() -> Model: Returns the transaction model.
+        get_purchase_order_model() -> Model: Returns the purchase order model.
+        get_ledger_model() -> Model: Returns the ledger model.
+        get_journal_entry_model() -> Model: Returns the journal entry model.
+        get_item_model() -> Model: Returns the item model.
+        get_item_transaction_model() -> Model: Returns the item transaction model.
+        get_customer_model() -> Model: Returns the customer model.
+        get_bill_model() -> Model: Returns the bill model.
+        get_invoice_model() -> Model: Returns the invoice model.
+        get_uom_model() -> Model: Returns the unit of measure model.
+        get_vendor_model() -> Model: Returns the vendor model.
+        get_estimate_model() -> Model: Returns the estimate model.
+        get_closing_entry_model() -> Model: Returns the closing entry model.
+        get_closing_entry_transaction_model() -> Model: Returns the closing entry transaction model.
+        get_entity_data_generator() -> EntityDataGenerator: Returns the EntityDataGenerator class.
+        get_balance_sheet_report_class() -> BalanceSheetReport: Returns the BalanceSheetReport class.
+        get_income_statement_report_class() -> IncomeStatementReport: Returns the IncomeStatementReport class.
+        get_cash_flow_statement_report_class() -> CashFlowStatementReport: Returns the CashFlowStatementReport class.
     """
-    ENTITY_MODEL = None
-    ENTITY_STATE_MODEL = None
-    UNIT_MODEL = None
-    ACCOUNT_MODEL = None
-    BANK_ACCOUNT_MODEL = None
-    LEDGER_MODEL = None
-    TXS_MODEL = None
-    JE_MODEL = None
-    ITEM_MODEL = None
-    ITEM_TRANSACTION_MODEL = None
-    CUSTOMER_MODEL = None
-    INVOICE_MODEL = None
-    BILL_MODEL = None
-    UOM_MODEL = None
-    VENDOR_MODEL = None
-    TRANSACTION_MODEL = None
-    ENTITY_UNIT_MODEL = None
-    PURCHASE_ORDER_MODEL = None
-    ESTIMATE_MODEL = None
-    CLOSING_ENTRY_MODEL = None
-    CLOSING_ENTRY_TRANSACTION_MODEL = None
 
-    ENTITY_DATA_GENERATOR = None
+    app_config = apps.get_app_config(app_label='django_ledger')
+
+    ENTITY_MODEL = 'entitymodel'
+    LEDGER_MODEL = 'ledgermodel'
+    JE_MODEL = 'journalentrymodel'
+    TRANSACTION_MODEL = 'transactionmodel'
+    ACCOUNT_MODEL = 'accountmodel'
+    COA_MODEL = 'chartofaccountmodel'
+
+    ENTITY_STATE_MODEL = 'entitystatemodel'
+    ENTITY_UNIT_MODEL = 'entityunitmodel'
+    CLOSING_ENTRY_MODEL = 'closingentrymodel'
+    CLOSING_ENTRY_TRANSACTION_MODEL = 'closingentrytransactionmodel'
+
+    BANK_ACCOUNT_MODEL = 'bankaccountmodel'
+    PURCHASE_ORDER_MODEL = 'purchaseordermodel'
+
+    CUSTOMER_MODEL = 'customermodel'
+    INVOICE_MODEL = 'invoicemodel'
+    BILL_MODEL = 'billmodel'
+    UOM_MODEL = 'unitofmeasuremodel'
+    VENDOR_MODEL = 'vendormodel'
+    ESTIMATE_MODEL = 'estimatemodel'
+    ITEM_MODEL = 'itemmodel'
+    ITEM_TRANSACTION_MODEL = 'itemtransactionmodel'
 
+    ENTITY_DATA_GENERATOR = None
     BALANCE_SHEET_REPORT_CLASS = None
     INCOME_STATEMENT_REPORT_CLASS = None
     CASH_FLOW_STATEMENT_REPORT_CLASS = None
 
+
+
     def get_entity_model(self):
-        if not self.ENTITY_MODEL:
-            from django_ledger.models import EntityModel
-            self.ENTITY_MODEL = EntityModel
-        return self.ENTITY_MODEL
+        return self.app_config.get_model(self.ENTITY_MODEL)
+
+    def get_entity_unit_model(self):
+        return self.app_config.get_model(self.ENTITY_UNIT_MODEL)
 
     def get_entity_state_model(self):
-        if not self.ENTITY_STATE_MODEL:
-            from django_ledger.models import EntityStateModel
-            self.ENTITY_STATE_MODEL = EntityStateModel
-        return self.ENTITY_STATE_MODEL
+        return self.app_config.get_model(self.ENTITY_STATE_MODEL)
 
     def get_bank_account_model(self):
-        if not self.BANK_ACCOUNT_MODEL:
-            from django_ledger.models import BankAccountModel
-            self.BANK_ACCOUNT_MODEL = BankAccountModel
-        return self.BANK_ACCOUNT_MODEL
+        return self.app_config.get_model(self.BANK_ACCOUNT_MODEL)
 
     def get_account_model(self):
-        if not self.ACCOUNT_MODEL:
-            from django_ledger.models import AccountModel
-            self.ACCOUNT_MODEL = AccountModel
-        return self.ACCOUNT_MODEL
+        return self.app_config.get_model(self.ACCOUNT_MODEL)
+
+    def get_coa_model(self):
+        return self.app_config.get_model(self.COA_MODEL)
 
     def get_txs_model(self):
-        if not self.TXS_MODEL:
-            from django_ledger.models import TransactionModel
-            self.TXS_MODEL = TransactionModel
-        return self.TXS_MODEL
+        return self.app_config.get_model(self.TRANSACTION_MODEL)
 
     def get_purchase_order_model(self):
-        if not self.PURCHASE_ORDER_MODEL:
-            from django_ledger.models import PurchaseOrderModel
-            self.PURCHASE_ORDER_MODEL = PurchaseOrderModel
-        return self.PURCHASE_ORDER_MODEL
+        return self.app_config.get_model(self.PURCHASE_ORDER_MODEL)
 
     def get_ledger_model(self):
-        if not self.LEDGER_MODEL:
-            from django_ledger.models import LedgerModel
-            self.LEDGER_MODEL = LedgerModel
-        return self.LEDGER_MODEL
-
-    def get_unit_model(self):
-        if not self.UNIT_MODEL:
-            from django_ledger.models import EntityUnitModel
-            self.UNIT_MODEL = EntityUnitModel
-        return self.UNIT_MODEL
+        self.get_entity_unit_model()
+        return self.app_config.get_model(self.LEDGER_MODEL)
 
     def get_journal_entry_model(self):
-        if not self.JE_MODEL:
-            from django_ledger.models import JournalEntryModel
-            self.JE_MODEL = JournalEntryModel
-        return self.JE_MODEL
+        return self.app_config.get_model(self.JE_MODEL)
 
     def get_item_model(self):
-        if not self.ITEM_MODEL:
-            from django_ledger.models import ItemModel
-            self.ITEM_MODEL = ItemModel
-        return self.ITEM_MODEL
+        return self.app_config.get_model(self.ITEM_MODEL)
 
     def get_item_transaction_model(self):
-        if not self.ITEM_TRANSACTION_MODEL:
-            from django_ledger.models import ItemTransactionModel
-            self.ITEM_TRANSACTION_MODEL = ItemTransactionModel
-        return self.ITEM_TRANSACTION_MODEL
+        return self.app_config.get_model(self.ITEM_TRANSACTION_MODEL)
 
     def get_customer_model(self):
-        if not self.CUSTOMER_MODEL:
-            from django_ledger.models import CustomerModel
-            self.CUSTOMER_MODEL = CustomerModel
-        return self.CUSTOMER_MODEL
+        return self.app_config.get_model(self.CUSTOMER_MODEL)
 
     def get_bill_model(self):
-        if not self.BILL_MODEL:
-            from django_ledger.models import BillModel
-            self.BILL_MODEL = BillModel
-        return self.BILL_MODEL
+        return self.app_config.get_model(self.BILL_MODEL)
 
     def get_invoice_model(self):
-        if not self.INVOICE_MODEL:
-            from django_ledger.models import InvoiceModel
-            self.INVOICE_MODEL = InvoiceModel
-        return self.INVOICE_MODEL
+        return self.app_config.get_model(self.INVOICE_MODEL)
 
     def get_uom_model(self):
-        if not self.UOM_MODEL:
-            from django_ledger.models import UnitOfMeasureModel
-            self.UOM_MODEL = UnitOfMeasureModel
-        return self.UOM_MODEL
+        return self.app_config.get_model(self.UOM_MODEL)
 
     def get_vendor_model(self):
-        if not self.VENDOR_MODEL:
-            from django_ledger.models import VendorModel
-            self.VENDOR_MODEL = VendorModel
-        return self.VENDOR_MODEL
+        return self.app_config.get_model(self.VENDOR_MODEL)
 
-    def get_transaction_model(self):
-        if not self.TRANSACTION_MODEL:
-            from django_ledger.models import TransactionModel
-            self.TRANSACTION_MODEL = TransactionModel
-        return self.TRANSACTION_MODEL
+    def get_estimate_model(self):
+        return self.app_config.get_model(self.ESTIMATE_MODEL)
 
-    def get_entity_unit_model(self):
-        if not self.ENTITY_UNIT_MODEL:
-            from django_ledger.models import EntityUnitModel
-            self.ENTITY_UNIT_MODEL = EntityUnitModel
-        return self.ENTITY_UNIT_MODEL
+    def get_closing_entry_model(self):
+        return self.app_config.get_model(self.CLOSING_ENTRY_MODEL)
 
-    def get_estimate_model(self):
-        if not self.ESTIMATE_MODEL:
-            from django_ledger.models import EstimateModel
-            self.ESTIMATE_MODEL = EstimateModel
-        return self.ESTIMATE_MODEL
+    def get_closing_entry_transaction_model(self):
+        return self.app_config.get_model(self.CLOSING_ENTRY_TRANSACTION_MODEL)
 
     def get_entity_data_generator(self):
         if not self.ENTITY_DATA_GENERATOR:
@@ -159,18 +172,6 @@ class LazyLoader:
             self.ENTITY_DATA_GENERATOR = EntityDataGenerator
         return self.ENTITY_DATA_GENERATOR
 
-    def get_closing_entry_model(self):
-        if not self.CLOSING_ENTRY_MODEL:
-            from django_ledger.models import ClosingEntryModel
-            self.CLOSING_ENTRY_MODEL = ClosingEntryModel
-        return self.CLOSING_ENTRY_MODEL
-
-    def get_closing_entry_transaction_model(self):
-        if not self.CLOSING_ENTRY_TRANSACTION_MODEL:
-            from django_ledger.models import ClosingEntryTransactionModel
-            self.CLOSING_ENTRY_TRANSACTION_MODEL = ClosingEntryTransactionModel
-        return self.CLOSING_ENTRY_TRANSACTION_MODEL
-
     def get_balance_sheet_report_class(self):
         if not self.BALANCE_SHEET_REPORT_CLASS:
             from django_ledger.report.balance_sheet import BalanceSheetReport

django_ledger/templates/django_ledger/account/account_list.html

@@ -15,9 +15,9 @@
                 {% accounts_table accounts %}
             </div>
             <div class="column is-12 has-text-centered">
-                <a href="{% url 'django_ledger:account-create' entity_slug=view.kwargs.entity_slug %}"
+                <a href="{{ coa_model.get_create_coa_account_url }}"
                    class="button is-success">{% trans 'Create Account' %}</a>
-                <a href="{% url 'django_ledger:coa-list' entity_slug=view.kwargs.entity_slug %}"
+                <a href="{{ coa_model.get_coa_list_url }}"
                    class="button is-dark">{% trans 'Back to CoA List' %}</a>
             </div>
         </div>

django_ledger/templates/django_ledger/account/account_update.html

@@ -4,22 +4,24 @@
 {% load django_ledger %}
 
 {% block view_content %}
-    <div class="columns is-centered">
-        <div class="column is-8-tablet is-6-desktop">
-            <div class="box">
-                <form action="{{ account.get_update_url }}" method="post">
-                    {% csrf_token %}
-                    {{ form.as_p }}
-                    <button class="button is-fullwidth is-primary my-3">{% trans 'Submit' %}</button>
-                    <a class="button is-fullwidth is-dark is-small my-3"
-                       href="{% url 'django_ledger:account-list' entity_slug=view.kwargs.entity_slug %}">
-                        {% trans 'Back' %}
-                    </a>
-                </form>
+    <div class="box">
+        <div class="columns is-centered is-multiline">
+            <div class="column is-12 has-text-centered">
+                <h2 class="is-size-3 has-text-info">{{ coa_model.name }}</h2>
+                <p class="is-size-4 has-text-danger is-italic">({{ coa_model.slug }})</p>
+                <h1 class="is-size-2">{{ account }}</h1>
+            </div>
+            <div class="column is-8-tablet is-6-desktop">
+                <div class="box">
+                    <form method="post">
+                        {% csrf_token %}
+                        {{ form.as_p }}
+                        <button type="submit" class="button is-primary is-fullwidth djetler_my_1">Submit</button>
+                        <a class="button is-dark is-small is-fullwidth"
+                           href="{% url 'django_ledger:account-list' entity_slug=view.kwargs.entity_slug %}">Back</a>
+                    </form>
+                </div>
             </div>
         </div>
     </div>
-{% endblock %}
-
-
-
+{% endblock %}

django_ledger/urls/chart_of_accounts.py

@@ -4,13 +4,13 @@ from django_ledger import views
 
 urlpatterns = [
     path('<slug:entity_slug>/list/',
-         views.ChartOfAccountsListView.as_view(),
+         views.ChartOfAccountModelListView.as_view(),
          name='coa-list'),
     path('<slug:entity_slug>/detail/<slug:coa_slug>/',
-         views.ChartOfAccountsListView.as_view(),
+         views.ChartOfAccountModelListView.as_view(),
          name='coa-detail'),
     path('<slug:entity_slug>/update/<slug:coa_slug>/',
-         views.ChartOfAccountsUpdateView.as_view(),
+         views.ChartOfAccountModelUpdateView.as_view(),
          name='coa-update'),
 
     # ACTIONS....

django_ledger/views/account.py

@@ -17,7 +17,7 @@ from django.views.generic.detail import SingleObjectMixin
 
 from django_ledger.forms.account import AccountModelUpdateForm, AccountModelCreateForm
 from django_ledger.io.io_core import get_localdate
-from django_ledger.models import lazy_loader
+from django_ledger.models import ChartOfAccountModel
 from django_ledger.models.accounts import AccountModel
 from django_ledger.views.mixins import (
     YearlyReportMixIn, MonthlyReportMixIn, QuarterlyReportMixIn, DjangoLedgerSecurityMixIn,
@@ -27,6 +27,18 @@ from django_ledger.views.mixins import (
 
 class BaseAccountModelViewQuerySetMixIn:
     queryset = None
+    coa_model = None
+
+    def get_coa_model(self) -> ChartOfAccountModel:
+        if self.coa_model is None:
+            coa_slug = self.kwargs.get('coa_slug')
+            if coa_slug:
+                coa_model_qs = self.AUTHORIZED_ENTITY_MODEL.chartofaccountmodel_set.all().active()
+                coa_model = get_object_or_404(coa_model_qs, slug__exact=coa_slug)
+            else:
+                coa_model = self.AUTHORIZED_ENTITY_MODEL.default_coa
+            self.coa_model = coa_model
+        return self.coa_model
 
     def get_queryset(self):
         if self.queryset is None:
@@ -39,8 +51,15 @@ class BaseAccountModelViewQuerySetMixIn:
             ).order_by(
                 'coa_model', 'role', 'code').not_coa_root()
 
-            if self.kwargs.get('coa_slug'):
-                qs = qs.filter(coa_model__slug__exact=self.kwargs.get('coa_slug'))
+            coa_slug = self.kwargs.get('coa_slug')
+            account_pk = self.kwargs.get('account_pk')
+
+            if coa_slug:
+                qs = qs.filter(coa_model__slug__exact=coa_slug)
+            elif account_pk:
+                qs = qs.filter(uuid__exact=account_pk)
+            else:
+                qs = qs.filter(coa_model__slug__exact=self.AUTHORIZED_ENTITY_MODEL.default_coa.slug)
 
             self.queryset = qs
         return super().get_queryset()
@@ -93,7 +112,8 @@ class AccountModelUpdateView(DjangoLedgerSecurityMixIn, BaseAccountModelViewQuer
         # Set here because user_model is needed to instantiate an instance of MoveNodeForm (AccountModelUpdateForm)
         account_model.USER_MODEL = self.request.user
         return AccountModelUpdateForm(
-            entity_slug=self.kwargs['entity_slug'],
+            entity_model=self.AUTHORIZED_ENTITY_MODEL,
+            coa_model=self.get_coa_model(),
             user_model=self.request.user,
             **self.get_form_kwargs()
         )
@@ -119,7 +139,8 @@ class AccountModelCreateView(DjangoLedgerSecurityMixIn, BaseAccountModelViewQuer
     def get_form(self, form_class=None):
         return AccountModelCreateForm(
             user_model=self.request.user,
-            entity_slug=self.kwargs['entity_slug'],
+            entity_model=self.AUTHORIZED_ENTITY_MODEL,
+            coa_model=self.get_coa_model(),
             **self.get_form_kwargs()
         )
 
@@ -129,20 +150,25 @@ class AccountModelCreateView(DjangoLedgerSecurityMixIn, BaseAccountModelViewQuer
         return context
 
     def form_valid(self, form):
-        EntityModel = lazy_loader.get_entity_model()
-        entity_model_qs = EntityModel.objects.for_user(user_model=self.request.user).select_related('default_coa')
-        entity_model: EntityModel = get_object_or_404(entity_model_qs, slug__exact=self.kwargs['entity_slug'])
+        entity_model = self.AUTHORIZED_ENTITY_MODEL
         account_model: AccountModel = form.save(commit=False)
 
         if not entity_model.has_default_coa():
             entity_model.create_chart_of_accounts(assign_as_default=True, commit=True)
 
-        coa_model = entity_model.default_coa
-        coa_model.allocate_account(account_model=account_model)
+        coa_model = self.get_coa_model()
+        coa_model.insert_account(account_model=account_model)
         return HttpResponseRedirect(self.get_success_url())
 
     def get_success_url(self):
         entity_slug = self.kwargs.get('entity_slug')
+        coa_slug = self.kwargs.get('coa_slug')
+        if coa_slug:
+            return reverse('django_ledger:account-list-coa',
+                           kwargs={
+                               'entity_slug': entity_slug,
+                               'coa_slug': coa_slug
+                           })
         return reverse('django_ledger:account-list',
                        kwargs={
                            'entity_slug': entity_slug,

django_ledger/views/coa.py

@@ -18,21 +18,21 @@ from django_ledger.models.coa import ChartOfAccountModel
 from django_ledger.views.mixins import DjangoLedgerSecurityMixIn
 
 
-class ChartOfAccountsModelModelViewQuerySetMixIn:
+class ChartOfAccountModelModelBaseViewMixIn(DjangoLedgerSecurityMixIn):
     queryset = None
 
     def get_queryset(self):
         if self.queryset is None:
-            self.queryset = ChartOfAccountModel.objects.for_entity(
-                entity_slug=self.kwargs['entity_slug'],
-                user_model=self.request.user,
-            ).select_related('entity').order_by('-updated')
+            entity_model = self.get_authorized_entity_instance()
+            self.queryset = entity_model.chartofaccountmodel_set.all().select_related(
+                'entity').order_by('-updated')
         return super().get_queryset()
 
 
-class ChartOfAccountsListView(DjangoLedgerSecurityMixIn, ChartOfAccountsModelModelViewQuerySetMixIn, ListView):
+class ChartOfAccountModelListView(ChartOfAccountModelModelBaseViewMixIn, ListView):
     template_name = 'django_ledger/chart_of_accounts/coa_list.html'
     extra_context = {
+        'page_title': _('Chart of Account List'),
         'header_title': _('Chart of Account List'),
     }
     context_object_name = 'coa_list'
@@ -65,7 +65,7 @@ class ChartOfAccountsListView(DjangoLedgerSecurityMixIn, ChartOfAccountsModelMod
         )
 
 
-class ChartOfAccountsUpdateView(DjangoLedgerSecurityMixIn, ChartOfAccountsModelModelViewQuerySetMixIn, UpdateView):
+class ChartOfAccountModelUpdateView(ChartOfAccountModelModelBaseViewMixIn, UpdateView):
     context_object_name = 'coa'
     slug_url_kwarg = 'coa_slug'
     template_name = 'django_ledger/chart_of_accounts/coa_update.html'
@@ -86,9 +86,8 @@ class ChartOfAccountsUpdateView(DjangoLedgerSecurityMixIn, ChartOfAccountsModelM
 
 
 # todo: centralize this functionality into a separate class for ALL Action views...
-class CharOfAccountModelActionView(DjangoLedgerSecurityMixIn,
+class CharOfAccountModelActionView(ChartOfAccountModelModelBaseViewMixIn,
                                    RedirectView,
-                                   ChartOfAccountsModelModelViewQuerySetMixIn,
                                    SingleObjectMixin):
     http_method_names = ['get']
     slug_url_kwarg = 'coa_slug'

django_ledger/views/mixins.py

@@ -299,14 +299,6 @@ class DjangoLedgerSecurityMixIn(PermissionRequiredMixin):
     def get_login_url(self):
         return reverse('django_ledger:login')
 
-    def get_authorized_entity_queryset(self):
-        return EntityModel.objects.for_user(
-            user_model=self.request.user).only(
-            'uuid', 'slug', 'name', 'default_coa', 'admin')
-
-    def get_authorized_entity_instance(self) -> Optional[EntityModel]:
-        return self.AUTHORIZED_ENTITY_MODEL
-
     def has_permission(self):
         if self.request.user.is_superuser:
             if 'entity_slug' in self.kwargs:
@@ -329,6 +321,16 @@ class DjangoLedgerSecurityMixIn(PermissionRequiredMixin):
             return True
         return False
 
+    def get_authorized_entity_queryset(self):
+        return EntityModel.objects.for_user(
+            user_model=self.request.user).only(
+            'uuid', 'slug', 'name', 'default_coa', 'admin')
+
+    def get_authorized_entity_instance(self) -> Optional[EntityModel]:
+        if self.AUTHORIZED_ENTITY_MODEL is None:
+            raise Http404()
+        return self.AUTHORIZED_ENTITY_MODEL
+
 
 class EntityUnitMixIn:
     UNIT_SLUG_KWARG = 'unit_slug'