Improve Markdown printing with "sbase print FILE.md"

Author: mdmintz

seleniumbase/console_scripts/rich_helper.py

@@ -15,6 +15,25 @@ def process_syntax(code, lang, theme, line_numbers, code_width, word_wrap):
     return syntax
 
 
+def get_code_without_tag(code, tag):
+    # Remove TAG from Code/HTML, but keep inner text.
+    # Eg: <a href="LINK">Hello!</a> becomes: "Hello!"
+    tag_start = "<%s " % tag
+    tag_solo = "<%s>" % tag
+    tag_end = "</%s>" % tag
+    while tag_start in code:
+        start = code.find(tag_start)
+        if start == -1:
+            break
+        end = code.find(">", start + 1) + 1
+        if end == 0:
+            break
+        code = code[:start] + code[end:]
+    code = code.replace(tag_solo, "")
+    code = code.replace(tag_end, "")
+    return code
+
+
 def display_markdown(code):
     try:
         markdown = Markdown(code)
@@ -38,6 +57,7 @@ def fix_emoji_spacing(code):
     try:
         # Fix the display width of certain emojis that take up two spaces
         double_width_emojis = [
+            "👁️",
             "🗺️",
             "🖼️",
             "🗄️",
@@ -63,6 +83,8 @@ def fix_emoji_spacing(code):
         for emoji in double_width_emojis:
             if emoji in code:
                 code = code.replace(emoji, emoji + " ")
+        code = code.replace("✅<", "✅ <")
+        code = code.replace("❌<", "❌ <")
     except Exception:
         pass
     return code

seleniumbase/console_scripts/sb_print.py

@@ -598,12 +598,6 @@ def main():
     print_success = False
     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>", "\n")
-        all_code = all_code.replace("</p>", "\n")
         if "<b>*" not in all_code and "*<b>" not in all_code:
             if "</b>*" not in all_code and "*</b>" not in all_code:
                 all_code = all_code.replace("<b>", "**")
@@ -617,6 +611,16 @@ def main():
         all_code = all_code.replace("\n<h2>", "\n# ").replace("</h2>", "")
         all_code = all_code.replace("\n<h3>", "\n# ").replace("</h3>", "")
         all_code = all_code.replace("\n<h4>", "\n# ").replace("</h4>", "")
+        all_code = rich_helper.get_code_without_tag(all_code, "summary")
+        all_code = rich_helper.get_code_without_tag(all_code, "details")
+        all_code = rich_helper.get_code_without_tag(all_code, "span")
+        all_code = rich_helper.get_code_without_tag(all_code, "div")
+        all_code = rich_helper.get_code_without_tag(all_code, "img")
+        all_code = rich_helper.get_code_without_tag(all_code, "li")
+        all_code = rich_helper.get_code_without_tag(all_code, "ul")
+        all_code = rich_helper.get_code_without_tag(all_code, "a")
+        all_code = rich_helper.get_code_without_tag(all_code, "p")
+        all_code = all_code.replace("<br />", "\n")
         print_success = rich_helper.display_markdown(all_code)
         if all_code.endswith("\n"):
             print()  # Because "rich" skips the last line if new-line