Add config for custom templates (#78)

Author: sindrehan

neoteroi/mkdocs/__init__.py

@@ -1 +1 @@
-__version__ = "1.1.3"
+__version__ = "1.2.0"

neoteroi/mkdocs/oad/__init__.py

@@ -11,6 +11,7 @@
 
 import re
 from pathlib import Path
+from typing import Optional
 
 from mkdocs.config.config_options import Type
 from mkdocs.plugins import BasePlugin
@@ -19,21 +20,30 @@
 
 
 class MkDocsOpenAPIDocumentationPlugin(BasePlugin):
-    config_scheme = (("use_pymdownx", Type(bool, default=False)),)
+    config_scheme = (
+        ("use_pymdownx", Type(bool, default=False)),
+        ("templates_path", Type(str, default=None)),
+    )
 
     rx = re.compile(r"\[OAD\(([^\)]+)\)\]")
 
     def _get_style(self) -> str:
         return "MKDOCS" if self.config.get("use_pymdownx", False) else "MARKDOWN"
 
+    def _get_templates_path(self) -> Optional[str]:
+        return self.config.get("templates_path", None)
+
     def _replacer(self, cwd):
         def replace(match) -> str:
             source = match.group(1).strip("'\"")
 
             data = read_from_source(source, cwd)
 
             handler = OpenAPIV3DocumentationHandler(
-                data, style=self._get_style(), source=source
+                data,
+                style=self._get_style(),
+                source=source,
+                templates_path=self._get_templates_path(),
             )
             return handler.write()
 

pyproject.toml

@@ -32,7 +32,7 @@ keywords = [
     "documentation",
 ]
 dependencies = [
-    "essentials-openapi>=1.2.1",
+    "essentials-openapi>=1.3.0",
     "mkdocs",
     "httpx",
     "click",

requirements.txt

@@ -1,6 +1,6 @@
 pytest
 pytest-cov
-essentials-openapi[full]>=1.2.1
+essentials-openapi[full]>=1.3.0
 mkdocs
 flake8
 black