fix log format for empty line

Author: SergeyPirogov

webdriver_manager/logger.py

@@ -4,7 +4,7 @@
 loggers = {}
 
 
-def log(text, level=logging.INFO, name="WDM", first_line=False):
+def log(text, level=logging.INFO, name="WDM", first_line=False, formatter='[%(name)s] - %(message)s'):
     log_level = os.getenv('WDM_LOG_LEVEL')
     if log_level:
         level = int(log_level)
@@ -14,7 +14,7 @@ def log(text, level=logging.INFO, name="WDM", first_line=False):
         _logger = logging.getLogger(name)
 
         handler = logging.StreamHandler()
-        formatter = logging.Formatter('[%(name)s] - %(message)s')
+        formatter = logging.Formatter(formatter)
         handler.setFormatter(formatter)
         _logger.addHandler(handler)
         _logger.setLevel(level)

webdriver_manager/manager.py

@@ -1,3 +1,5 @@
+import os
+
 from webdriver_manager.driver_cache import DriverCache
 from webdriver_manager.logger import log
 from webdriver_manager.utils import download_file
@@ -6,8 +8,8 @@
 class DriverManager(object):
     def __init__(self, root_dir=None, log_level=None, print_first_line=None, cache_valid_range=1):
         self.driver_cache = DriverCache(root_dir, cache_valid_range)
-        if print_first_line:
-            print("\n")
+        if os.environ.get('WDM_PRINT_FIRST_LINE', str(print_first_line)) == 'True':
+            log("\n", formatter='%(message)s')
         log("====== WebDriver manager ======")
 
     def install(self):