add option disable progress bar by env variable (#438)

SergeyPirogov · Oct 2, 2022 · dd39e87 · dd39e87
1 parent 2cceda2
commit dd39e87
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/webdriver_manager/core/config.py b/webdriver_manager/core/config.py
@@ -29,5 +29,13 @@ def wdm_log_level():
         return default_level
 
 
+def wdm_progress_bar():
+    default_level = 1
+    try:
+        return int(os.getenv("WDM_PROGRESS_BAR", default_level))
+    except Exception:
+        return default_level
+
+
 def get_xdist_worker_id():
     return os.getenv("PYTEST_XDIST_WORKER", '')
diff --git a/webdriver_manager/core/http.py b/webdriver_manager/core/http.py
@@ -1,7 +1,7 @@
 import requests
 from requests import Response
 
-from webdriver_manager.core.config import ssl_verify
+from webdriver_manager.core.config import ssl_verify, wdm_progress_bar
 from webdriver_manager.core.utils import show_download_progress
 
 
@@ -31,6 +31,7 @@ def __init__(self):
     def get(self, url, **kwargs) -> Response:
         resp = requests.get(url=url, verify=self._ssl_verify, stream=True, **kwargs)
         self.validate_response(resp)
-        show_download_progress(resp)
+        if wdm_progress_bar():
+            show_download_progress(resp)
         return resp