Extend config scheme with templates_path

Author: sindrehan

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()