Merge pull request #664 from seleniumbase/improve-seleniumbase-driver-install

mdmintz · web-flow · commit d05d1a70b134 · 2020-08-20T23:50:14.000-04:00
Improve the "seleniumbase install DRIVER" command
diff --git a/examples/decryption_test.py b/examples/decryption_test.py
@@ -14,9 +14,9 @@ def test_rate_limited_printing(self):
         self.update_text("#user-name", "standard_user")
 
         encrypted_password = "$^*ENCRYPT=S3BDTAdCWzMmKEY8Gjg=?&#$"
-        print("Encrypted Password = %s" % encrypted_password)
+        print("\nEncrypted Password = %s" % encrypted_password)
         password = encryption.decrypt(encrypted_password)
-        print("Password = %s" % password)
+        print("Decrypted Password = %s" % password)
         self.update_text("#password", password)
 
         self.click('input[type="submit"]')
diff --git a/examples/test_chromedriver.py b/examples/test_chromedriver.py
@@ -0,0 +1,59 @@
+"""
+This test is only for Chrome!
+(Verify that your chromedriver is compatible with your version of Chrome.)
+"""
+import colorama
+from seleniumbase import BaseCase
+
+
+class ChromeTestClass(BaseCase):
+
+    def test_chromedriver_matches_chrome(self):
+        if self.browser != "chrome":
+            print("\n  This test is only for Chrome!")
+            print("  (Run with: '--browser=chrome')")
+            self.skip("This test is only for Chrome!")
+        driver_capabilities = self.driver.__dict__["capabilities"]
+        if "version" in driver_capabilities:
+            chrome_version = driver_capabilities["version"]
+        else:
+            chrome_version = driver_capabilities["browserVersion"]
+        major_chrome_version = chrome_version.split('.')[0]
+        chrome_dict = self.driver.__dict__["capabilities"]["chrome"]
+        chromedriver_version = chrome_dict["chromedriverVersion"]
+        chromedriver_version = chromedriver_version.split(' ')[0]
+        major_chromedriver_version = chromedriver_version.split('.')[0]
+        c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
+        c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
+        c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
+        c4 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
+        c5 = colorama.Fore.RED + colorama.Back.LIGHTGREEN_EX
+        cr = colorama.Style.RESET_ALL
+        pr_chromedriver_version = c3 + chromedriver_version + cr
+        pr_chrome_version = c2 + chrome_version + cr
+        message = (
+            "\n"
+            "* Your version of chromedriver is: %s\n"
+            "*\n* And your version of Chrome is: %s"
+            "" % (pr_chromedriver_version, pr_chrome_version))
+        print(message)
+        if major_chromedriver_version < major_chrome_version:
+            install_sb = (
+                "seleniumbase install chromedriver %s" % major_chrome_version)
+            pr_install_sb = c1 + install_sb + cr
+            up_msg = "You may want to upgrade your version of chromedriver:"
+            up_msg = c4 + up_msg + cr
+            message = ("*\n* %s\n*\n* >>> %s" % (up_msg, pr_install_sb))
+            print(message)
+        elif major_chromedriver_version > major_chrome_version:
+            up_msg = "You may want to upgrade your version of Chrome:"
+            up_msg = c5 + up_msg + cr
+            up_url = c1 + "chrome://settings/help" + cr
+            message = ("*\n* %s\n*\n* See: %s" % (up_msg, up_url))
+            print(message)
+        else:
+            up_msg = (
+                "Success! Your chromedriver is compatible with your Chrome!")
+            up_msg = c1 + up_msg + cr
+            message = ("*\n* %s\n" % up_msg)
+            print(message)
diff --git a/requirements.txt b/requirements.txt
@@ -16,7 +16,7 @@ requests==2.24.0
 selenium==3.141.0
 msedge-selenium-tools==3.141.2
 pluggy==0.13.1
-attrs>=19.3.0
+attrs>=20.1.0
 py==1.8.1;python_version<"3.5"
 py==1.9.0;python_version>="3.5"
 pytest==4.6.11;python_version<"3.5"
@@ -52,7 +52,7 @@ brython>=3.8.9
 pyotp==2.4.0
 boto==2.49.0
 cffi==1.14.2
-rich==5.2.0;python_version>="3.6" and python_version<"4.0"
+rich==5.2.1;python_version>="3.6" and python_version<"4.0"
 flake8==3.7.9;python_version<"3.5"
 flake8==3.8.3;python_version>="3.5"
 pyflakes==2.1.1;python_version<"3.5"
diff --git a/seleniumbase/console_scripts/sb_install.py b/seleniumbase/console_scripts/sb_install.py
@@ -13,11 +13,12 @@
         seleniumbase install chromedriver
         seleniumbase install geckodriver
         seleniumbase install edgedriver
+        seleniumbase install edgedriver 84.0.522.61
+        seleniumbase install chromedriver 84
         seleniumbase install chromedriver 84.0.4147.30
         seleniumbase install chromedriver latest
         seleniumbase install chromedriver -p
         seleniumbase install chromedriver latest -p
-        seleniumbase install edgedriver 84.0.522.61
 Output:
         Installs the chosen webdriver to seleniumbase/drivers/
         (chromedriver is required for Chrome automation)
@@ -27,6 +28,7 @@
         (operadriver is required for Opera Browser automation)
 """
 
+import colorama
 import os
 import platform
 import requests
@@ -108,18 +110,28 @@ def main(override=None):
     platform_code = None
     inner_folder = None
     copy_to_path = False
+    latest_version = ""
     use_version = ""
     new_file = ""
     f_name = ""
+    c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
+    c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
+    c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
+    cr = colorama.Style.RESET_ALL
 
     if name == "chromedriver":
+        last = "https://chromedriver.storage.googleapis.com/LATEST_RELEASE"
         use_version = DEFAULT_CHROMEDRIVER_VERSION
         get_latest = False
+        get_v_latest = False
         if num_args == 4 or num_args == 5:
             if "-p" not in sys.argv[3].lower():
                 use_version = sys.argv[3]
-                if use_version.lower() == "latest":
+                uv_low = use_version.lower()
+                if uv_low == "latest":
                     get_latest = True
+                elif len(uv_low) < 4 and uv_low.isdigit() and int(uv_low) > 69:
+                    get_v_latest = True
             else:
                 copy_to_path = True
         if num_args == 5:
@@ -138,20 +150,48 @@ def main(override=None):
                             "to download!")
         found_chromedriver = False
         if get_latest:
-            last = "http://chromedriver.storage.googleapis.com/LATEST_RELEASE"
             url_request = requests.get(last)
             if url_request.ok:
                 found_chromedriver = True
                 use_version = url_request.text
-        download_url = ("http://chromedriver.storage.googleapis.com/"
+        elif get_v_latest:
+            url_req = requests.get(last)
+            if url_req.ok:
+                latest_version = url_req.text
+            last = last + "_" + use_version
+            url_request = requests.get(last)
+            if url_request.ok:
+                found_chromedriver = True
+                use_version = url_request.text
+                if use_version == latest_version:
+                    get_latest = True
+        download_url = ("https://chromedriver.storage.googleapis.com/"
                         "%s/%s" % (use_version, file_name))
         url_request = None
         if not found_chromedriver:
+            url_req = requests.get(last)
+            if url_req.ok:
+                latest_version = url_req.text
+                if use_version == latest_version:
+                    get_latest = True
             url_request = requests.get(download_url)
         if found_chromedriver or url_request.ok:
-            print("\n* chromedriver version for download = %s" % use_version)
+            p_version = use_version
+            p_version = c3 + use_version + cr
+            if get_latest:
+                p_version = p_version + " (Latest)"
+            else:
+                p_version = p_version + " (NOT Latest)"
+            msg = c2 + "chromedriver version for download" + cr
+            print("\n*** %s = %s" % (msg, p_version))
         else:
             raise Exception("Could not find chromedriver to download!\n")
+        if not get_latest:
+            to_upgrade = " " + c3 + "To upgrade" + cr
+            run_this = c3 + "run this" + cr
+            install_sb = c1 + "seleniumbase install chromedriver latest" + cr
+            print("\n#%s to the latest version of chromedriver," % to_upgrade)
+            print('#  %s: >>> %s' % (run_this, install_sb))
     elif name == "geckodriver" or name == "firefoxdriver":
         use_version = DEFAULT_GECKODRIVER_VERSION
         if "win32" in sys_plat or "win64" in sys_plat or "x64" in sys_plat:
@@ -196,7 +236,9 @@ def main(override=None):
         if not found_geckodriver:
             url_request = requests.get(download_url)
         if found_geckodriver or url_request.ok:
-            print("\n* geckodriver version for download = %s" % use_version)
+            msg = c2 + "geckodriver version for download" + cr
+            p_version = c3 + use_version + cr
+            print("\n*** %s = %s" % (msg, p_version))
         else:
             raise Exception("\nCould not find the specified geckodriver "
                             "version to download!\n")
@@ -226,6 +268,9 @@ def main(override=None):
                             "only for Windows or Mac operating systems!")
         download_url = ("https://msedgedriver.azureedge.net/"
                         "%s/%s" % (use_version, file_name))
+        msg = c2 + "edgedriver version for download" + cr
+        p_version = c3 + use_version + cr
+        print("\n*** %s = %s" % (msg, p_version))
     elif name == "iedriver":
         major_version = "3.14"
         full_version = "3.14.0"
@@ -237,7 +282,7 @@ def main(override=None):
         else:
             raise Exception("Sorry! IEDriver is only for "
                             "Windows-based operating systems!")
-        download_url = ("http://selenium-release.storage.googleapis.com/"
+        download_url = ("https://selenium-release.storage.googleapis.com/"
                         "%s/%s" % (major_version, file_name))
     elif name == "operadriver" or name == "operachromiumdriver":
         name = "operadriver"
@@ -290,6 +335,9 @@ def main(override=None):
         download_url = ("https://github.com/operasoftware/operachromiumdriver/"
                         "releases/download/"
                         "%s/%s" % (use_version, file_name))
+        msg = c2 + "operadriver version for download" + cr
+        p_version = c3 + use_version + cr
+        print("\n*** %s = %s" % (msg, p_version))
     else:
         invalid_run_command()
 
@@ -326,15 +374,16 @@ def main(override=None):
             print('Unzip Complete!\n')
             for f_name in contents:
                 new_file = downloads_folder + '/' + str(f_name)
-                print("The file [%s] was saved to:\n%s\n" % (f_name, new_file))
+                pr_file = c3 + new_file + cr
+                print("The file [%s] was saved to:\n%s\n" % (f_name, pr_file))
                 print("Making [%s %s] executable ..." % (f_name, use_version))
                 make_executable(new_file)
-                print("[%s] is now ready for use!" % f_name)
+                print("%s[%s] is now ready for use!%s" % (c1, f_name, cr))
                 if copy_to_path and os.path.exists(LOCAL_PATH):
                     path_file = LOCAL_PATH + f_name
                     shutil.copyfile(new_file, path_file)
                     make_executable(path_file)
-                    print("Also copied to: %s" % path_file)
+                    print("Also copied to: %s%s%s" % (c3, path_file, cr))
             print("")
         elif name == "edgedriver" or name == "msedgedriver":
             if "darwin" in sys_plat or "linux" in sys_plat:
@@ -364,7 +413,7 @@ def main(override=None):
                     if os.path.exists(new_file):
                         os.remove(new_file)
             if not driver_file or not driver_path:
-                raise Exception("Operadriver missing from Zip file!")
+                raise Exception("msedgedriver missing from Zip file!")
             print('Extracting %s from %s ...' % (contents, file_name))
             zip_ref.extractall(downloads_folder)
             zip_ref.close()
@@ -382,7 +431,7 @@ def main(override=None):
                 driver_file, driver_path))
             print("Making [%s %s] executable ..." % (driver_file, use_version))
             make_executable(driver_path)
-            print("[%s] is now ready for use!" % driver_file)
+            print("%s[%s] is now ready for use!%s" % (c1, driver_file, cr))
             print("")
         elif name == "operadriver":
             if len(contents) > 3:
@@ -409,16 +458,17 @@ def main(override=None):
             inner_driver = downloads_folder + '/' + inner_folder + driver_file
             inner_sha = downloads_folder + '/' + inner_folder + "sha512_sum"
             shutil.copyfile(inner_driver, driver_path)
+            pr_driver_path = c3 + driver_path + cr
             print("The file [%s] was saved to:\n%s\n" % (
-                driver_file, driver_path))
+                driver_file, pr_driver_path))
             print("Making [%s %s] executable ..." % (driver_file, use_version))
             make_executable(driver_path)
-            print("[%s] is now ready for use!" % driver_file)
+            print("%s[%s] is now ready for use!%s" % (c1, driver_file, cr))
             if copy_to_path and os.path.exists(LOCAL_PATH):
                 path_file = LOCAL_PATH + driver_file
                 shutil.copyfile(driver_path, path_file)
                 make_executable(path_file)
-                print("Also copied to: %s" % path_file)
+                print("Also copied to: %s%s%s" % (c3, path_file, cr))
             # Clean up extra files
             if os.path.exists(inner_driver):
                 os.remove(inner_driver)
@@ -450,15 +500,16 @@ def main(override=None):
             print('Unzip Complete!\n')
             for f_name in contents:
                 new_file = downloads_folder + '/' + str(f_name)
-                print("The file [%s] was saved to:\n%s\n" % (f_name, new_file))
+                pr_file = c3 + new_file + cr
+                print("The file [%s] was saved to:\n%s\n" % (f_name, pr_file))
                 print("Making [%s %s] executable ..." % (f_name, use_version))
                 make_executable(new_file)
-                print("[%s] is now ready for use!" % f_name)
+                print("%s[%s] is now ready for use!%s" % (c1, f_name, cr))
                 if copy_to_path and os.path.exists(LOCAL_PATH):
                     path_file = LOCAL_PATH + f_name
                     shutil.copyfile(new_file, path_file)
                     make_executable(path_file)
-                    print("Also copied to: %s" % path_file)
+                    print("Also copied to: %s%s%s" % (c3, path_file, cr))
             print("")
         elif len(contents) == 0:
             raise Exception("Tar file %s is empty!" % tar_file_path)
@@ -469,7 +520,7 @@ def main(override=None):
         if "Driver" in file_name or "driver" in file_name:
             print("Making [%s] executable ..." % file_name)
             make_executable(file_path)
-            print("[%s] is now ready for use!" % file_name)
+            print("%s[%s] is now ready for use!%s" % (c1, file_name, cr))
             print("Location of [%s]:\n%s\n" % (file_name, file_path))
 
 
diff --git a/seleniumbase/fixtures/base_case.py b/seleniumbase/fixtures/base_case.py
@@ -1236,7 +1236,37 @@ def hover_on_element(self, selector, by=By.CSS_SELECTOR):
         self.__demo_mode_highlight_if_active(selector, by)
         self.scroll_to(selector, by=by)
         time.sleep(0.05)  # Settle down from scrolling before hovering
-        return page_actions.hover_on_element(self.driver, selector)
+        if self.browser != "chrome":
+            return page_actions.hover_on_element(self.driver, selector)
+        # Using Chrome
+        # (Pure hover actions won't work on early chromedriver versions)
+        try:
+            return page_actions.hover_on_element(self.driver, selector)
+        except WebDriverException as e:
+            driver_capabilities = self.driver.__dict__["capabilities"]
+            if "version" in driver_capabilities:
+                chrome_version = driver_capabilities["version"]
+            else:
+                chrome_version = driver_capabilities["browserVersion"]
+            major_chrome_version = chrome_version.split('.')[0]
+            chrome_dict = self.driver.__dict__["capabilities"]["chrome"]
+            chromedriver_version = chrome_dict["chromedriverVersion"]
+            chromedriver_version = chromedriver_version.split(' ')[0]
+            major_chromedriver_version = chromedriver_version.split('.')[0]
+            install_sb = (
+                "seleniumbase install chromedriver %s" % major_chrome_version)
+            if major_chromedriver_version < major_chrome_version:
+                # Upgrading the driver is required for performing hover actions
+                message = (
+                    "\n"
+                    "You need a newer chromedriver to perform hover actions!\n"
+                    "Your version of chromedriver is: %s\n"
+                    "And your version of Chrome is: %s\n"
+                    "You can fix this issue by running:\n>>> %s\n"
+                    "" % (chromedriver_version, chrome_version, install_sb))
+                raise Exception(message)
+            else:
+                raise Exception(e)
 
     def hover_and_click(self, hover_selector, click_selector,
                         hover_by=By.CSS_SELECTOR, click_by=By.CSS_SELECTOR,
diff --git a/setup.py b/setup.py
@@ -54,7 +54,7 @@
 
 setup(
     name='seleniumbase',
-    version='1.47.0',
+    version='1.47.1',
     description='Web Automation and Test Framework - https://seleniumbase.io',
     long_description=long_description,
     long_description_content_type='text/markdown',
@@ -108,7 +108,7 @@
         'selenium==3.141.0',
         'msedge-selenium-tools==3.141.2',
         'pluggy==0.13.1',
-        'attrs>=19.3.0',
+        'attrs>=20.1.0',
         'py==1.8.1;python_version<"3.5"',
         'py==1.9.0;python_version>="3.5"',
         'pytest==4.6.11;python_version<"3.5"',
@@ -144,7 +144,7 @@
         'pyotp==2.4.0',
         'boto==2.49.0',
         'cffi==1.14.2',
-        'rich==5.2.0;python_version>="3.6" and python_version<"4.0"',
+        'rich==5.2.1;python_version>="3.6" and python_version<"4.0"',
         'flake8==3.7.9;python_version<"3.5"',
         'flake8==3.8.3;python_version>="3.5"',
         'pyflakes==2.1.1;python_version<"3.5"',
