Merge pull request #262 from mayankkholia/proxy-feature

Added Proxy Feature #261

Author: 4n4nd

prometheus_api_client/prometheus_connect.py

@@ -36,6 +36,8 @@ class PrometheusConnect:
     :param retry: (Retry) Retry adapter to retry on HTTP errors
     :param auth: (optional) Auth tuple to enable Basic/Digest/Custom HTTP Auth. See python
         requests library auth parameter for further explanation.
+    :param proxy: (Optional) Proxies dictonary to enable connection through proxy. 
+        Example: {"http_proxy": "<ip_address/hostname:port>", "https_proxy": "<ip_address/hostname:port>"}  
     """
 
     def __init__(
@@ -44,7 +46,8 @@ def __init__(
         headers: dict = None,
         disable_ssl: bool = False,
         retry: Retry = None,
-        auth: tuple = None
+        auth: tuple = None,
+        proxy: dict = None
     ):
         """Functions as a Constructor for the class PrometheusConnect."""
         if url is None:
@@ -66,6 +69,8 @@ def __init__(
         self.auth = auth
 
         self._session = requests.Session()
+        if proxy is not None:
+            self._session.proxies = proxy
         self._session.mount(self.url, HTTPAdapter(max_retries=retry))
 
     def check_prometheus_connection(self, params: dict = None) -> bool: