Refactoring

Author: mdmintz

seleniumbase/console_scripts/rich_helper.py

@@ -38,7 +38,15 @@ def fix_emoji_spacing(code):
     try:
         # Fix the display width of certain emojis that take up two spaces
         double_width_emojis = [
-            "🗺️", "🖼️", "🗄️", "⏺️", "♻️", "🗂️", "🖥️", "🕹️", "🎞️"
+            "🗺️",
+            "🖼️",
+            "🗄️",
+            "⏺️",
+            "♻️",
+            "🗂️",
+            "🖥️",
+            "🕹️",
+            "🎞️",
         ]
         for emoji in double_width_emojis:
             if emoji in code:

seleniumbase/console_scripts/sb_install.py

@@ -196,7 +196,7 @@ def main(override=None):
                 use_version = url_request.text
                 if get_latest_minus_one:
                     get_v_latest = True
-                    use_version = str(int(use_version.split('.')[0]) - 1)
+                    use_version = str(int(use_version.split(".")[0]) - 1)
         if get_v_latest:
             url_req = requests_get(last)
             if url_req.ok:

seleniumbase/console_scripts/sb_mkdir.py

@@ -362,9 +362,9 @@ def main():
     data.append('        self.assert_text("Automation Practice", "h3")')
     data.append("        try:")
     data.append("            self.hover_and_click(")
-    data.append('                "#myDropdown", "#dropOption2", timeout=2)')
+    data.append('                "#myDropdown", "#dropOption2", timeout=1)')
     data.append("        except Exception:")
-    data.append("            # If a human moves the mouse while the test runs")
+    data.append("            # If someone moves the mouse while the test runs")
     data.append('            self.js_click("#dropOption2")')
     data.append('        self.assert_text("Link Two Selected", "h3")')
     data.append('        self.assert_text("This Text is Green", "#pText")')
@@ -412,7 +412,8 @@ def main():
     data.append('        self.assert_true(self.is_selected(".fBox"))')
     data.append("        self.switch_to_default_content()")
     data.append(
-        '        self.assert_element_not_visible("div#drop2 img#logo")')
+        '        self.assert_element_not_visible("div#drop2 img#logo")'
+    )
     data.append('        self.drag_and_drop("img#logo", "div#drop2")')
     data.append('        self.assert_element("div#drop2 img#logo")')
     data.append('        self.assert_link_text("seleniumbase.com")')

seleniumbase/console_scripts/sb_mkfile.py

@@ -231,8 +231,8 @@ def main():
     if not recorder:
         data.append('        self.open("%s")' % url)
     else:
-        data.append('        if self.recorder_ext and not self.xvfb:')
-        data.append('            import ipdb; ipdb.set_trace()')
+        data.append("        if self.recorder_ext and not self.xvfb:")
+        data.append("            import ipdb; ipdb.set_trace()")
     if not basic and not recorder:
         data.append(
             '        self.type("input", "%s")' "  # selector, text" % goodbye

seleniumbase/console_scripts/sb_mkrec.py

@@ -116,7 +116,7 @@ def main():
     file_path = os.path.join(dir_name, file_name)
 
     if (
-        "--overwrite" in ' '.join(command_args).lower()
+        "--overwrite" in " ".join(command_args).lower()
         and os.path.exists(file_path)
     ):
         os.remove(file_path)
@@ -132,9 +132,7 @@ def main():
                 help_me = True
             elif option.lower() == "--edge":
                 use_edge = True
-            elif (
-                option.lower() in ("--gui", "--headed")
-            ):
+            elif option.lower() in ("--gui", "--headed"):
                 if "linux" in platform:
                     force_gui = True
             elif option.lower().startswith("--url="):
@@ -163,14 +161,14 @@ def main():
     data.append("")
     data.append("class RecorderTests(BaseCase):")
     data.append("    def test_recording(self):")
-    data.append('        if self.recorder_ext and not self.xvfb:')
-    data.append('            import ipdb; ipdb.set_trace()')
+    data.append("        if self.recorder_ext and not self.xvfb:")
+    data.append("            import ipdb; ipdb.set_trace()")
     data.append("")
     file = codecs.open(file_path, "w+", "utf-8")
     file.writelines("\r\n".join(data))
     file.close()
     success = (
-        "\n" + c0 + '* RECORDING initialized:' + cr + " "
+        "\n" + c0 + "* RECORDING initialized:" + cr + " "
         "" + c1 + file_name + "" + cr + "\n"
     )
     print(success)

seleniumbase/console_scripts/sb_print.py

@@ -595,12 +595,9 @@ def main():
     if use_rich and code_lang == "markdown" and not line_numbers:
         all_code = rich_helper.fix_emoji_spacing(all_code)
         all_code = all_code.replace("<br />", "\n")
-        all_code = all_code.replace(
-            '<p align="center">', '\n')
-        all_code = all_code.replace(
-            '<p align="left">', '\n')
-        all_code = all_code.replace(
-            '<p align="right">', '\n')
+        all_code = all_code.replace('<p align="center">', "\n")
+        all_code = all_code.replace('<p align="left">', "\n")
+        all_code = all_code.replace('<p align="right">', "\n")
         all_code = all_code.replace("<p>", "\n")
         all_code = all_code.replace("</p>", "\n")
         if "<b>*" not in all_code and "*<b>" not in all_code:

seleniumbase/console_scripts/sb_recorder.py

@@ -68,7 +68,8 @@ def do_recording(file_name, url, overwrite_enabled, use_chrome, window):
     error_msg = file_name_error(file_name)
     if error_msg:
         messagebox.showwarning(
-            "Invalid filename", "Invalid filename: %s" % error_msg)
+            "Invalid filename", "Invalid filename: %s" % error_msg
+        )
         return
 
     url = url.strip()
@@ -77,14 +78,15 @@ def do_recording(file_name, url, overwrite_enabled, use_chrome, window):
             url = "https://" + url
     if not page_utils.is_valid_url(url):
         messagebox.showwarning(
-            "Invalid URL", "Enter a valid URL! (Eg. seleniumbase.io)")
+            "Invalid URL", "Enter a valid URL! (Eg. seleniumbase.io)"
+        )
     else:
         if os.path.exists(os.getcwd() + "/" + file_name):
             if not overwrite_enabled:
                 msgbox = tk.messagebox.askquestion(
                     "Overwrite?",
                     'Are you sure you want to overwrite "%s"?' % file_name,
-                    icon="warning"
+                    icon="warning",
                 )
                 if msgbox == "yes":
                     os.remove(file_name)
@@ -105,12 +107,13 @@ def do_playback(file_name, use_chrome, window, demo_mode=False):
     error_msg = file_name_error(file_name)
     if error_msg:
         messagebox.showwarning(
-            "Invalid filename", "Invalid filename: %s" % error_msg)
+            "Invalid filename", "Invalid filename: %s" % error_msg
+        )
         return
     if not os.path.exists(os.getcwd() + "/" + file_name):
         messagebox.showwarning(
             "File does not exist",
-            'File "%s" does not exist in the current directory!' % file_name
+            'File "%s" does not exist in the current directory!' % file_name,
         )
         return
     command = "pytest %s -q -s" % file_name
@@ -154,28 +157,38 @@ def create_tkinter_gui():
     entry.focus()
     entry.bind(
         "<Return>",
-        (lambda _: do_recording(
-            fname.get(), url.get(), cbx.get(), cbb.get(), window))
+        (
+            lambda _: do_recording(
+                fname.get(), url.get(), cbx.get(), cbb.get(), window
+            )
+        ),
     )
     tk.Button(
-        window, text="Record", fg="red",
+        window,
+        text="Record",
+        fg="red",
         command=lambda: do_recording(
-            fname.get(), url.get(), cbx.get(), cbb.get(), window)
+            fname.get(), url.get(), cbx.get(), cbb.get(), window
+        ),
     ).pack()
     tk.Label(window, text="").pack()
-    tk.Label(
-        window, text="Playback recording (Normal Mode):").pack()
+    tk.Label(window, text="Playback recording (Normal Mode):").pack()
     tk.Button(
-        window, text="Playback", fg="green",
-        command=lambda: do_playback(fname.get(), cbb.get(), window)
+        window,
+        text="Playback",
+        fg="green",
+        command=lambda: do_playback(fname.get(), cbb.get(), window),
     ).pack()
     tk.Label(window, text="").pack()
-    tk.Label(
-        window, text="Playback recording (Demo Mode):").pack()
+    tk.Label(window, text="Playback recording (Demo Mode):").pack()
     tk.Button(
-        window, text="Playback (Demo Mode)", fg="teal",
+        window,
+        text="Playback (Demo Mode)",
+        fg="teal",
         command=lambda: do_playback(
-            fname.get(), cbb.get(), window, demo_mode=True)).pack()
+            fname.get(), cbb.get(), window, demo_mode=True
+        ),
+    ).pack()
 
     # Bring form window to front
     send_window_to_front(window)