add feature that finialize can remove current record

beancount · Jan 3, 2025 · e7d13bd · e7d13bd
1 parent f8d2bc5
commit e7d13bd
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/beangulp/importers/csvbase.py b/beangulp/importers/csvbase.py
@@ -310,7 +310,8 @@ def extract(self, filepath, existing):
             txn = self.finalize(txn, row)
 
             # Add the transaction to the output list.
-            entries.append(txn)
+            if txn is not None:
+                entries.append(txn)
 
             # Add balance to balances list.
             balance = getattr(row, 'balance', None)

diff --git a/beangulp/importers/csvbase_test.py b/beangulp/importers/csvbase_test.py
@@ -481,3 +481,21 @@ def finalize(self, txn, row):
           Assets:CSV  -1.00 EUR
           Expenses:Testing  1.00 EUR
         """)
+
+    @docfile
+    def test_extract_finalize_remove(self, filename):
+        """\
+        2021-05-17, Test, -1.00, Testing
+        """
+        class CSVImporter(Base):
+            date = Date(0)
+            narration = Column(1)
+            amount = Amount(2)
+            category = Column(3)
+            names = False
+
+            def finalize(self, txn, row):
+                return None
+
+        importer = CSVImporter('Assets:CSV', 'EUR')
+        entries = importer.extract(filename, [])