|
26 | 26 | from django_ledger.io.io_context import (RoleContextManager, GroupContextManager, ActivityContextManager, |
27 | 27 | BalanceSheetStatementContextManager, IncomeStatementContextManager, |
28 | 28 | CashFlowStatementContextManager) |
29 | | -from django_ledger.io.io_digest import IODigest |
| 29 | +from django_ledger.io.io_digest import IODigestContextManager |
30 | 30 | from django_ledger.io.ratios import FinancialRatioManager |
31 | 31 | from django_ledger.models.utils import lazy_loader |
32 | 32 | from django_ledger.settings import (DJANGO_LEDGER_TRANSACTION_MAX_TOLERANCE, |
@@ -169,7 +169,7 @@ def digest_balance_sheet(self, |
169 | 169 | to_date: Union[date, datetime], |
170 | 170 | user_model: UserModel, |
171 | 171 | txs_queryset: Optional[QuerySet] = None, |
172 | | - **kwargs: Dict) -> Union[IODigest, Tuple[QuerySet, Dict]]: |
| 172 | + **kwargs: Dict) -> Union[IODigestContextManager, Tuple[QuerySet, Dict]]: |
173 | 173 | return self.digest( |
174 | 174 | user_model=user_model, |
175 | 175 | to_date=to_date, |
@@ -217,7 +217,7 @@ def digest_income_statement(self, |
217 | 217 | to_date: Union[date, datetime], |
218 | 218 | user_model: Optional[UserModel] = None, |
219 | 219 | txs_queryset: Optional[QuerySet] = None, |
220 | | - **kwargs) -> Union[IODigest, Tuple[QuerySet, Dict]]: |
| 220 | + **kwargs) -> Union[IODigestContextManager, Tuple[QuerySet, Dict]]: |
221 | 221 | return self.digest( |
222 | 222 | user_model=user_model, |
223 | 223 | from_date=from_date, |
@@ -267,7 +267,7 @@ def digest_cash_flow_statement(self, |
267 | 267 | to_date: Union[date, datetime], |
268 | 268 | user_model: UserModel, |
269 | 269 | txs_queryset: Optional[QuerySet] = None, |
270 | | - **kwargs) -> Union[IODigest, Tuple[QuerySet, Dict]]: |
| 270 | + **kwargs) -> Union[IODigestContextManager, Tuple[QuerySet, Dict]]: |
271 | 271 | return self.digest( |
272 | 272 | user_model=user_model, |
273 | 273 | from_date=from_date, |
@@ -611,7 +611,7 @@ def digest(self, |
611 | 611 | balance_sheet_statement: bool = False, |
612 | 612 | income_statement: bool = False, |
613 | 613 | cash_flow_statement: bool = False, |
614 | | - ) -> Union[Tuple, IODigest]: |
| 614 | + ) -> Union[Tuple, IODigestContextManager]: |
615 | 615 |
|
616 | 616 | if balance_sheet_statement: |
617 | 617 | from_date = None |
@@ -704,7 +704,7 @@ def digest(self, |
704 | 704 | io_data = cfs.digest() |
705 | 705 |
|
706 | 706 | if as_io_digest: |
707 | | - return IODigest(io_data=io_data) |
| 707 | + return IODigestContextManager(io_data=io_data) |
708 | 708 |
|
709 | 709 | if not digest_name: |
710 | 710 | digest_name = 'tx_digest' |
|
0 commit comments