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

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

@@ -138,6 +138,15 @@ class JournalEntryModelManager(models.Manager):
     EntityModel and authenticated UserModel.
     """
 
+    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):
         """
         Fetches a QuerySet of JournalEntryModels associated with a specific EntityModel & UserModel.
@@ -161,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):
@@ -345,8 +345,18 @@ class JournalEntryModelAbstract(CreateUpdateMixIn):
 
     def __str__(self):
         if self.je_number:
-            return 'JE: {x1} - Desc: {x2}'.format(x1=self.je_number, x2=self.description)
-        return 'JE ID: {x1} - Desc: {x2}'.format(x1=self.pk, x2=self.description)
+            return 'JE: {x1} (posted={p}, locked={l}) - Desc: {x2}'.format(
+                x1=self.je_number,
+                x2=self.description,
+                p=self.posted,
+                l=self.locked
+            )
+        return 'JE ID: {x1} (posted={p}, locked={l}) - Desc: {x2}'.format(
+            x1=self.pk,
+            x2=self.description,
+            p=self.posted,
+            l=self.locked
+        )
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -1011,7 +1021,7 @@ class JournalEntryModelAbstract(CreateUpdateMixIn):
         """
         if self.can_generate_je_number():
 
-            with transaction.atomic(durable=True):
+            with transaction.atomic():
 
                 state_model = None
                 while not state_model:
@@ -1198,12 +1208,45 @@ class JournalEntryModelAbstract(CreateUpdateMixIn):
 
         return reverse('django_ledger:je-detail',
                        kwargs={
-                           'je_pk': self.id,
+                           'je_pk': self.uuid,
                            'ledger_pk': self.ledger_id,
-                           # pylint: disable=no-member
                            'entity_slug': self.ledger.entity.slug
                        })
 
+    def get_detail_url(self) -> str:
+        """
+        Determines the update URL of the LedgerModel instance.
+        Results in additional Database query if entity field is not selected in QuerySet.
+
+        Returns
+        -------
+        str
+            URL as a string.
+        """
+        return reverse('django_ledger:je-detail',
+                       kwargs={
+                           'entity_slug': self.ledger.entity.slug,
+                           'ledger_pk': self.ledger_id,
+                           'je_pk': self.uuid
+                       })
+
+    def get_detail_txs_url(self) -> str:
+        """
+        Determines the update URL of the LedgerModel instance.
+        Results in additional Database query if entity field is not selected in QuerySet.
+
+        Returns
+        -------
+        str
+            URL as a string.
+        """
+        return reverse('django_ledger:je-detail-txs',
+                       kwargs={
+                           'entity_slug': self.ledger.entity.slug,
+                           'ledger_pk': self.ledger_id,
+                           'je_pk': self.uuid
+                       })
+
     def get_unlock_url(self):
         return reverse('django_ledger:je-mark-as-unlocked',
                        kwargs={

django_ledger/models/ledger.py

@@ -38,7 +38,7 @@ from django.db.models import Q, Min, F, Count
 from django.urls import reverse
 from django.utils.translation import gettext_lazy as _
 
-from django_ledger.io import IOMixIn
+from django_ledger.io.io_core import IOMixIn
 from django_ledger.models import lazy_loader
 from django_ledger.models.mixins import CreateUpdateMixIn
 
@@ -54,6 +54,28 @@ class LedgerModelQuerySet(models.QuerySet):
     Custom defined LedgerModel QuerySet.
     """
 
+    def locked(self):
+        """
+        Filters the QuerySet to only locked LedgerModel.
+
+        Returns
+        -------
+        LedgerModelQuerySet
+            A QuerySet with applied filters.
+        """
+        return self.filter(locked=True)
+
+    def unlocked(self):
+        """
+        Filters the QuerySet to only un-locked LedgerModel.
+
+        Returns
+        -------
+        LedgerModelQuerySet
+            A QuerySet with applied filters.
+        """
+        return self.filter(locked=False)
+
     def posted(self):
         """
         Filters the QuerySet to only posted LedgerModel.
@@ -65,6 +87,17 @@ class LedgerModelQuerySet(models.QuerySet):
         """
         return self.filter(posted=True)
 
+    def unposted(self):
+        """
+        Filters the QuerySet to only un-posted LedgerModel.
+
+        Returns
+        -------
+        LedgerModelQuerySet
+            A QuerySet with applied filters.
+        """
+        return self.filter(posted=True)
+
     def hidden(self):
         return self.filter(hidden=True)
 
@@ -73,7 +106,8 @@ class LedgerModelQuerySet(models.QuerySet):
 
     def current(self):
         return self.filter(
-            earliest_timestamp__gt=F('entity__last_closing_date')
+            Q(earliest_timestamp__gt=F('entity__last_closing_date'))
+            | Q(earliest_timestamp__isnull=True)
         )
 
 
@@ -90,6 +124,15 @@ class LedgerModelManager(models.Manager):
                                    filter=Q(journal_entries__posted=True)),
         )
 
+    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, user_model):
         """
         Returns a QuerySet of LedgerModels associated with a specific EntityModel & UserModel.
@@ -107,21 +150,13 @@ class LedgerModelManager(models.Manager):
         LedgerModelQuerySet
             A Filtered LedgerModelQuerySet.
         """
-        qs = self.get_queryset()
+        qs = self.for_user(user_model)
         if isinstance(entity_slug, lazy_loader.get_entity_model()):
             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)
         )
 
 
@@ -135,6 +170,8 @@ class LedgerModelAbstract(CreateUpdateMixIn, IOMixIn):
         This is a unique primary key generated for the table. The default value of this field is uuid4().
     name: str
         Human-readable name of the LedgerModel. Maximum 150 characters.
+    ledger_xid: str
+        A unique user-defined identifier for the LedgerModel. Unique for the Entity Model.
     entity: EntityModel
         The EntityModel associated with the LedgerModel instance.
     posted: bool
@@ -146,6 +183,9 @@ class LedgerModelAbstract(CreateUpdateMixIn, IOMixIn):
     """
     _WRAPPED_MODEL_KEY = 'wrapped_model'
     uuid = models.UUIDField(default=uuid4, editable=False, primary_key=True)
+    ledger_xid = models.SlugField(allow_unicode=True, max_length=150, null=True, blank=True,
+                                  verbose_name=_('Ledger Slug'),
+                                  help_text=_('User Defined Ledger ID'))
     name = models.CharField(max_length=150, null=True, blank=True, verbose_name=_('Ledger Name'))
 
     # todo: rename to entity_model...
@@ -173,6 +213,9 @@ class LedgerModelAbstract(CreateUpdateMixIn, IOMixIn):
             models.Index(fields=['entity', 'posted']),
             models.Index(fields=['entity', 'locked']),
         ]
+        unique_together = [
+            ('entity', 'ledger_xid')
+        ]
 
     def __str__(self):
         return self.name
@@ -192,6 +235,7 @@ class LedgerModelAbstract(CreateUpdateMixIn, IOMixIn):
                 return getattr(self, model_id)
             except ObjectDoesNotExist:
                 pass
+        return False
 
     def remove_wrapped_model_info(self):
         if self.has_wrapped_model_info():
@@ -375,7 +419,7 @@ class LedgerModelAbstract(CreateUpdateMixIn, IOMixIn):
         if not self.can_post():
             if raise_exception:
                 raise LedgerModelValidationError(
-                    message=_(f'Ledger {self.uuid} cannot be posted.')
+                    message=_(f'Ledger {self.name} cannot be posted. UUID: {self.uuid}')
                 )
             return
         self.posted = True
@@ -417,7 +461,7 @@ class LedgerModelAbstract(CreateUpdateMixIn, IOMixIn):
                 'updated'
             ])
 
-    def lock(self, commit: bool = False, **kwargs):
+    def lock(self, commit: bool = False, raise_exception: bool = True, **kwargs):
         """
         Locks the LedgerModel.
 
@@ -425,14 +469,22 @@ class LedgerModelAbstract(CreateUpdateMixIn, IOMixIn):
         ----------
         commit: bool
             If True, saves the LedgerModel instance instantly. Defaults to False.
+        raise_exception: bool
+            Raises LedgerModelValidationError if locking not allowed.
         """
-        if self.can_lock():
-            self.locked = True
-            if commit:
-                self.save(update_fields=[
-                    'locked',
-                    'updated'
-                ])
+
+        if not self.can_lock():
+            if raise_exception:
+                raise LedgerModelValidationError(
+                    message=_(f'Ledger {self.name} cannot be locked. UUID: {self.uuid}')
+                )
+            return
+        self.locked = True
+        if commit:
+            self.save(update_fields=[
+                'locked',
+                'updated'
+            ])
 
     def lock_journal_entries(self, commit: bool = True, **kwargs):
         je_model_qs = self.journal_entries.unlocked()
@@ -493,9 +545,8 @@ class LedgerModelAbstract(CreateUpdateMixIn, IOMixIn):
         str
             URL as a string.
         """
-        return reverse('django_ledger:ledger-detail',
+        return reverse(viewname='django_ledger:je-list',
                        kwargs={
-                           # pylint: disable=no-member
                            'entity_slug': self.entity.slug,
                            'ledger_pk': self.uuid
                        })
@@ -531,6 +582,21 @@ class LedgerModelAbstract(CreateUpdateMixIn, IOMixIn):
                            'ledger_pk': self.uuid
                        })
 
+    def get_list_url(self) -> str:
+        """
+        Determines the list URL of the LedgerModel instances.
+        Results in additional Database query if entity field is not selected in QuerySet.
+
+        Returns
+        -------
+        str
+            URL as a string.
+        """
+        return reverse('django_ledger:ledger-list',
+                       kwargs={
+                           'entity_slug': self.entity.slug
+                       })
+
     def get_balance_sheet_url(self):
         return reverse(
             viewname='django_ledger:ledger-bs',

django_ledger/models/mixins.py

@@ -13,7 +13,7 @@ from collections import defaultdict
 from datetime import timedelta, date, datetime
 from decimal import Decimal
 from itertools import groupby
-from typing import Optional, Union, Dict, List
+from typing import Optional, Union, Dict
 from uuid import UUID
 
 from django.conf import settings
@@ -27,8 +27,7 @@ from django.utils.timezone import localdate, localtime
 from django.utils.translation import gettext_lazy as _
 from markdown import markdown
 
-from django_ledger.io import (check_tx_balance, ASSET_CA_CASH, ASSET_CA_PREPAID, LIABILITY_CL_DEFERRED_REVENUE,
-                              validate_io_date)
+from django_ledger.io.io_core import validate_io_date, check_tx_balance
 from django_ledger.models.utils import lazy_loader
 
 logging.basicConfig(format='%(asctime)s %(message)s', datefmt='%m/%d/%Y %I:%M:%S %p')

django_ledger/models/purchase_order.py

@@ -101,6 +101,15 @@ class PurchaseOrderModelManager(models.Manager):
     A custom defined PurchaseOrderModel Manager.
     """
 
+    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, user_model) -> PurchaseOrderModelQuerySet:
         """
         Fetches a QuerySet of PurchaseOrderModel associated with a specific EntityModel & UserModel.
@@ -111,15 +120,10 @@ class PurchaseOrderModelManager(models.Manager):
         PurchaseOrderModelQuerySet
             A PurchaseOrderModelQuerySet with applied filters.
         """
-        qs = self.get_queryset()
+        qs = self.for_user(user_model)
         if isinstance(entity_slug, EntityModel):
             qs = qs.filter(entity=entity_slug)
-        elif isinstance(entity_slug, str):
-            qs = qs.filter(entity__slug__exact=entity_slug)
-        return qs.filter(
-            Q(entity__admin=user_model) |
-            Q(entity__managers__in=[user_model])
-        )
+        return qs.filter(entity__slug__exact=entity_slug)
 
 
 class PurchaseOrderModelAbstract(CreateUpdateMixIn,

django_ledger/models/transactions.py

@@ -26,7 +26,7 @@ from django.db import models
 from django.db.models import Q, QuerySet
 from django.utils.translation import gettext_lazy as _
 
-from django_ledger.io import validate_io_date
+from django_ledger.io.io_core import validate_io_date
 from django_ledger.models.accounts import AccountModel
 from django_ledger.models.bill import BillModel
 from django_ledger.models.entity import EntityModel
@@ -220,6 +220,8 @@ class TransactionModelAdmin(models.Manager):
             Returns a TransactionModelQuerySet with applied filters.
         """
         qs = self.get_queryset()
+        if user_model.is_superuser:
+            return qs
         return qs.filter(
             Q(journal_entry__ledger__entity__admin=user_model) |
             Q(journal_entry__ledger__entity__managers__in=[user_model])

django_ledger/models/unit.py

@@ -34,7 +34,7 @@ from django.utils.text import slugify
 from django.utils.translation import gettext_lazy as _
 from treebeard.mp_tree import MP_Node, MP_NodeManager, MP_NodeQuerySet
 
-from django_ledger.io.io_mixin import IOMixIn
+from django_ledger.io.io_core import IOMixIn
 from django_ledger.models import lazy_loader
 from django_ledger.models.mixins import CreateUpdateMixIn, SlugNameMixIn
 
@@ -53,6 +53,15 @@ class EntityUnitModelQuerySet(MP_NodeQuerySet):
 
 class EntityUnitModelManager(MP_NodeManager):
 
+    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: str, user_model):
         """
         Fetches a QuerySet of EntityUnitModels associated with a specific EntityModel & UserModel.
@@ -76,23 +85,14 @@ class EntityUnitModelManager(MP_NodeManager):
         EntityUnitModelQuerySet
             Returns a EntityUnitModelQuerySet with applied filters.
         """
-        qs = self.get_queryset()
+        qs = self.for_user(user_model)
         if isinstance(entity_slug, lazy_loader.get_entity_model()):
             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)
         )
 
 
@@ -212,6 +212,15 @@ class EntityUnitModelAbstract(MP_Node,
             self.slug = unit_slug
         return self.slug
 
+    def get_absolute_url(self):
+        return reverse(
+            viewname='django_ledger:unit-detail',
+            kwargs={
+                'entity_slug': self.entity.slug,
+                'unit_slug': self.slug
+            }
+        )
+
 
 class EntityUnitModel(EntityUnitModelAbstract):
     """

django_ledger/models/vendor.py

@@ -91,6 +91,15 @@ class VendorModelManager(models.Manager):
     Custom defined VendorModel Manager, which defines many methods for initial query of the Database.
     """
 
+    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) -> VendorModelQuerySet:
         """
         Fetches a QuerySet of VendorModel associated with a specific EntityModel & UserModel.
@@ -114,21 +123,13 @@ class VendorModelManager(models.Manager):
         VendorModelQuerySet
             A filtered VendorModel 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/report/cash_flow_statement.py

@@ -1,7 +1,7 @@
 from datetime import datetime, date
 from typing import Optional, Dict, Union
 
-from django_ledger.io import IODigestContextManager
+from django_ledger.io.io_digest import IODigestContextManager
 from django_ledger.report.core import BaseReportSupport, PDFReportValidationError
 from django_ledger.settings import DJANGO_LEDGER_CURRENCY_SYMBOL
 from django_ledger.templatetags.django_ledger import currency_format

django_ledger/report/core.py

@@ -3,8 +3,9 @@ from typing import Optional, Dict
 from django.contrib.staticfiles import finders
 from django.core.exceptions import ValidationError
 
-from django_ledger.io import IODigestContextManager
-from django_ledger.models import LedgerModel, EntityUnitModel
+from django_ledger.io.io_digest import IODigestContextManager
+from django_ledger.models.ledger import LedgerModel
+from django_ledger.models.unit import EntityUnitModel
 from django_ledger.settings import DJANGO_LEDGER_PDF_SUPPORT_ENABLED
 from django_ledger.templatetags.django_ledger import currency_symbol, currency_format
 

django_ledger/templates/django_ledger/journal_entry/includes/card_journal_entry.html

@@ -35,7 +35,7 @@
     <div class="card-footer">
         {% if journal_entry.can_lock %}
             <a href="{{ journal_entry.get_lock_url }}" class="card-footer-item has-text-success has-text-weight-bold">{% trans 'Lock' %}</a>
-        {% endif %}t
+        {% endif %}
         {% if journal_entry.can_unlock %}
             <a href="{{ journal_entry.get_unlock_url }}" class="card-footer-item has-text-warning has-text-weight-bold">{% trans 'UnLock' %}</a>
         {% endif %}

django_ledger/templates/django_ledger/journal_entry/je_list.html

@@ -4,6 +4,9 @@
 {% load django_ledger %}
 
 {% block view_content %}
+    
+{#    {% include 'django_ledger/journal_entry/includes/card_journal_entry.html' je_model= %}#}
+    
     <div class="card">
         <div class="card-content">
             <div class="level">

django_ledger/templatetags/django_ledger.py

@@ -19,7 +19,7 @@ from django_ledger import __version__
 from django_ledger.forms.app_filters import EntityFilterForm, ActivityFilterForm
 from django_ledger.forms.feedback import BugReportForm, RequestNewFeatureForm
 from django_ledger.io import CREDIT, DEBIT, ROLES_ORDER_ALL
-from django_ledger.io.io_mixin import validate_activity
+from django_ledger.io.io_core import validate_activity
 from django_ledger.models import TransactionModel, BillModel, InvoiceModel, EntityUnitModel
 from django_ledger.settings import (
     DJANGO_LEDGER_FINANCIAL_ANALYSIS, DJANGO_LEDGER_CURRENCY_SYMBOL,

django_ledger/tests/base.py

@@ -12,7 +12,7 @@ from django.test import TestCase
 from django.test.client import Client
 from django.utils.timezone import get_default_timezone
 
-from django_ledger.io.data_generator import EntityDataGenerator
+from django_ledger.io.io_generator import EntityDataGenerator
 from django_ledger.models.entity import EntityModel, EntityModelQuerySet
 
 UserModel = get_user_model()