From dd39e872f722b6f3e43d9146f25b1983231ccb80 Mon Sep 17 00:00:00 2001
From: Arsen Gabdulin <63103528+DAKExDUCK@users.noreply.github.com>
Date: Sun, 2 Oct 2022 13:59:30 +0600
Subject: [PATCH] add option disable progress bar by env variable (#438)

---
 webdriver_manager/core/config.py | 8 ++++++++
 webdriver_manager/core/http.py   | 5 +++--
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/webdriver_manager/core/config.py b/webdriver_manager/core/config.py
index 9b1e1fac..f7fea5d6 100644
--- 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
index f4d5e5a2..1e924a3a 100644
--- 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