plugin: Refactor hooks to use new payment-related methods.

conservancy_beancount/plugin/meta_receipt.py

@@ -15,6 +15,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 from . import core
+from .. import config as configmod
 from .. import data
 from .. import errors as errormod
 from ..beancount_types import (
@@ -23,14 +24,16 @@ from ..beancount_types import (
 )
 
 class MetaReceipt(core._RequireLinksPostingMetadataHook):
-    DEFAULT_STOP_AMOUNT = 0
     METADATA_KEY = 'receipt'
 
+    def __init__(self, config: configmod.Config) -> None:
+        self.payment_threshold = -abs(config.payment_threshold())
+
     def _run_on_post(self, txn: Transaction, post: data.Posting) -> bool:
         return bool(
             (post.account.is_real_asset() or post.account.is_under('Liabilities'))
-            and post.units.number
+            and post.units.number is not None
-            and post.units.number < self.DEFAULT_STOP_AMOUNT
+            and post.units.number < self.payment_threshold
         )
 
     def post_run(self, txn: Transaction, post: data.Posting) -> errormod.Iter:

conservancy_beancount/plugin/meta_tax_implication.py

@@ -17,13 +17,12 @@
 import decimal
 
 from . import core
+from .. import config as configmod
 from .. import data
 from ..beancount_types import (
     Transaction,
 )
 
-DEFAULT_STOP_AMOUNT = decimal.Decimal(0)
-
 class MetaTaxImplication(core._NormalizePostingMetadataHook):
     VALUES_ENUM = core.MetadataEnum('tax-implication', [
         '1099',
@@ -45,9 +44,8 @@ class MetaTaxImplication(core._NormalizePostingMetadataHook):
         'W2',
     ])
 
+    def __init__(self, config: configmod.Config) -> None:
+        self.payment_threshold = config.payment_threshold()
+
     def _run_on_post(self, txn: Transaction, post: data.Posting) -> bool:
-        return bool(
+        return post.is_payment(self.payment_threshold)
-            post.account.is_real_asset()
-            and post.units.number
-            and post.units.number < DEFAULT_STOP_AMOUNT
-        )

tests/testutil.py

@@ -144,10 +144,12 @@ class Transaction:
 
 
 class TestConfig:
-    def __init__(self,
+    def __init__(self, *,
+                 payment_threshold=0,
                  repo_path=None,
                  rt_client=None,
     ):
+        self._payment_threshold = Decimal(payment_threshold)
         self.repo_path = test_path(repo_path)
         self._rt_client = rt_client
         if rt_client is None:
@@ -155,6 +157,9 @@ class TestConfig:
         else:
             self._rt_wrapper = rtutil.RT(rt_client)
 
+    def payment_threshold(self):
+        return self._payment_threshold
+
     def repository_path(self):
         return self.repo_path