add compatibility for mkdocs-print-site-plugin

.devcontainer/Dockerfile

@@ -1,3 +1,3 @@
 FROM mcr.microsoft.com/vscode/devcontainers/python:3-bullseye
 
-RUN pip3 install wheel mkdocs mkdocs-material BeautifulSoup4
+RUN pip3 install wheel mkdocs mkdocs-material mkdocs-print-site-plugin BeautifulSoup4

example/mkdocs.yml

@@ -11,5 +11,6 @@ nav:
 plugins:
     - search
     - drawio
+    - print-site
 
 repo_url: https://github.com/tuunit/mkdocs-drawio

mkdocs_drawio/plugin.py

@@ -37,10 +37,15 @@ def __init__(self):
         self.pool = None
 
     def on_post_page(self, output_content, config, page, **kwargs):
+        return self.render_drawio_diagrams(output_content, page)
+
+    def render_drawio_diagrams(self, output_content, page):
         if ".drawio" not in output_content.lower():
             # Skip unecessary HTML parsing
             return output_content
 
+        plugin_config = self.config.copy()
+
         soup = BeautifulSoup(output_content, "html.parser")
 
         # search for images using drawio extension
@@ -49,7 +54,7 @@ def on_post_page(self, output_content, config, page, **kwargs):
             return output_content
 
         # add drawio library to body
-        lib = soup.new_tag("script", src=self.config["viewer_js"])
+        lib = soup.new_tag("script", src=plugin_config["viewer_js"])
         soup.body.append(lib)
 
         # substitute images with embedded drawio diagram
@@ -58,7 +63,7 @@ def on_post_page(self, output_content, config, page, **kwargs):
         for diagram in diagrams:
             diagram.replace_with(
                 BeautifulSoup(
-                    self.substitute_image(path, diagram["src"], diagram["alt"]),
+                    DrawioPlugin.substitute_image(path, diagram["src"], diagram["alt"]),
                     "html.parser",
                 )
             )
@@ -68,8 +73,8 @@ def on_post_page(self, output_content, config, page, **kwargs):
     @staticmethod
     def substitute_image(path: Path, src: str, alt: str):
         diagram_xml = etree.parse(path.joinpath(src).resolve())
-        diagram = self.parse_diagram(diagram_xml, alt)
-        escaped_xml = self.escape_diagram(diagram)
+        diagram = DrawioPlugin.parse_diagram(diagram_xml, alt)
+        escaped_xml = DrawioPlugin.escape_diagram(diagram)
 
         return SUB_TEMPLATE.substitute(xml_drawio=escaped_xml)
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mkdocs-drawio"
-version = "1.5.5"
+version = "1.6.0"
 description = "MkDocs plugin for embedding Drawio files"
 authors = [
     "Sergey Lukin <onixpro@gmail.com>",

setup.py

@@ -9,7 +9,7 @@ def read(name):
 
 setup(
     name="mkdocs-drawio",
-    version="1.5.5",
+    version="1.6.0",
     packages=find_packages(),
     url="https://github.com/tuunit/mkdocs-drawio",
     license="MIT",