Extract method from writing contents to document file

hypernode/common/docs.py

@@ -1,6 +1,9 @@
 import os.path
 from pathlib import Path
-from typing import List
+from typing import List, Optional
+
+import mdformat
+import yaml
 
 from hypernode.common.settings import DOCS_DIR
 
@@ -15,3 +18,14 @@ def get_all_docs() -> List[Path]:
         for file in markdown_files:
             result.append(Path(root) / file)
     return result
+
+
+def write_doc(path: Path, contents: str, frontmatter: Optional[dict]) -> None:
+    if frontmatter:
+        fm_yaml = yaml.dump(frontmatter, default_flow_style=False)
+        contents = "---\n" + fm_yaml + "---\n\n" + contents
+
+    contents = mdformat.text(contents, extensions=["frontmatter", "myst"])
+
+    with open(path, mode="w", encoding="utf-8") as f:
+        f.write(contents)

hypernode/redirect/generate_redirects_from_source_links.py

@@ -3,10 +3,9 @@
 from typing import Optional
 from urllib.parse import urlparse
 
-import yaml
 from frontmatter import Frontmatter
 
-from hypernode.common.docs import get_all_docs
+from hypernode.common.docs import get_all_docs, write_doc
 
 SOURCE_PATTERN = re.compile(r"^<!-- source: (.+) -->$")
 
@@ -25,16 +24,13 @@ def set_source_path_redirect(doc: Path, source_path: str) -> None:
     fm = Frontmatter.read_file(doc)
     attributes = fm["attributes"] or {}
     attributes["redirect_from"] = [source_path]
-    fm_yaml = yaml.dump(attributes, default_flow_style=False)
 
     body = fm["body"]
     if not body:
         with open(doc, mode="r", encoding="utf-8") as f:
             body = f.read()
 
-    with open(doc, mode="w", encoding="utf-8") as f:
-        contents = "---\n" + fm_yaml + "---\n\n" + body
-        f.write(contents)
+    write_doc(doc, body, attributes)
 
 
 def main():