Add count to specify how often a mapping is used

src/toldg/process.py

@@ -64,29 +64,16 @@ def get_transactions(csv_file: str, config: CsvConfig) -> List[Transaction]:
     return transactions
 
 
-def find_duplicates(transactions: List[Transaction]):
-    rows = set()
-    for t in transactions:
-        row = t.row
-        if row in rows:
-            logging.critical(f"'{row}' is duplicated.")
-            logging.critical("Exit because of duplicated transactions.")
-            sys.exit(1)
-        else:
-            rows.add(row)
-
-
 def apply_mappings(transactions: List[Transaction], mappings: Dict[str, Mapping]):
     """Apply mappings to transactions."""
-    unused_mappings = set(mappings.keys())
-
     for t in transactions:
         if t.row in mappings:
             mapping = mappings[t.row]
-
-            assert isinstance(
-                mapping, Mapping
-            ), "Only new mappings format is supported."
+            assert isinstance(mapping, Mapping)
+            assert (
+                mapping.count > 0
+            ), f"{mapping} used by {t} but count is not greater than '0'."
+            mapping.count -= 1
             t.account2 = mapping.account2
 
             if mapping.narration:
@@ -94,15 +81,11 @@ def apply_mappings(transactions: List[Transaction], mappings: Dict[str, Mapping]
 
             if mapping.payee:
                 t.payee = mapping.payee
-
-            unused_mappings.discard(t.row)
         else:
             logging.warning(f"No mapping for '{t}'.")
 
-    for row in unused_mappings:
-        mapping_info = mappings[row]
-        account2 = mapping_info["account2"]
-        logging.warning(f"Unused mapping '{row}' -> {account2}")
+    for mapping in mappings.values():
+        assert mapping.count == 0, f"{mapping} was not used as often as expected!"
 
 
 def process_csv_files(config: Config):
@@ -113,9 +96,6 @@ def process_csv_files(config: Config):
         csv_config = get_csv_config(csv_file, config.csv_configs)
         transactions += get_transactions(csv_file, csv_config)
 
-    if config.find_duplicates:
-        find_duplicates(transactions)
-
     mappings = toldg.utils.read_mappings(config.mappings_file)
     apply_mappings(transactions, mappings)
     toldg.predict.add_account2(transactions, config.categories)