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

seleniumbase/console_scripts/sb_mkrec.py

@@ -39,21 +39,21 @@ import sys
 def invalid_run_command(msg=None):
     exp = "  ** mkrec / record / codegen **\n\n"
     exp += "  Usage:\n"
-    exp += "           seleniumbase mkrec [FILE.py]\n"
-    exp += "           OR:    sbase mkrec [FILE.py]\n"
+    exp += "     seleniumbase mkrec [FILE.py]\n"
+    exp += "     OR:    sbase mkrec [FILE.py]\n"
     exp += "  Examples:\n"
-    exp += "           sbase mkrec new_test.py\n"
-    exp += "           sbase mkrec new_test.py --url=wikipedia.org\n"
+    exp += "     sbase mkrec new_test.py\n"
+    exp += "     sbase mkrec new_test.py --url=wikipedia.org\n"
     exp += "  Options:\n"
-    exp += "           --url=URL  (Sets the initial start page URL.)\n"
-    exp += "           --edge  (Use Edge browser instead of Chrome.)\n"
-    exp += "           --gui / --headed  (Use headed mode on Linux.)\n"
-    exp += "           --uc / --undetected  (Use undetectable mode.)\n"
-    exp += "           --overwrite  (Overwrite file when it exists.)\n"
-    exp += "           --behave  (Also output Behave/Gherkin files.)\n"
+    exp += "     --url=URL  (Sets the initial start page URL.)\n"
+    exp += "     --edge  (Use Edge browser instead of Chrome.)\n"
+    exp += "     --gui / --headed  (Use headed mode on Linux.)\n"
+    exp += "     --uc / --undetected  (Use undetectable mode.)\n"
+    exp += "     --overwrite  (Overwrite file when it exists.)\n"
+    exp += "     --behave  (Also output Behave/Gherkin files.)\n"
     exp += "  Output:\n"
-    exp += "           Creates a new SeleniumBase test using the Recorder.\n"
-    exp += "           If the filename already exists, an error is raised.\n"
+    exp += "     Creates a new SeleniumBase test using the Recorder.\n"
+    exp += "     If the filename already exists, an error is raised.\n"
     if not msg:
         raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
     elif msg == "help":
@@ -71,13 +71,6 @@ def set_colors(use_colors):
     c7 = ""
     cr = ""
     if use_colors:
-        if (
-            "win32" in sys.platform
-            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.RED + colorama.Back.LIGHTYELLOW_EX
         c2 = colorama.Fore.LIGHTRED_EX + colorama.Back.LIGHTYELLOW_EX
@@ -93,6 +86,7 @@ def main():
     invalid_cmd = None
     use_edge = False
     use_uc = False
+    esc_end = False
     start_page = None
     next_is_url = False
     use_colors = True
@@ -145,10 +139,14 @@ def main():
                 help_me = True
             elif option.lower() == "--edge":
                 use_edge = True
+            elif option.lower() == "--ee":
+                esc_end = True
             elif option.lower() in ("--gui", "--headed"):
                 if "linux" in sys.platform:
                     force_gui = True
-            elif option.lower() in ("--uc", "--undetected", "--undetectable"):
+            elif option.lower() in (
+                "--uc", "--cdp", "--undetected", "--undetectable"
+            ):
                 use_uc = True
             elif option.lower() in ("--rec-behave", "--behave", "--gherkin"):
                 rec_behave = True
@@ -178,11 +176,69 @@ def main():
     data.append("")
     data.append("class RecorderTest(BaseCase):")
     data.append("    def test_recording(self):")
+    if use_uc:
+        data.append("        if self.undetectable:")
+        if (
+            start_page
+            and (
+                start_page.startswith("http:")
+                or start_page.startswith("https:")
+                or start_page.startswith("file:")
+            )
+        ):
+            used_sp = start_page
+            if '"' not in start_page:
+                used_sp = '"%s"' % start_page
+            elif "'" not in start_page:
+                used_sp = "'%s'" % start_page
+            data.append(
+                "            self.uc_open_with_disconnect(\n"
+                "                %s\n"
+                "            )" % used_sp
+            )
+        else:
+            data.append("            self.disconnect()")
     data.append("        if self.recorder_ext:")
     data.append("            # When done recording actions,")
     data.append('            # type "c", and press [Enter].')
     data.append("            import pdb; pdb.set_trace()")
     data.append("")
+
+    if esc_end:
+        msg = ">>> Use [SHIFT + ESC] in the browser to end recording!"
+        d2 = []
+        d2.append("from seleniumbase import BaseCase")
+        d2.append("")
+        d2.append("")
+        d2.append("class RecorderTest(BaseCase):")
+        d2.append("    def test_recording(self):")
+        d2.append("        if self.recorder_ext:")
+        d2.append("            print(")
+        d2.append('                "\\n\\n%s\\n"' % msg)
+        d2.append("            )")
+        d2.append('            script = self._get_rec_shift_esc_script()')
+        d2.append('            esc = "return document.sb_esc_end;"')
+        d2.append("            start_time = self.time()")
+        d2.append("            last_handles_num = self._get_num_handles()")
+        d2.append("            for i in range(1200):")
+        d2.append("                try:")
+        d2.append("                    self.execute_script(script)")
+        d2.append("                    handles_num = self._get_num_handles()")
+        d2.append("                    if handles_num < 1:")
+        d2.append("                        return")
+        d2.append("                    elif handles_num != last_handles_num:")
+        d2.append("                        self.switch_to_window(-1)")
+        d2.append("                        last_handles_num = handles_num")
+        d2.append('                    if self.execute_script(esc) == "yes":')
+        d2.append("                        return")
+        d2.append("                    elif self.time() - start_time > 600:")
+        d2.append("                        return")
+        d2.append("                    self.sleep(0.5)")
+        d2.append("                except Exception:")
+        d2.append("                    return")
+        d2.append("")
+        data = d2
+
     file = codecs.open(file_path, "w+", "utf-8")
     file.writelines("\r\n".join(data))
     file.close()
@@ -192,7 +248,18 @@ def main():
     )
     print(success)
     run_cmd = None
-    if not start_page:
+    if (
+        not start_page
+        or (
+            use_uc
+            and (
+                start_page.startswith("http:")
+                or start_page.startswith("https:")
+                or start_page.startswith("file:")
+            )
+            and not esc_end
+        )
+    ):
         run_cmd = "%s -m pytest %s --rec -q -s" % (sys_executable, file_name)
     else:
         run_cmd = "%s -m pytest %s --rec -q -s --url=%s" % (

seleniumbase/console_scripts/sb_objectify.py

@@ -2,11 +2,11 @@
 Converts a SeleniumBase Python file into one that uses the Page Object Pattern.
 
 Usage:
-        seleniumbase objectify [SELENIUMBASE_PYTHON_FILE].py
+    seleniumbase objectify [SELENIUMBASE_PYTHON_FILE].py
 Output:
-        A modified version of the file where the selectors
-        have been replaced with variable names defined in
-        "page_objects.py", supporting the Page Object Pattern.
+    A modified version of the file where the selectors
+    have been replaced with variable names defined in
+    "page_objects.py", supporting the Page Object Pattern.
 """
 import codecs
 import os
@@ -35,57 +35,57 @@ def invalid_run_command(shell_command):
 def invalid_objectify_run_command():
     exp = "  ** objectify **\n\n"
     exp += "  Usage:\n"
-    exp += "         seleniumbase objectify [SELENIUMBASE_PYTHON_FILE]\n"
+    exp += "     seleniumbase objectify [SELENIUMBASE_PYTHON_FILE]\n"
     exp += "  Options:\n"
-    exp += "         -c, --comments  (Add object selectors to the comments.)\n"
-    exp += "                         (Default: No added comments.)\n"
+    exp += "     -c, --comments  (Add object selectors to the comments.)\n"
+    exp += "                     (Default: No added comments.)\n"
     exp += "  Output:\n"
-    exp += "         Converts a SeleniumBase Python file into one that uses\n"
-    exp += "         the Page Object Pattern by converting page selectors\n"
-    exp += '         into objects stored in a "page_objects.py" file that is\n'
-    exp += "         autogenerated and stored in the same folder as tests.\n"
-    exp += '         (seleniumbase "objectify" has the same outcome as\n'
-    exp += '         combining "extract-objects" with "inject-objects")\n'
+    exp += "     Converts a SeleniumBase Python file into one that uses\n"
+    exp += "     the Page Object Pattern by converting page selectors\n"
+    exp += '     into objects stored in a "page_objects.py" file that is\n'
+    exp += "     autogenerated and stored in the same folder as tests.\n"
+    exp += '     (seleniumbase "objectify" has the same outcome as\n'
+    exp += '     combining "extract-objects" with "inject-objects")\n'
     raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
 
 
 def invalid_inject_objects_run_command():
     exp = "  ** inject-objects **\n\n"
     exp += "  Usage:\n"
-    exp += "         seleniumbase inject-objects [SELENIUMBASE_PYTHON_FILE]\n"
+    exp += "     seleniumbase inject-objects [SELENIUMBASE_PYTHON_FILE]\n"
     exp += "  Options:\n"
-    exp += "         -c, --comments  (Add object selectors to the comments.)\n"
-    exp += "                         (Default: No added comments.)\n"
+    exp += "     -c, --comments  (Add object selectors to the comments.)\n"
+    exp += "                     (Default: No added comments.)\n"
     exp += "  Output:\n"
-    exp += '         Takes the page objects found in the "page_objects.py"\n'
-    exp += "         file and uses those to replace matching selectors in\n"
-    exp += "         the selected seleniumbase Python file.\n"
+    exp += '     Takes the page objects found in the "page_objects.py"\n'
+    exp += "     file and uses those to replace matching selectors in\n"
+    exp += "     the selected seleniumbase Python file.\n"
     raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
 
 
 def invalid_extract_objects_run_command():
     exp = "  ** extract-objects **\n\n"
     exp += "  Usage:\n"
-    exp += "         seleniumbase extract-objects [SELENIUMBASE_PYTHON_FILE]\n"
+    exp += "     seleniumbase extract-objects [SELENIUMBASE_PYTHON_FILE]\n"
     exp += "  Output:\n"
-    exp += "         Creates page objects based on selectors found in a\n"
-    exp += "         seleniumbase Python file and saves those objects to the\n"
-    exp += '         "page_objects.py" file in the same folder as the tests.\n'
+    exp += "     Creates page objects based on selectors found in a\n"
+    exp += "     seleniumbase Python file and saves those objects to the\n"
+    exp += '     "page_objects.py" file in the same folder as the tests.\n'
     raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
 
 
 def invalid_revert_objects_run_command():
     exp = "  ** revert-objects **\n\n"
     exp += "  Usage:\n"
-    exp += "         seleniumbase revert-objects [SELENIUMBASE_PYTHON_FILE]\n"
+    exp += "     seleniumbase revert-objects [SELENIUMBASE_PYTHON_FILE]\n"
     exp += "  Options:\n"
-    exp += "         -c, --comments  (Keep existing comments for the lines.)\n"
-    exp += "                         (Default: No comments are kept.)\n"
+    exp += "     -c, --comments  (Keep existing comments for the lines.)\n"
+    exp += "                     (Default: No comments are kept.)\n"
     exp += "  Output:\n"
-    exp += '         Reverts the changes made by "seleniumbase objectify" or\n'
-    exp += '         "seleniumbase inject-objects" when run against a\n'
-    exp += "         seleniumbase Python file. Objects will get replaced by\n"
-    exp += '         selectors stored in the "page_objects.py" file.\n'
+    exp += '     Reverts the changes made by "seleniumbase objectify" or\n'
+    exp += '     "seleniumbase inject-objects" when run against a\n'
+    exp += "     seleniumbase Python file. Objects will get replaced by\n"
+    exp += '     selectors stored in the "page_objects.py" file.\n'
     raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
 
 

seleniumbase/console_scripts/sb_print.py

@@ -18,13 +18,13 @@ import sys
 def invalid_run_command(msg=None):
     exp = "  ** print **\n\n"
     exp += "  Usage:\n"
-    exp += "         seleniumbase print [FILE] [OPTIONS]\n"
-    exp += "         OR:    sbase print [FILE] [OPTIONS]\n"
+    exp += "     seleniumbase print [FILE] [OPTIONS]\n"
+    exp += "     OR:    sbase print [FILE] [OPTIONS]\n"
     exp += "  Options:\n"
-    exp += "         -n   (Add line Numbers to the rows)\n"
+    exp += "     -n   (Add line Numbers to the rows)\n"
     exp += "  Output:\n"
-    exp += "         Prints the code/text of any file\n"
-    exp += "         with syntax-highlighting.\n"
+    exp += "     Prints the code/text of any file\n"
+    exp += "     with syntax-highlighting.\n"
     if not msg:
         raise Exception("INVALID RUN COMMAND!\n\n%s" % exp)
     else:
@@ -61,13 +61,6 @@ def get_width(line):
 
 
 def main():
-    if (
-        "win32" in sys.platform
-        and hasattr(colorama, "just_fix_windows_console")
-    ):
-        colorama.just_fix_windows_console()
-    else:
-        colorama.init(autoreset=True)
     c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
     c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA
     cr = colorama.Style.RESET_ALL

seleniumbase/console_scripts/sb_recorder.py

@@ -9,6 +9,7 @@ Usage:
 
 Options:
     --uc / --undetected  (Use undetectable mode.)
+    --cdp  (Same as "--uc" and "--undetectable".)
     --behave  (Also output Behave/Gherkin files.)
 
 Output:
@@ -18,23 +19,23 @@ import colorama
 import os
 import subprocess
 import sys
+import tkinter as tk
 from seleniumbase import config as sb_config
 from seleniumbase.fixtures import page_utils
 from seleniumbase.fixtures import shared_utils
+from tkinter import messagebox
 
 sb_config.rec_subprocess_p = None
 sb_config.rec_subprocess_used = False
 sys_executable = sys.executable
 if " " in sys_executable:
     sys_executable = "python"
-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* Recorder Desktop requires Python 3.7 or newer!"
+        "\n* Recorder Desktop requires Python 3.8 or newer!"
         "\n*** You are currently using Python %s" % current_version
     )
-import tkinter as tk  # noqa: E402
-from tkinter import messagebox  # noqa: E402
 
 
 def set_colors(use_colors):
@@ -45,13 +46,6 @@ def set_colors(use_colors):
     c4 = ""
     cr = ""
     if use_colors:
-        if (
-            "win32" in sys.platform
-            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
@@ -158,10 +152,13 @@ def do_recording(file_name, url, overwrite_enabled, use_chrome, window):
             command += " --edge"
         if (
             "--uc" in command_args
+            or "--cdp" in command_args
             or "--undetected" in command_args
             or "--undetectable" in command_args
         ):
             command += " --uc"
+        if "--ee" in command_args:
+            command += " --ee"
         command += add_on
         poll = None
         if sb_config.rec_subprocess_used:
@@ -195,6 +192,14 @@ def do_playback(file_name, use_chrome, window, demo_mode=False):
         command += " --edge"
     if demo_mode:
         command += " --demo"
+    command_args = sys.argv[2:]
+    if (
+        "--uc" in command_args
+        or "--cdp" in command_args
+        or "--undetected" in command_args
+        or "--undetectable" in command_args
+    ):
+        command += " --uc"
     poll = None
     if sb_config.rec_subprocess_used:
         poll = sb_config.rec_subprocess_p.poll()