NPO-Accounting/conservancy_beancount Changeset - 4420873c967e

Changeset - 4420873c967e

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 4 years ago 2020-04-22 14:34:55
brettcsmith@brettcsmith.org

filters: Add filter_meta_match function.

2 files changed with 27 insertions and 0 deletions:

conservancy_beancount/filters.py

tests/test_filters.py

0 comments (0 inline, 0 general)

conservancy_beancount/filters.py

➞

Show inline comments

@@ ... / @@ -14,10 +14,14 @@ @@
 # 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 re
 from . import data
 from typing import (
     Iterable,
     Pattern,
     Union,
+)
 from .beancount_types import (
     MetaKey,
@@ ... / @@ -25,6 +29,7 @@ from .beancount_types import ( @@
+)
 Postings = Iterable[data.Posting]
 Regexp = Union[str, Pattern]
 def filter_meta_equal(postings: Postings, key: MetaKey, value: MetaValue) -> Postings:
     for post in postings:
@@ ... @@
                 yield post
         except KeyError:
             pass
 def filter_meta_match(postings: Postings, key: MetaKey, regexp: Regexp) -> Postings:
     for post in postings:
         try:
             if re.search(regexp, post.meta[key]):
                 yield post
         except (KeyError, TypeError):
             pass

tests/test_filters.py

➞

Show inline comments

@@ ... @@
     postings = data.Posting.from_entries(cc_txn_pair)
     actual = filters.filter_meta_equal(postings, key, value)
     check_filter(actual, cc_txn_pair, expected_indexes)
 @pytest.mark.parametrize('key,regexp,expected_indexes', [
     ('entity', '^Smith-', range(5)),
     ('receipt', r'\.pdf$', range(5)),
     ('receipt', 'Receipt', range(3)),
     ('statement', '.', [4]),
     ('metadate', 'foo', ()),
     ('BadKey', '.', ()),
     ('emptykey', '.', ()),
 ])
 def test_filter_meta_match(cc_txn_pair, key, regexp, expected_indexes):
     postings = data.Posting.from_entries(cc_txn_pair)
     actual = filters.filter_meta_match(postings, key, regexp)
     check_filter(actual, cc_txn_pair, expected_indexes)

0 comments (0 inline, 0 general)