fix: update example bank enum and column name

benjamin-awd · Oct 26, 2023 · e8307d1 · e8307d1
1 parent 3f89319
commit e8307d1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/monopoly/constants.py b/monopoly/constants.py
@@ -13,6 +13,7 @@ class BankNames(Enum):
     HSBC = "hsbc"
     OCBC = "ocbc"
     STANDARD_CHARTERED = "standard_chartered"
+    EXAMPLE = "monopoly"
 
 
 class StatementFields(str, Enum):

diff --git a/monopoly/examples/example_bank.py b/monopoly/examples/example_bank.py
@@ -1,16 +1,16 @@
 from monopoly.banks.base import BankBase
 from monopoly.config import StatementConfig
-from monopoly.constants import AccountType
+from monopoly.constants import AccountType, BankNames
 
 
 # fmt: off
 class MonopolyBank(BankBase):
     """Dummy class to help with reading the example PDF statement"""
     statement_config = StatementConfig(
-        bank_name="monopoly",
+        bank_name=BankNames.EXAMPLE,
         account_type=AccountType.CREDIT,
         transaction_pattern=(
-            r"(?P<date>\d+/\d+)\s*"
+            r"(?P<transaction_date>\d+/\d+)\s*"
             r"(?P<description>.*?)\s*"
             r"(?P<amount>[\d.,]+)$"
         ),