seleniumbase 4.24.11__py3-none-any.whl → 4.33.15__py3-none-any.whl

seleniumbase/console_scripts/sb_behave_gui.py

@@ -16,16 +16,16 @@ Output:
 import colorama
 import subprocess
 import sys
+import tkinter as tk
+from seleniumbase.fixtures import shared_utils
+from tkinter.scrolledtext import ScrolledText
 
-if sys.version_info <= (3, 7):
+if sys.version_info <= (3, 8):
     current_version = ".".join(str(ver) for ver in sys.version_info[:3])
     raise Exception(
-        "\n* SBase Commander requires Python 3.7 or newer!"
+        "\n* SBase Commander requires Python 3.8 or newer!"
         "\n** You are currently using Python %s" % current_version
     )
-from seleniumbase.fixtures import shared_utils
-import tkinter as tk  # noqa: E402
-from tkinter.scrolledtext import ScrolledText  # noqa: E402
 
 
 def set_colors(use_colors):
@@ -38,13 +38,6 @@ def set_colors(use_colors):
     c6 = ""
     cr = ""
     if use_colors:
-        if (
-            shared_utils.is_windows()
-            and hasattr(colorama, "just_fix_windows_console")
-        ):
-            colorama.just_fix_windows_console()
-        else:
-            colorama.init(autoreset=True)
         c0 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
         c2 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX

seleniumbase/console_scripts/sb_caseplans.py

@@ -20,17 +20,17 @@ import colorama
 import os
 import subprocess
 import sys
+import tkinter as tk
+from seleniumbase.fixtures import shared_utils
+from tkinter import messagebox
+from tkinter.scrolledtext import ScrolledText
 
-if sys.version_info <= (3, 7):
+if sys.version_info <= (3, 8):
     current_version = ".".join(str(ver) for ver in sys.version_info[:3])
     raise Exception(
-        "\n* SBase Case Plans Generator requires Python 3.7 or newer!"
+        "\n* SBase Case Plans Generator requires Python 3.8 or newer!"
         "\n** You are currently using Python %s" % current_version
     )
-from seleniumbase.fixtures import shared_utils
-import tkinter as tk  # noqa: E402
-from tkinter import messagebox  # noqa: E402
-from tkinter.scrolledtext import ScrolledText  # noqa: E402
 
 
 def set_colors(use_colors):
@@ -42,13 +42,6 @@ def set_colors(use_colors):
     c5 = ""
     cr = ""
     if use_colors:
-        if (
-            shared_utils.is_windows()
-            and hasattr(colorama, "just_fix_windows_console")
-        ):
-            colorama.just_fix_windows_console()
-        else:
-            colorama.init(autoreset=True)
         c0 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
         c2 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX

seleniumbase/console_scripts/sb_commander.py

@@ -21,16 +21,16 @@ import colorama
 import os
 import subprocess
 import sys
+import tkinter as tk
+from seleniumbase.fixtures import shared_utils
+from tkinter.scrolledtext import ScrolledText
 
-if sys.version_info <= (3, 7):
+if sys.version_info <= (3, 8):
     current_version = ".".join(str(ver) for ver in sys.version_info[:3])
     raise Exception(
-        "\n* SBase Commander requires Python 3.7 or newer!"
+        "\n* SBase Commander requires Python 3.8 or newer!"
         "\n** You are currently using Python %s" % current_version
     )
-from seleniumbase.fixtures import shared_utils
-import tkinter as tk  # noqa: E402
-from tkinter.scrolledtext import ScrolledText  # noqa: E402
 
 
 def set_colors(use_colors):
@@ -42,13 +42,6 @@ def set_colors(use_colors):
     c5 = ""
     cr = ""
     if use_colors:
-        if (
-            shared_utils.is_windows()
-            and hasattr(colorama, "just_fix_windows_console")
-        ):
-            colorama.just_fix_windows_console()
-        else:
-            colorama.init(autoreset=True)
         c0 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
         c2 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX

seleniumbase/console_scripts/sb_install.py

@@ -2,28 +2,28 @@
 Downloads the specified webdriver to "seleniumbase/drivers/"
 
 Usage:
-         sbase get {chromedriver|geckodriver|edgedriver|
-                    iedriver|uc_driver} [OPTIONS]
+    sbase get {chromedriver|geckodriver|edgedriver|
+               iedriver|uc_driver} [OPTIONS]
 Options:
-         VERSION         Specify the version.
-                         Tries to detect the needed version.
-                         If using chromedriver or edgedriver,
-                         you can use the major version integer.
-         -p OR --path    Also copy the driver to /usr/local/bin
+    VERSION         Specify the version.
+                    Tries to detect the needed version.
+                    If using chromedriver or edgedriver,
+                    you can use the major version integer.
+    -p OR --path    Also copy the driver to /usr/local/bin
 Examples:
-         sbase get chromedriver
-         sbase get geckodriver
-         sbase get edgedriver
-         sbase get chromedriver 114
-         sbase get chromedriver 114.0.5735.90
-         sbase get chromedriver stable
-         sbase get chromedriver beta
-         sbase get chromedriver -p
+    sbase get chromedriver
+    sbase get geckodriver
+    sbase get edgedriver
+    sbase get chromedriver 114
+    sbase get chromedriver 114.0.5735.90
+    sbase get chromedriver stable
+    sbase get chromedriver beta
+    sbase get chromedriver -p
 Output:
-         Downloads the webdriver to seleniumbase/drivers/
-         (chromedriver is required for Chrome automation)
-         (geckodriver is required for Firefox automation)
-         (edgedriver is required for MS__Edge automation)
+    Downloads the webdriver to seleniumbase/drivers/
+    (chromedriver is required for Chrome automation)
+    (geckodriver is required for Firefox automation)
+    (edgedriver is required for MS__Edge automation)
 """
 import colorama
 import logging
@@ -50,39 +50,40 @@ IS_WINDOWS = shared_utils.is_windows()
 DRIVER_DIR = os.path.dirname(os.path.realpath(drivers.__file__))
 LOCAL_PATH = "/usr/local/bin/"  # On Mac and Linux systems
 DEFAULT_CHROMEDRIVER_VERSION = "114.0.5735.90"  # (If can't find LATEST_STABLE)
-DEFAULT_GECKODRIVER_VERSION = "v0.34.0"
+DEFAULT_GECKODRIVER_VERSION = "v0.35.0"
 DEFAULT_EDGEDRIVER_VERSION = "115.0.1901.183"  # (If can't find LATEST_STABLE)
 
 
 def invalid_run_command():
     exp = "  ** get / install **\n\n"
     exp += "  Usage:\n"
-    exp += "           seleniumbase install [DRIVER] [OPTIONS]\n"
-    exp += "           OR     sbase install [DRIVER] [OPTIONS]\n"
-    exp += "           OR  seleniumbase get [DRIVER] [OPTIONS]\n"
-    exp += "           OR         sbase get [DRIVER] [OPTIONS]\n"
-    exp += "                (Drivers: chromedriver, geckodriver, edgedriver,\n"
-    exp += "                          iedriver, uc_driver)\n"
+    exp += "     seleniumbase install [DRIVER_NAME] [OPTIONS]\n"
+    exp += "     OR     sbase install [DRIVER_NAME] [OPTIONS]\n"
+    exp += "     OR  seleniumbase get [DRIVER_NAME] [OPTIONS]\n"
+    exp += "     OR         sbase get [DRIVER_NAME] [OPTIONS]\n"
+    exp += "         (Drivers: chromedriver, geckodriver,\n"
+    exp += "                   edgedriver, iedriver, uc_driver)\n"
     exp += "  Options:\n"
-    exp += "           VERSION        Specify the version.\n"
-    exp += "                          Tries to detect the needed version.\n"
-    exp += "                          If using chromedriver or edgedriver,\n"
-    exp += "                          you can use the major version integer.\n"
-    exp += "           -p OR --path   Also copy the driver to /usr/local/bin\n"
+    exp += "     VERSION    Specify the version.\n"
+    exp += "                Tries to detect the needed version.\n"
+    exp += "                If using chromedriver or edgedriver,\n"
+    exp += "                you can use the major version integer.\n"
+    exp += "\n"
+    exp += "     -p OR --path   Also copy the driver to /usr/local/bin\n"
     exp += "  Examples:\n"
-    exp += "           sbase get chromedriver\n"
-    exp += "           sbase get geckodriver\n"
-    exp += "           sbase get edgedriver\n"
-    exp += "           sbase get chromedriver 114\n"
-    exp += "           sbase get chromedriver 114.0.5735.90\n"
-    exp += "           sbase get chromedriver stable\n"
-    exp += "           sbase get chromedriver beta\n"
-    exp += "           sbase get chromedriver -p\n"
+    exp += "     sbase get chromedriver\n"
+    exp += "     sbase get geckodriver\n"
+    exp += "     sbase get edgedriver\n"
+    exp += "     sbase get chromedriver 114\n"
+    exp += "     sbase get chromedriver 114.0.5735.90\n"
+    exp += "     sbase get chromedriver stable\n"
+    exp += "     sbase get chromedriver beta\n"
+    exp += "     sbase get chromedriver -p\n"
     exp += "  Output:\n"
-    exp += "          Downloads the webdriver to seleniumbase/drivers/\n"
-    exp += "          (chromedriver is required for Chrome automation)\n"
-    exp += "          (geckodriver is required for Firefox automation)\n"
-    exp += "          (edgedriver is required for MS__Edge automation)\n"
+    exp += "     Downloads the webdriver to seleniumbase/drivers/\n"
+    exp += "     (chromedriver is required for Chrome automation)\n"
+    exp += "     (geckodriver is required for Firefox automation)\n"
+    exp += "     (edgedriver is required for MS__Edge automation)\n"
     print("")
     raise Exception("%s\n\n%s" % (constants.Warnings.INVALID_RUN_COMMAND, exp))
 
@@ -293,10 +294,6 @@ def main(override=None, intel_for_uc=None, force_uc=None):
     use_version = ""
     new_file = ""
     f_name = ""
-    if IS_WINDOWS and hasattr(colorama, "just_fix_windows_console"):
-        colorama.just_fix_windows_console()
-    else:
-        colorama.init(autoreset=True)
     c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
     c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
     c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
@@ -578,10 +575,18 @@ def main(override=None, intel_for_uc=None, force_uc=None):
             else:
                 invalid_run_command()
         if IS_MAC:
-            file_name = "geckodriver-%s-macos.tar.gz" % use_version
+            if IS_ARM_MAC:
+                file_name = "geckodriver-%s-macos-aarch64.tar.gz" % use_version
+            else:
+                file_name = "geckodriver-%s-macos.tar.gz" % use_version
         elif IS_LINUX:
             if "64" in ARCH:
-                file_name = "geckodriver-%s-linux64.tar.gz" % use_version
+                if "aarch64" in platform.processor():
+                    file_name = (
+                        "geckodriver-%s-linux-aarch64.tar.gz" % use_version
+                    )
+                else:
+                    file_name = "geckodriver-%s-linux64.tar.gz" % use_version
             else:
                 file_name = "geckodriver-%s-linux32.tar.gz" % use_version
         elif IS_WINDOWS:
@@ -806,7 +811,7 @@ def main(override=None, intel_for_uc=None, force_uc=None):
         zip_ref.extractall(downloads_folder)
         zip_ref.close()
         os.remove(zip_file_path)
-        shutil.copyfile(driver_path, os.path.join(downloads_folder, filename))
+        shutil.copy3(driver_path, os.path.join(downloads_folder, filename))
         log_d("%sUnzip Complete!%s\n" % (c2, cr))
         to_remove = [
             "%s/%s/ruby_example/Gemfile" % (downloads_folder, h_ie_fn),
@@ -936,7 +941,10 @@ def main(override=None, intel_for_uc=None, force_uc=None):
                         f_name = "uc_driver"
                 new_file = os.path.join(downloads_folder, str(f_name))
                 pr_file = c3 + new_file + cr
-                log_d("The file [%s] was saved to:\n%s\n" % (f_name, pr_file))
+                d_folder = os.sep.join(pr_file.split(os.sep)[:-1]) + os.sep
+                d_file = pr_file.split(os.sep)[-1]
+                d_ff = c3 + d_folder + cr + "\n" + c3 + d_file + cr
+                log_d("The file [%s] was saved to:\n%s\n" % (f_name, d_ff))
                 log_d("Making [%s %s] executable ..." % (f_name, use_version))
                 make_executable(new_file)
                 log_d(
@@ -945,7 +953,7 @@ def main(override=None, intel_for_uc=None, force_uc=None):
                 )
                 if copy_to_path and os.path.exists(LOCAL_PATH):
                     path_file = LOCAL_PATH + f_name
-                    shutil.copyfile(new_file, path_file)
+                    shutil.copy2(new_file, path_file)
                     make_executable(path_file)
                     log_d("Also copied to: %s%s%s" % (c3, path_file, cr))
             log_d("")
@@ -1034,7 +1042,7 @@ def main(override=None, intel_for_uc=None, force_uc=None):
             )
             if copy_to_path and os.path.exists(LOCAL_PATH):
                 path_file = LOCAL_PATH + f_name
-                shutil.copyfile(new_file, path_file)
+                shutil.copy2(new_file, path_file)
                 make_executable(path_file)
                 log_d("Also copied to: %s%s%s" % (c3, path_file, cr))
             log_d("")
@@ -1070,7 +1078,7 @@ def main(override=None, intel_for_uc=None, force_uc=None):
                 )
                 if copy_to_path and os.path.exists(LOCAL_PATH):
                     path_file = LOCAL_PATH + f_name
-                    shutil.copyfile(new_file, path_file)
+                    shutil.copy2(new_file, path_file)
                     make_executable(path_file)
                     log_d("Also copied to: %s%s%s" % (c3, path_file, cr))
             log_d("")

seleniumbase/console_scripts/sb_mkchart.py

@@ -31,22 +31,22 @@ import sys
 def invalid_run_command(msg=None):
     exp = "  ** mkchart **\n\n"
     exp += "  Usage:\n"
-    exp += "          seleniumbase mkchart [FILE.py] [LANG]\n"
-    exp += "          OR     sbase mkchart [FILE.py] [LANG]\n"
+    exp += "     seleniumbase mkchart [FILE.py] [LANG]\n"
+    exp += "     OR     sbase mkchart [FILE.py] [LANG]\n"
     exp += "  Example:\n"
-    exp += "          sbase mkchart new_chart.py --en\n"
+    exp += "     sbase mkchart new_chart.py --en\n"
     exp += "  Language Options:\n"
-    exp += "          --en / --English    |    --zh / --Chinese\n"
-    exp += "          --nl / --Dutch      |    --fr / --French\n"
-    exp += "          --it / --Italian    |    --ja / --Japanese\n"
-    exp += "          --ko / --Korean     |    --pt / --Portuguese\n"
-    exp += "          --ru / --Russian    |    --es / --Spanish\n"
+    exp += "     --en / --English    |    --zh / --Chinese\n"
+    exp += "     --nl / --Dutch      |    --fr / --French\n"
+    exp += "     --it / --Italian    |    --ja / --Japanese\n"
+    exp += "     --ko / --Korean     |    --pt / --Portuguese\n"
+    exp += "     --ru / --Russian    |    --es / --Spanish\n"
     exp += "  Output:\n"
-    exp += "          Creates a new SeleniumBase chart presentation.\n"
-    exp += "          If the file already exists, an error is raised.\n"
-    exp += "          By default, the slides are written in English,\n"
-    exp += '          and use a "sky" theme with "slide" transition.\n'
-    exp += "          The chart can be used as a basic boilerplate.\n"
+    exp += "     Creates a new SeleniumBase chart presentation.\n"
+    exp += "     If the file already exists, an error is raised.\n"
+    exp += "     By default, the slides are written in English,\n"
+    exp += '     and use a "sky" theme with "slide" transition.\n'
+    exp += "     The chart can be used as a basic boilerplate.\n"
     if not msg:
         raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
     elif msg == "help":
@@ -62,13 +62,6 @@ def main():
     c7 = ""
     cr = ""
     if "linux" not in sys.platform:
-        if (
-            "win32" in sys.platform
-            and hasattr(colorama, "just_fix_windows_console")
-        ):
-            colorama.just_fix_windows_console()
-        else:
-            colorama.init(autoreset=True)
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
         c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA

seleniumbase/console_scripts/sb_mkdir.py

@@ -27,18 +27,18 @@ import sys
 def invalid_run_command(msg=None):
     exp = "  ** mkdir **\n\n"
     exp += "  Usage:\n"
-    exp += "          seleniumbase mkdir [DIRECTORY] [OPTIONS]\n"
-    exp += "          OR     sbase mkdir [DIRECTORY] [OPTIONS]\n"
+    exp += "     seleniumbase mkdir [DIRECTORY] [OPTIONS]\n"
+    exp += "     OR     sbase mkdir [DIRECTORY] [OPTIONS]\n"
     exp += "  Example:\n"
-    exp += "          sbase mkdir ui_tests\n"
+    exp += "     sbase mkdir ui_tests\n"
     exp += "  Options:\n"
-    exp += "          -b / --basic  (Only config files. No tests added.)\n"
+    exp += "     -b / --basic  (Only config files. No tests added.)\n"
     exp += "  Output:\n"
-    exp += "          Creates a new folder for running SBase scripts.\n"
-    exp += "          The new folder contains default config files,\n"
-    exp += "          sample tests for helping new users get started,\n"
-    exp += "          and Python boilerplates for setting up customized\n"
-    exp += "          test frameworks.\n"
+    exp += "     Creates a new folder for running SBase scripts.\n"
+    exp += "     The new folder contains default config files,\n"
+    exp += "     sample tests for helping new users get started,\n"
+    exp += "     and Python boilerplates for setting up customized\n"
+    exp += "     test frameworks.\n"
     if not msg:
         raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
     elif msg == "help":
@@ -54,13 +54,6 @@ def main():
     c7 = ""
     cr = ""
     if "linux" not in sys.platform:
-        if (
-            "win32" in sys.platform
-            and hasattr(colorama, "just_fix_windows_console")
-        ):
-            colorama.just_fix_windows_console()
-        else:
-            colorama.init(autoreset=True)
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
         c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA
@@ -127,7 +120,7 @@ def main():
 
     data = []
     data.append("[pytest]")
-    data.append("addopts = --capture=no -p no:cacheprovider")
+    data.append("addopts = --capture=tee-sys -p no:cacheprovider")
     data.append("norecursedirs = .* build dist recordings temp assets")
     data.append("filterwarnings =")
     data.append("    ignore::pytest.PytestWarning")
@@ -266,6 +259,7 @@ def main():
     data.append("report_archives")
     data.append("archived_reports")
     data.append("html_report.html")
+    data.append("last_report.html")
     data.append("report.html")
     data.append("report.xml")
     data.append("dashboard.html")

seleniumbase/console_scripts/sb_mkfile.py

@@ -9,9 +9,10 @@ Example:
     sbase mkfile new_test.py
 
 Options:
+    --uc  (UC Mode boilerplate using SB context manager)
     -b / --basic  (Basic boilerplate / single-line test)
-    -r / --rec  (adds Pdb+ breakpoint for Recorder Mode)
-    --url=URL  (makes the test start on a specific page)
+    -r / --rec  (Adds Pdb+ breakpoint for Recorder Mode)
+    --url=URL  (Makes the test start on a specific page)
 
 Language Options:
     --en / --English    |    --zh / --Chinese
@@ -37,6 +38,7 @@ Output:
     and "assert_text". If using the basic boilerplate
     option, only the "open" method is included. Only the
     BaseCase format supports Languages or Recorder Mode.
+    UC Mode automatically uses English with SB() format.
 """
 import codecs
 import colorama
@@ -47,36 +49,38 @@ import sys
 def invalid_run_command(msg=None):
     exp = "  ** mkfile **\n\n"
     exp += "  Usage:\n"
-    exp += "           seleniumbase mkfile [FILE.py] [OPTIONS]\n"
-    exp += "           OR     sbase mkfile [FILE.py] [OPTIONS]\n"
+    exp += "     seleniumbase mkfile [FILE.py] [OPTIONS]\n"
+    exp += "     OR     sbase mkfile [FILE.py] [OPTIONS]\n"
     exp += "  Example:\n"
-    exp += "           sbase mkfile new_test.py\n"
+    exp += "     sbase mkfile new_test.py\n"
     exp += "  Options:\n"
-    exp += "           -b / --basic  (Basic boilerplate / single-line test)\n"
-    exp += "           -r / --rec  (adds Pdb+ breakpoint for Recorder Mode)\n"
-    exp += "           --url=URL  (makes the test start on a specific page)\n"
+    exp += "     --uc  (UC Mode boilerplate using SB context manager)\n"
+    exp += "     -b / --basic  (Basic boilerplate / single-line test)\n"
+    exp += "     -r / --rec  (Adds Pdb+ breakpoint for Recorder Mode)\n"
+    exp += "     --url=URL  (Makes the test start on a specific page)\n"
     exp += "  Language Options:\n"
-    exp += "           --en / --English    |    --zh / --Chinese\n"
-    exp += "           --nl / --Dutch      |    --fr / --French\n"
-    exp += "           --it / --Italian    |    --ja / --Japanese\n"
-    exp += "           --ko / --Korean     |    --pt / --Portuguese\n"
-    exp += "           --ru / --Russian    |    --es / --Spanish\n"
+    exp += "     --en / --English    |    --zh / --Chinese\n"
+    exp += "     --nl / --Dutch      |    --fr / --French\n"
+    exp += "     --it / --Italian    |    --ja / --Japanese\n"
+    exp += "     --ko / --Korean     |    --pt / --Portuguese\n"
+    exp += "     --ru / --Russian    |    --es / --Spanish\n"
     exp += "  Syntax Formats:\n"
-    exp += "           --bc / --basecase       (BaseCase class inheritance)\n"
-    exp += "           --pf / --pytest-fixture          (sb pytest fixture)\n"
-    exp += "           --cf / --class-fixture   (class + sb pytest fixture)\n"
-    exp += "           --cm / --context-manager        (SB context manager)\n"
-    exp += "           --dc / --driver-context      (DriverContext manager)\n"
-    exp += "           --dm / --driver-manager             (Driver manager)\n"
+    exp += "     --bc / --basecase       (BaseCase class inheritance)\n"
+    exp += "     --pf / --pytest-fixture          (sb pytest fixture)\n"
+    exp += "     --cf / --class-fixture   (class + sb pytest fixture)\n"
+    exp += "     --cm / --context-manager        (SB context manager)\n"
+    exp += "     --dc / --driver-context      (DriverContext manager)\n"
+    exp += "     --dm / --driver-manager             (Driver manager)\n"
     exp += "  Output:\n"
-    exp += "           Creates a new SBase test file with boilerplate code.\n"
-    exp += "           If the file already exists, an error is raised.\n"
-    exp += "           By default, uses English with BaseCase inheritance,\n"
-    exp += "           and creates a boilerplate with common SeleniumBase\n"
-    exp += '           methods: "open", "type", "click", "assert_element",\n'
-    exp += '           and "assert_text". If using the basic boilerplate\n'
-    exp += '           option, only the "open" method is included. Only the\n'
-    exp += "           BaseCase format supports Languages or Recorder Mode.\n"
+    exp += "     Creates a new SBase test file with boilerplate code.\n"
+    exp += "     If the file already exists, an error is raised.\n"
+    exp += "     By default, uses English with BaseCase inheritance,\n"
+    exp += "     and creates a boilerplate with common SeleniumBase\n"
+    exp += '     methods: "open", "type", "click", "assert_element",\n'
+    exp += '     and "assert_text". If using the basic boilerplate\n'
+    exp += '     option, only the "open" method is included. Only the\n'
+    exp += "     BaseCase format supports Languages or Recorder Mode.\n"
+    exp += "     UC Mode automatically uses English with SB() format.\n"
     if not msg:
         raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
     elif msg == "help":
@@ -92,19 +96,13 @@ def main():
     c7 = ""
     cr = ""
     if "linux" not in sys.platform:
-        if (
-            "win32" in sys.platform
-            and hasattr(colorama, "just_fix_windows_console")
-        ):
-            colorama.just_fix_windows_console()
-        else:
-            colorama.init(autoreset=True)
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
         c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA
         cr = colorama.Style.RESET_ALL
 
     basic = False
+    use_uc = False
     help_me = False
     recorder = False
     error_msg = None
@@ -152,6 +150,9 @@ def main():
                 recorder = True
             elif option == "--record" or option == "--recorder":
                 recorder = True
+            elif use_uc:
+                # UC must use English & ContextManager formats
+                continue
             elif option == "--en" or option == "--english":
                 language = "English"
             elif option == "--zh" or option == "--chinese":
@@ -184,6 +185,11 @@ def main():
                 syntax = "DriverContext"
             elif option == "--dm" or option == "--driver-manager":
                 syntax = "DriverManager"
+            elif option == "--uc":
+                basic = True
+                language = "English"
+                syntax = "ContextManager"
+                use_uc = True
             else:
                 invalid_cmd = "\n===> INVALID OPTION: >> %s <<\n" % option
                 invalid_cmd = invalid_cmd.replace(">> ", ">>" + c5 + " ")
@@ -319,16 +325,22 @@ def main():
         data = []
         data.append("from seleniumbase import SB")
         data.append("")
-        data.append('with SB(browser="chrome") as sb:')
-        data.append(
-            '    sb.open("data:text/html,<div>Hello<br><input></div>")'
-        )
+        if use_uc:
+            data.append('with SB(uc=True) as sb:')
+        else:
+            data.append('with SB(browser="chrome") as sb:')
+        if use_uc:
+            data.append('    url = "%s"' % url)
+            data.append("    sb.uc_open_with_reconnect(url, 4)")
+            data.append("    sb.uc_gui_click_captcha()")
+        else:
+            data.append('    sb.open("%s")' % url)
         if not basic:
             data.append('    sb.type("input", "Goodbye")  # selector, text')
-            data.append('    sb.click("html body > div")  # selector')
+            data.append('    sb.click("html body > p")  # selector')
             data.append('    sb.assert_element("input")  # selector')
-            data.append('    sb.assert_text("Hello", "div")  # text, selector')
-            data.append('    sb.highlight("div")  # selector')
+            data.append('    sb.assert_text("Hello", "p")  # text, selector')
+            data.append('    sb.highlight("p")  # selector')
             data.append("    sb.sleep(0.5)  # seconds")
         data.append("")
         new_data = data
@@ -337,7 +349,14 @@ def main():
         data.append("from seleniumbase import DriverContext")
         data.append("")
         data.append('with DriverContext(browser="chrome") as driver:')
-        data.append('    driver.get("data:text/html,<p>Hello<br><input>")')
+        data.append('    driver.get("%s")' % url)
+        if not basic:
+            data.append('    driver.type("input", "Goodbye")  # sel, text')
+            data.append('    driver.click("html body > p")  # selector')
+            data.append('    driver.assert_element("input")  # selector')
+            data.append('    driver.assert_text("Hello", "p")  # text, sel')
+            data.append('    driver.highlight("p")  # selector')
+            data.append("    driver.sleep(0.5)  # seconds")
         data.append("")
         new_data = data
     elif language == "English" and syntax == "DriverManager":
@@ -346,7 +365,14 @@ def main():
         data.append("")
         data.append('driver = Driver(browser="chrome")')
         data.append("try:")
-        data.append('    driver.get("data:text/html,<p>Hello<br><input>")')
+        data.append('    driver.get("%s")' % url)
+        if not basic:
+            data.append('    driver.type("input", "Goodbye")  # sel, text')
+            data.append('    driver.click("html body > p")  # selector')
+            data.append('    driver.assert_element("input")  # selector')
+            data.append('    driver.assert_text("Hello", "p")  # text, sel')
+            data.append('    driver.highlight("p")  # selector')
+            data.append("    driver.sleep(0.5)  # seconds")
         data.append("finally:")
         data.append("    driver.quit()")
         data.append("")

seleniumbase/console_scripts/sb_mkpres.py

@@ -31,22 +31,22 @@ import sys
 def invalid_run_command(msg=None):
     exp = "  ** mkpres **\n\n"
     exp += "  Usage:\n"
-    exp += "          seleniumbase mkpres [FILE.py] [LANG]\n"
-    exp += "          OR     sbase mkpres [FILE.py] [LANG]\n"
+    exp += "     seleniumbase mkpres [FILE.py] [LANG]\n"
+    exp += "     OR     sbase mkpres [FILE.py] [LANG]\n"
     exp += "  Example:\n"
-    exp += "          sbase mkpres new_presentation.py --en\n"
+    exp += "     sbase mkpres new_presentation.py --en\n"
     exp += "  Language Options:\n"
-    exp += "          --en / --English    |    --zh / --Chinese\n"
-    exp += "          --nl / --Dutch      |    --fr / --French\n"
-    exp += "          --it / --Italian    |    --ja / --Japanese\n"
-    exp += "          --ko / --Korean     |    --pt / --Portuguese\n"
-    exp += "          --ru / --Russian    |    --es / --Spanish\n"
+    exp += "     --en / --English    |    --zh / --Chinese\n"
+    exp += "     --nl / --Dutch      |    --fr / --French\n"
+    exp += "     --it / --Italian    |    --ja / --Japanese\n"
+    exp += "     --ko / --Korean     |    --pt / --Portuguese\n"
+    exp += "     --ru / --Russian    |    --es / --Spanish\n"
     exp += "  Output:\n"
-    exp += "          Creates a new presentation with 3 example slides.\n"
-    exp += "          If the file already exists, an error is raised.\n"
-    exp += "          By default, the slides are written in English,\n"
-    exp += '          and use "serif" theme with "slide" transition.\n'
-    exp += "          The slides can be used as a basic boilerplate.\n"
+    exp += "     Creates a new presentation with 3 example slides.\n"
+    exp += "     If the file already exists, an error is raised.\n"
+    exp += "     By default, the slides are written in English,\n"
+    exp += '     and use "serif" theme with "slide" transition.\n'
+    exp += "     The slides can be used as a basic boilerplate.\n"
     if not msg:
         raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
     elif msg == "help":
@@ -62,13 +62,6 @@ def main():
     c7 = ""
     cr = ""
     if "linux" not in sys.platform:
-        if (
-            "win32" in sys.platform
-            and hasattr(colorama, "just_fix_windows_console")
-        ):
-            colorama.just_fix_windows_console()
-        else:
-            colorama.init(autoreset=True)
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
         c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA