[intelowlproject#160][IMP] Add proxy support on IntelOwl class

Ektoplasma · Nov 5, 2022 · 5a8ac30 · 5a8ac30
1 parent 99e6b1a
commit 5a8ac30
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pyintelowl/pyintelowl.py b/pyintelowl/pyintelowl.py
@@ -26,11 +26,13 @@ def __init__(
         certificate: str = None,
         logger: logging.Logger = None,
         cli: bool = False,
+        proxies: dict = None
     ):
         self.token = token
         self.instance = instance_url
         self.certificate = certificate
         self.cli = cli
+        self.proxies = proxies
         if logger:
             self.logger = logger
         else:
@@ -45,6 +47,8 @@ def session(self) -> requests.Session:
             session = requests.Session()
             if self.certificate is not True:
                 session.verify = self.certificate
+            if self.proxies:
+                session.proxies.update(self.proxies)
             session.headers.update(
                 {
                     "Authorization": f"Token {self.token}",