feat: add remove_bank function to manage bank entries

cuenca-mx · Dec 23, 2024 · d84ee8a · d84ee8a
1 parent 4eb15b9
commit d84ee8a
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/clabe/__init__.py b/clabe/__init__.py
@@ -8,6 +8,7 @@
     'get_bank_name',
     'validate_clabe',
     'configure_additional_bank',
+    'remove_bank',
 ]
 
 from .banks import BANK_NAMES, BANKS
@@ -17,6 +18,7 @@
     configure_additional_bank,
     generate_new_clabes,
     get_bank_name,
+    remove_bank,
     validate_clabe,
 )
 from .version import __version__
diff --git a/clabe/validations.py b/clabe/validations.py
@@ -98,3 +98,18 @@ def configure_additional_bank(bank_code_banxico: str, bank_name: str) -> None:
     )
     BANKS[request.bank_code_abm] = request.bank_code_banxico
     BANK_NAMES[request.bank_code_banxico] = request.bank_name
+
+
+def remove_bank(bank_code_banxico: str) -> None:
+    bank_code_abm = next(
+        (
+            abm
+            for abm, banxico in BANKS.items()
+            if banxico == bank_code_banxico
+        ),
+        None,
+    )
+
+    if bank_code_abm:
+        BANKS.pop(bank_code_abm)
+        BANK_NAMES.pop(bank_code_banxico)
diff --git a/tests/test_clabe.py b/tests/test_clabe.py
@@ -5,6 +5,7 @@
     configure_additional_bank,
     generate_new_clabes,
     get_bank_name,
+    remove_bank,
     validate_clabe,
 )
 
@@ -64,3 +65,9 @@ def test_configure_additional_bank_success(abm_code, banxico_code, name):
 def test_configure_additional_bank_invalid_inputs(banxico_code, name):
     with pytest.raises(ValueError):
         configure_additional_bank(banxico_code, name)
+
+
+def test_remove_bank():
+    remove_bank('40138')
+    with pytest.raises(ValueError):
+        get_bank_name('138')