Function main

seleniumbase/console_scripts/run.py:984–1356

()

def main():
    command = None
    command_args = None
    num_args = len(sys.argv)
    if num_args == 1:
        show_basic_usage()
        return
    elif num_args == 2:
        command = sys.argv[1]
        command_args = []
    elif num_args > 2:
        command = sys.argv[1]
        command_args = sys.argv[2:]
    command = command.lower()

    if command == "get" or command == "install":
        if len(command_args) >= 1:
            from seleniumbase.console_scripts import sb_install

            need_retry = False
            need_another_retry = False
            retry_msg_1 = "* Unable to download driver! Retrying in 3s..."
            retry_msg_2 = "** Unable to download driver! Retrying in 5s..."
            if " --proxy=" in " ".join(sys.argv):
                from seleniumbase.core import proxy_helper
                for arg in sys.argv:
                    if arg.startswith("--proxy="):
                        proxy_string = arg.split("--proxy=")[1]
                        if "@" in proxy_string:
                            proxy_string = proxy_string.split("@")[1]
                        proxy_helper.validate_proxy_string(proxy_string)
                        break
            try:
                settings.HIDE_DRIVER_DOWNLOADS = False
                sb_install.main()
            except Exception as e:
                invalid_run_cmd = constants.Warnings.INVALID_RUN_COMMAND
                if invalid_run_cmd in str(e):
                    raise
                print()
                print(retry_msg_1)
                time.sleep(3)
                print()
                need_retry = True
            if need_retry:
                try:
                    sb_install.main()
                except Exception:
                    print(retry_msg_2)
                    time.sleep(5)
                    print()
                    need_another_retry = True
            if need_another_retry:
                sb_install.main()
        else:
            show_basic_usage()
            show_install_usage()
    elif command == "commander" or command == "gui":