Formatting fixes in markdown conversion for formulas.

Author: jamalex

notion/markdown.py

@@ -81,9 +81,11 @@ def _extract_text_and_format_from_ast(item):
     if item["type"] == "html_inline":
         if item.get("literal", "") == "<s>":
             return "", ("s",)
-        if item.get("literal", "").startswith('<latex'):
-            elem = minidom.parseString(item.get("literal", "") + '</latex>').documentElement
-            equation = elem.attributes['equation'].value
+        if item.get("literal", "").startswith("<latex"):
+            elem = minidom.parseString(
+                item.get("literal", "") + "</latex>"
+            ).documentElement
+            equation = elem.attributes["equation"].value
             return "", ("e", equation)
 
     if item["type"] == "emph":
@@ -126,8 +128,15 @@ def markdown_to_notion(markdown):
 
     # commonmark doesn't support latex blocks, so we need to handle it ourselves
     def handle_latex(match):
-        return f'<latex equation="{html.escape(match.group(0)[2:-2])}">\u204d</latex>'
-    markdown = re.sub(r'(?<!\\\\|\$\$)(?:\\\\)*((\$\$)+)(?!(\$\$))(.+?)(?<!(\$\$))\1(?!(\$\$))', handle_latex, markdown)
+        return '<latex equation="{}">\u204d</latex>'.format(
+            html.escape(match.group(0)[2:-2])
+        )
+
+    markdown = re.sub(
+        r"(?<!\\\\|\$\$)(?:\\\\)*((\$\$)+)(?!(\$\$))(.+?)(?<!(\$\$))\1(?!(\$\$))",
+        handle_latex,
+        markdown,
+    )
 
     # we don't want to touch dashes, so temporarily replace them here
     markdown = markdown.replace("-", "⸻")
@@ -160,7 +169,7 @@ def handle_latex(match):
                 literal = ""
 
             if item["type"] == "html_inline" and literal == "</latex>":
-                for f in filter(lambda f: f[0] == 'e', format):
+                for f in filter(lambda f: f[0] == "e", format):
                     format.remove(f)
                     break
                 literal = ""
@@ -195,16 +204,18 @@ def handle_latex(match):
 
     return cleanup_dashes(consolidated)
 
+
 def cleanup_dashes(thing):
-    regex_pattern = re.compile('⸻|%E2%B8%BB')
+    regex_pattern = re.compile("⸻|%E2%B8%BB")
     if type(thing) is list:
         for counter, value in enumerate(thing):
             thing[counter] = cleanup_dashes(value)
     elif type(thing) is str:
-        return regex_pattern.sub('-', thing)
+        return regex_pattern.sub("-", thing)
 
     return thing
 
+
 def notion_to_markdown(notion):
 
     markdown_chunks = []
@@ -247,7 +258,7 @@ def notion_to_markdown(notion):
         # Check wheter a format modifies the content
         content_changed = False
         for f in sorted_format:
-            if f[0] == 'e':
+            if f[0] == "e":
                 markdown += f[1]
                 content_changed = True