Merge pull request #17 from LeithMerrifield/Dev

Created auto updater/installer for the chrome driver

Author: LeithMerrifield

Main.py

@@ -1,6 +1,7 @@
 import os
 import sys
 import logging
+import threading
 import json
 from datetime import datetime
 
@@ -13,6 +14,7 @@
 from src.LoginScreen import LoginScreen
 from src.ScannerScreen import ScannerScreen
 from src.SettingsScreen import SettingsScreen
+from src.CheckDriver import compare_and_download as driver_update
 from kivy.base import Builder
 
 Builder.load_file("./src/kv/Login.kv")
@@ -50,5 +52,13 @@ def Bypass_Reopen_Tabs():
 
 
 if __name__ == "__main__":
-    Bypass_Reopen_Tabs()
+    threading.Thread(
+        target=Bypass_Reopen_Tabs,
+        daemon=True,
+    ).start()
+
+    threading.Thread(
+        target=driver_update,
+        daemon=True,
+    ).start()
     LoginApp().run()

src/CheckDriver.py

@@ -0,0 +1,84 @@
+import subprocess
+import json
+from os.path import exists
+from os import remove
+from zipfile import ZipFile
+from urllib.request import urlretrieve
+
+
+def get_driver_version():
+    ps = "curl https://googlechromelabs.github.io/chrome-for-testing/LATEST_RELEASE_STABLE"
+    return subprocess.check_output(ps, shell=True).decode("utf-8")
+
+
+def get_download_link(version_number):
+    downloads_json = "curl https://googlechromelabs.github.io/chrome-for-testing/known-good-versions-with-downloads.json"
+    downloads = subprocess.check_output(downloads_json, shell=True)
+    link = None
+    # converts to json format and grabs only the versions dict
+    downloads = json.loads(downloads)["versions"]
+
+    # loops through dict looking for the correct version number from download list
+    # returns download link
+    for version in downloads:
+        if version["version"] == version_number:
+            platforms = version["downloads"]["chromedriver"]
+            for platform in platforms:
+                if platform["platform"] == "win64":
+                    link = platform["url"]
+    return link
+
+
+def check_file():
+    if exists("./src/driver_version.json"):
+        return
+    else:
+        with open("./src/driver_version.json", "w", encoding="utf-8") as openfile:
+            template = {"Chrome_Driver_Version": ""}
+            json.dump(template, openfile, indent=4)
+    return
+
+
+def get_local_driver_version():
+    version = None
+    try:
+        with open("./src/driver_version.json", "r", encoding="utf-8") as openfile:
+            json_object = json.load(openfile)
+            version = json_object["Chrome_Driver_Version"]
+    except KeyError:
+        version = ""
+    return version
+
+
+def update_version(version):
+    json_object = None
+    with open("./src/driver_version.json", "r", encoding="utf-8") as openfile:
+        json_object = json.load(openfile)
+
+    with open("./src/driver_version.json", "w", encoding="utf-8") as openfile:
+        json_object["Chrome_Driver_Version"] = version
+        json.dump(json_object, openfile, indent=4)
+
+
+def clean():
+    remove("./driver.zip")
+    return
+
+
+def compare_and_download():
+    check_file()
+    local_version = get_local_driver_version()
+    remote_version = get_driver_version()
+
+    if local_version.split(".")[0] == remote_version.split(".")[0]:
+        print("chrome driver up to date")
+        return
+
+    remote_download_link = get_download_link(remote_version)
+
+    urlretrieve(remote_download_link, "./driver.zip")
+
+    with ZipFile("./driver.zip", "r") as zipObj:
+        zipObj.extractall()
+    update_version(remote_version)
+    clean()

src/LoginScreen.py

@@ -68,12 +68,12 @@ def store_login(self, username=None, password=None):
             "remember": True,
         }
 
-        with open(r"src\database.json", "w") as openfile:
+        with open(r"src\database.json", "w", encoding="utf-8") as openfile:
             json.dump(login_object, openfile, indent=4)
 
     def reset_database(self):
         default_object = {}
-        with open(r"src\database.json", "w") as openfile:
+        with open(r"src\database.json", "w", encoding="utf-8") as openfile:
             json.dump(default_object, openfile, indent=4)
 
     def decrypt_pass(self, password, key):
@@ -84,7 +84,7 @@ def read_json(self):
         if not os.path.exists(r"src\database.json"):
             return None
 
-        with open(r"src\database.json", "r") as openfile:
+        with open(r"src\database.json", "r", encoding="utf-8") as openfile:
             json_object = json.load(openfile)
 
         return json_object
@@ -93,7 +93,7 @@ def do_login(self):
         username = self.ids.username.text
         password = self.ids.password.text
         # if nothing, need to store new login
-        #if "@bollebrands.com" not in username.lower() or password == "":
+        # if "@bollebrands.com" not in username.lower() or password == "":
         #    return
 
         if self.ids.login_checkbox.active:

src/MainWebDriver.py

@@ -8,6 +8,7 @@
 from selenium.webdriver.chrome.options import Options
 from src import Elements
 from src.state import State
+from src.CheckDriver import compare_and_download as update_driver
 from selenium.webdriver.chrome.service import (
     Service as ChromeService,
 )  # Similar thing for firefox also!
@@ -356,18 +357,33 @@ def login(
                 self.run_driver(chrome_service, chrome_options, True)
             else:
                 self.run_driver(chrome_service, chrome_options, False)
-        except exceptions.SessionNotCreatedException:
+        except (
+            exceptions.SessionNotCreatedException,
+            exceptions.SeleniumManagerException,
+        ):
             login_callback()
             Clock.schedule_once(
                 partial(
                     self.test_popup,
                     "Chrome Driver Issue",
-                    "  You need to disable the manual override.\n"
-                    + "                   If the issue persists\n"
-                    + "Update and manually set the chrome driver.",
+                    "  I've downloaded the new driver\n \
+                       Try again",
                 ),
                 1,
             )
+            update_driver()
+            return
+        except exceptions.WebDriverException as e:
+            login_callback()
+            if "cannot find chrome binary" in e.msg.lower():
+                Clock.schedule_once(
+                    partial(
+                        self.test_popup,
+                        "Chrome Issue",
+                        "Chrome isn't installed",
+                    ),
+                    1,
+                )
             return
         sleep(3)
 

src/SettingsScreen.py

@@ -22,7 +22,7 @@ def save_and_return(self):
             "Timeout_Avoidance": self.timeout_avoidance,
         }
 
-        with open(r"src\settings.json", "w",encoding="utf-8") as openfile:
+        with open(r"src\settings.json", "w", encoding="utf-8") as openfile:
             json.dump(json_object, openfile, indent=4)
 
         self.manager.transition = SlideTransition(direction="left")
@@ -33,16 +33,16 @@ def set_previous_screen(self, screen):
 
     def load_settings(self):
         if not os.path.exists(r"src\settings.json"):
-            with open(r".\src\settings.json", "w") as openfile:
+            with open(r".\src\settings.json", "w", encoding="utf-8") as openfile:
                 json_object = {
                     "Netsuite_SSO": "Replace with Netsuite SSO",
                     "Netsuite": "https://5230881.app.netsuite.com/app/center/card.nl?sc=-29&whence=",
-                    "Chrome_Driver": ["None", False],
+                    "Chrome_Driver": ["./chromedriver-win64/chromedriver.exe", False],
                     "Timeout_Avoidance": True,
                 }
                 json.dump(json_object, openfile, indent=4)
 
-        with open(r".\src\settings.json", "r") as openfile:
+        with open(r".\src\settings.json", "r", encoding="utf-8") as openfile:
             json_object = json.load(openfile)
 
         self.netsuite_sso = json_object["Netsuite_SSO"]

src/settings.json

@@ -2,7 +2,7 @@
     "Netsuite_SSO": "https://launcher.myapps.microsoft.com/api/signin/fd4bc304-663f-4e7b-9245-93596454cc99?tenantId=7261aa19-728c-457c-bb6e-a31f9a21516d",
     "Netsuite": "https://5230881.app.netsuite.com/app/center/card.nl?sc=-29&whence=",
     "Chrome_Driver": [
-        "C:\\chromedriver-win32\\chromedriver.exe",
+        "./chromedriver-win64/chromedriver.exe",
         true
     ],
     "Timeout_Avoidance": true