NPO-Accounting/conservancy_beancount Changeset - 96a363633f5e

Changeset - 96a363633f5e

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 4 years ago 2020-04-21 15:58:28
brettcsmith@brettcsmith.org

books: Add Loader.load_fy_range() method.

2 files changed with 28 insertions and 0 deletions:

conservancy_beancount/books.py

tests/test_books_loader.py

0 comments (0 inline, 0 general)

conservancy_beancount/books.py

➞

Show inline comments

@@ ... / @@ -15,19 +15,25 @@ @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 import datetime
 from pathlib import Path
 from beancount import loader as bc_loader
 from typing import (
     Any,
     Iterable,
     Mapping,
     NamedTuple,
     Optional,
     Union,
+)
 from .beancount_types import (
     LoadResult,
+)
 PathLike = Union[str, Path]
 PluginsSpec = Mapping[str, Optional[str]]
 Year = Union[int, datetime.date]
 class FiscalYear(NamedTuple):
@@ ... / @@ -144,6 +150,21 @@ class Loader: @@
             return ''
         return '\n'.join([
             *(self._format_plugin(name, opts) for name, opts in plugins.items()),
             books_start,
             *(self._format_include(year) for year in years),
         ])
     def load_fy_range(self,
                       from_fy: Year,
                       to_fy: Optional[Year]=None,
                       plugins: Optional[PluginsSpec]=None,
     ) -> LoadResult:
         """Load books for a range of fiscal years
         This is a convenience wrapper to call
         self.fy_range_string(from_fy, to_fy, plugins)
         and load the result with beancount.loader.load_string.
         """
         return bc_loader.load_string(  # type:ignore[no-untyped-call, no-any-return]
             self.fy_range_string(from_fy, to_fy, plugins),
+        )

tests/test_books_loader.py

➞

Show inline comments

@@ ... / @@ -11,12 +11,14 @@ @@
 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 # GNU Affero General Public License for more details.
+#
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 import hashlib
 from datetime import date
 from pathlib import Path
 import pytest
 from . import testutil
@@ ... @@
 def test_fy_range_string_plugins_override(conservancy_loader, plugins):
     expected = expect_string([2019, 2020], plugins)
     assert conservancy_loader.fy_range_string(2019, 2020, plugins) == expected
 def test_fy_range_string_empty_range(conservancy_loader):
     assert conservancy_loader.fy_range_string(2020, 2019) == ''
 def test_load_fy_range_empty(conservancy_loader):
     entries, errors, options_map = conservancy_loader.load_fy_range(2020, 2019)
     assert not entries
     assert options_map.get('input_hash') == hashlib.md5().hexdigest()

0 comments (0 inline, 0 general)