NPO-Accounting/conservancy_beancount
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

data.Account: Add is_income method.

Browse source
This commit is contained in:
Brett Smith 2020-03-19 10:23:55 -04:00
parent b7089f5ddf
commit 7cd569be7b
3 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
conservancy_beancount/data.py
View file

@ -35,6 +35,9 @@ from .beancount_types import (
class Account(str):
SEP = bc_account.sep
def is_income(self) -> bool:
return self.is_under('Income:', 'UnearnedIncome:') is not None
def is_real_asset(self) -> bool:
return bool(
self.is_under('Assets:')

2
conservancy_beancount/plugin/meta_income_type.py
View file

@ -43,7 +43,7 @@ class MetaIncomeType(core._NormalizePostingMetadataHook):
}
def _run_on_post(self, txn: Transaction, post: data.Posting) -> bool:
return post.account.is_under('Income:', 'UnearnedIncome:') is not None
return post.account.is_income()
def _default_value(self, txn: Transaction, post: data.Posting) -> MetaValueEnum:
try:

13
tests/test_data_account.py
View file

@ -45,6 +45,19 @@ def test_is_under_multi_arg(acct_name, expected):
expected += ':'
assert data.Account(acct_name).is_under('Income:', 'UnearnedIncome:') == expected
@pytest.mark.parametrize('acct_name,expected', [
('Accrued:AccountsReceivable', False),
('Assets:Cash', False),
('Expenses:General', False),
('Income:Donations', True),
('Income:Sales', True),
('Income:Other', True),
('Liabilities:CreditCard', False),
('UnearnedIncome:MatchPledges', True),
])
def test_is_income(acct_name, expected):
assert data.Account(acct_name).is_income() == expected
@pytest.mark.parametrize('acct_name,expected', [
('Accrued:AccountsPayable', False),
('Accrued:AccountsReceivable', False),