🔄 synced local 'find_unused_media.py' with remote 'find_unused_media.py'

Enveloppe · Nov 6, 2023 · 43d765c · 43d765c
1 parent 5ad0be4
commit 43d765c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/find_unused_media.py b/find_unused_media.py
@@ -1,13 +1,17 @@
 from pathlib import Path
 import os
 import argparse
+from typing import Optional
 import yaml
 
 
-def find_unused_media(img_path: Path = None, dry_run: bool = False):
+def find_unused_media(img_path: Optional[Path] = None, dry_run: bool = False):
     # load mkdocs.yml
     with open("mkdocs.yml", "r", encoding="utf-8") as f:
-        config = yaml.load(f, Loader=yaml.UnsafeLoader)
+        # remove all !! pairs from the yaml file
+        data = f.read()
+        data = data.replace("!!", "")
+        config = yaml.safe_load(data)
 
     docs_dir = Path.cwd() / Path(config.get("docs_dir", "docs"))
     assets_dir = Path(docs_dir, config["extra"]["attachments"])