From 56d92d9547a74eec5045eb25ba05b6f7c292722b Mon Sep 17 00:00:00 2001 From: Ugochukwu Onyebuchi Date: Tue, 7 Mar 2023 14:47:08 +0100 Subject: [PATCH] feat: add method to retrieve bank list from API --- decide/models/enums.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/decide/models/enums.py b/decide/models/enums.py index d838837..4e6bec8 100644 --- a/decide/models/enums.py +++ b/decide/models/enums.py @@ -1,5 +1,8 @@ +import requests from enum import Enum +BANK_LIST_URL = "https://api.indicina.co/api/v3/banks" + class Currency(Enum): NGN = "NGN" @@ -31,6 +34,15 @@ class Bank(Enum): UNION = "032" ZENITH = "057" + @classmethod + def get_bank_list(cls): + response = requests.get(BANK_LIST_URL) + if response.status_code == 200: + bank_list = response.json()['data'] + return [(bank['name'], bank['code']) for bank in bank_list] + else: + raise Exception(f"Failed to get bank list. Status code: {response.status_code}") + class StatementType(Enum): JSON = "json"