NPO-Accounting/conservancy_beancount Changeset - e3dceb601c00

Changeset - e3dceb601c00

Parent rev.

Child rev.

[Not reviewed]

0 3 0

Brett Smith - 4 years ago 2020-06-11 14:46:06
brettcsmith@brettcsmith.org

filters: Add iter_unique() function.

3 files changed with 21 insertions and 11 deletions:

conservancy_beancount/filters.py

conservancy_beancount/reports/accrual.py

tests/test_filters.py

0 comments (0 inline, 0 general)

conservancy_beancount/filters.py

➞

Show inline comments

@@ ... / @@ -26,5 +26,9 @@ from typing import ( @@
     cast,
     Hashable,
     Iterable,
     Iterator,
     Optional,
     Pattern,
     Set,
     TypeVar,
     Union,
@@ ... / @@ -39,2 +43,4 @@ from .beancount_types import ( @@
 # Saying Optional works around <https://github.com/python/mypy/issues/8768>.
 HashT = TypeVar('HashT', bound=Optional[Hashable])
 Postings = Iterable[data.Posting]
@@ ... @@
 def iter_unique(seq: Iterable[HashT]) -> Iterator[HashT]:
     seen: Set[HashT] = set()
     for item in seq:
         if item not in seen:
             seen.add(item)
             yield item
 def remove_opening_balance_txn(entries: Entries) -> Optional[Transaction]:

conservancy_beancount/reports/accrual.py

➞

Show inline comments

@@ ... / @@ -209,13 +209,7 @@ class AccrualPostings(core.RelatedPostings): @@
     def entities(self, pred: Callable[[data.Posting], bool]=bool) -> Iterator[MetaValue]:
         seen: Set[MetaValue] = set()
         for post in self:
             if pred(post):
                 try:
                     entity = post.meta['entity']
                 except KeyError:
                     pass
                 else:
                     if entity not in seen:
                         yield entity
                         seen.add(entity)
         return filters.iter_unique(
             post.meta['entity']
             for post in self
             if pred(post) and 'entity' in post.meta
+        )

tests/test_filters.py

➞

Show inline comments

@@ ... / @@ -185 +185,4 @@ def test_audit_date(entry): @@
         assert actual == entry.date
 def test_iter_unique():
     assert list(filters.iter_unique('1213231')) == list('123')

0 comments (0 inline, 0 general)