Update sb_mkrec.py

Author: mdmintz

seleniumbase/console_scripts/sb_mkrec.py

@@ -43,7 +43,7 @@ def invalid_run_command(msg=None):
     exp += "  Options:\n"
     exp += "           --url=URL  (Sets the initial start page URL.)\n"
     exp += "           --edge  (Use Edge browser instead of Chrome.)\n"
-    exp += "           --ui or --headed  (For linux: use browser instead of headless.)\n"
+    exp += "           --gui / --headed  (Use headed mode on Linux.)\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"
@@ -55,6 +55,7 @@ def invalid_run_command(msg=None):
     else:
         raise Exception("INVALID RUN COMMAND!\n\n%s\n%s\n" % (exp, msg))
 
+
 def set_colors(use_colors):
     c0 = ""
     c1 = ""
@@ -74,16 +75,18 @@ def set_colors(use_colors):
 
 
 def main():
-
     help_me = False
     error_msg = None
     invalid_cmd = None
     use_edge = False
     start_page = None
     next_is_url = False
+    use_colors = True
+    force_gui = False
 
-    use_browser = ("linux" not in sys.platform)
-    c0, c1, c2, c5, c7, cr = set_colors(use_browser)
+    if "linux" in sys.platform:
+        use_colors = False
+    c0, c1, c2, c5, c7, cr = set_colors(use_colors)
 
     command_args = sys.argv[2:]
     file_name = command_args[0]
@@ -110,16 +113,20 @@ def main():
                 help_me = True
             elif option.lower() == "--edge":
                 use_edge = True
+            elif (
+                option.lower() in ("--gui", "--headed")
+                and "linux" in sys.platform
+            ):
+                use_colors = True
+                c0, c1, c2, c5, c7, cr = set_colors(use_colors)
+                force_gui = True
             elif option.lower().startswith("--url="):
                 start_page = option[len("--url="):]
             elif option.lower() == "--url":
                 next_is_url = True
             elif next_is_url:
                 start_page = option
                 next_is_url = False
-            elif option.lower() in ('--gui', '--headed'):
-                use_browser = True
-                c0, c1, c2, c5, c7, cr = set_colors(use_browser)
             else:
                 invalid_cmd = "\n===> INVALID OPTION: >> %s <<\n" % option
                 invalid_cmd = invalid_cmd.replace(">> ", ">>" + c5 + " ")
@@ -153,17 +160,18 @@ def main():
     print(success)
     if not start_page:
         run_cmd = "pytest %s --rec -q -s" % file_name
-        if use_browser:
-            if use_edge:
-                run_cmd += " --edge"
-            else:
-                run_cmd += " --chrome"
+        if use_edge:
+            run_cmd += " --edge"
+        if force_gui:
+            run_cmd += " --gui"
         print(run_cmd)
         os.system(run_cmd)
     else:
         run_cmd = "pytest %s --rec -q -s --url=%s" % (file_name, start_page)
         if use_edge:
             run_cmd += " --edge"
+        if force_gui:
+            run_cmd += " --gui"
         print(run_cmd)
         os.system(run_cmd)
     if os.path.exists(file_path):