feature: Added Headers param

Author: IvanildoBarauna

src/api_to_dataframe/controller/client_builder.py

@@ -5,6 +5,7 @@
 class ClientBuilder:
     def __init__(self,
                  endpoint: str,
+                 headers: dict = {},
                  retry_strategy: RetryStrategies = RetryStrategies.NoRetryStrategy,
                  timeout: int = 5):
         """
@@ -24,6 +25,7 @@ def __init__(self,
             self.endpoint = endpoint
             self.retry_strategy = retry_strategy
             self.timeout = timeout
+            self.headers = headers
 
     def get_api_data(self):
         """
@@ -32,7 +34,7 @@ def get_api_data(self):
         Returns:
             dict: The response from the API.
         """
-        response = GetData.get_response(self.endpoint, self.retry_strategy, self.timeout)
+        response = GetData.get_response(self.endpoint, self.headers, self.retry_strategy, self.timeout)
         return response
 
     @staticmethod

src/api_to_dataframe/models/get_data.py

@@ -7,9 +7,9 @@
 
 class GetData:
     @staticmethod
-    def get_response(endpoint: str, retry_strategies: RetryStrategies, timeout: int):
+    def get_response(endpoint: str, headers: dict, retry_strategies: RetryStrategies, timeout: int):
         try:
-            response = requests.get(endpoint, timeout=timeout)
+            response = requests.get(endpoint, timeout=timeout, headers=headers)
             response.raise_for_status()
         except HTTPError as http_err:
             print(f'HTTP error occurred: {http_err}')