NPO-Accounting/conservancy_beancount Changeset - f76fa35fad2a

Changeset - f76fa35fad2a

Parent rev.

Child rev.

[Not reviewed]

0 3 0

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

reports: RelatedPostings.all_meta_links() returns an iterator.

This preserves order.

3 files changed with 21 insertions and 8 deletions:

conservancy_beancount/reports/accrual.py

conservancy_beancount/reports/core.py

tests/test_reports_related_postings.py

0 comments (0 inline, 0 general)

conservancy_beancount/reports/accrual.py

➞

Show inline comments

@@ ... / @@ -554,5 +554,5 @@ class OutgoingReport(BaseReport): @@
             balance_s = f'{raw_balance} ({balance_s})'
         contract_links = posts.all_meta_links('contract')
+        contract_links = list(posts.all_meta_links('contract'))
         if contract_links:
             contract_s = ' , '.join(self.rt_wrapper.iter_urls(

conservancy_beancount/reports/core.py

➞

Show inline comments

@@ ... / @@ -39,4 +39,5 @@ from beancount.core import amount as bc_amount @@
 from .. import data
 from .. import filters
 from typing import (
@@ ... / @@ -294,12 +295,13 @@ class RelatedPostings(Sequence[data.Posting]): @@
         return len(self._postings)
     def all_meta_links(self, key: MetaKey) -> Set[str]:
         retval: Set[str] = set()
     def _all_meta_links(self, key: MetaKey) -> Iterator[str]:
         for post in self:
             try:
-                retval.update(post.meta.get_links(key))
+                yield from post.meta.get_links(key)
             except TypeError:
                 pass
         return retval
     def all_meta_links(self, key: MetaKey) -> Iterator[str]:
         return filters.iter_unique(self._all_meta_links(key))
     def iter_with_balance(self) -> Iterator[Tuple[data.Posting, Balance]]:

tests/test_reports_related_postings.py

➞

Show inline comments

@@ ... / @@ -257,5 +257,5 @@ def test_all_meta_links_zero(count): @@
         for post in postings
+    )
-    assert related.all_meta_links('approval') == set()
+    assert next(related.all_meta_links('approval'), None) is None
 def test_all_meta_links_singletons():
@@ ... / @@ -271,5 +271,5 @@ def test_all_meta_links_singletons(): @@
         for post in postings
+    )
     assert related.all_meta_links('statement') == testutil.LINK_METADATA_STRINGS
+    assert set(related.all_meta_links('statement')) == testutil.LINK_METADATA_STRINGS
 def test_all_meta_links_multiples():
@@ ... / @@ -282,5 +282,16 @@ def test_all_meta_links_multiples(): @@
         for post in postings
+    )
     assert related.all_meta_links('approval') == testutil.LINK_METADATA_STRINGS
     assert set(related.all_meta_links('approval')) == testutil.LINK_METADATA_STRINGS
 def test_all_meta_links_preserves_order():
     postings = (
         testutil.Posting('Income:Donations', -10, approval=c)
         for c in '121323'
+    )
     related = core.RelatedPostings(
         post._replace(meta=data.Metadata(post.meta))
         for post in postings
+    )
     assert list(related.all_meta_links('approval')) == list('123')
 def test_group_by_meta_zero():

0 comments (0 inline, 0 general)