feat: add sinks protocols

src/ficamp/classifier/google_apis.py

@@ -94,7 +94,7 @@ def query_gmaps_category(concept):
         cached_category = cached.get(concept)
         if not cached_category:
             try:
-                #ams = "52.3676,4.9041"
+                # ams = "52.3676,4.9041"
                 gmaps_category = find_business_category_in_google(concept)
             except GoogleException as error:
                 print(f"error: {error}")

src/ficamp/classifier/keywords.py

@@ -1,6 +1,7 @@
 """
 Logic to sort transactions based on keywords.
 """
+
 import json
 import pathlib
 

src/ficamp/sinks/protocols.py

@@ -0,0 +1,17 @@
+from enum import Enum
+from typing import Protocol
+
+from ficamp.datastructures import Tx
+
+
+class BaseSinkException(Exception):
+    """Base exception for all sink exceptions.
+
+    If your sink function is going to raise an exception, it should inherit from this class.
+    """
+
+    pass
+
+
+class SinkProtocol(Protocol):
+    def write(self, transations: list[Tx]): ...

tests/test_preprocessing.py

@@ -5,9 +5,9 @@
     remove_colon,
     remove_comma,
     remove_digits,
+    remove_isolated_digits,
     remove_pipes,
     remove_punctuation,
-    remove_isolated_digits,
     remove_short_words,
 )