diff --git a/src/docgen/generators/html.py b/src/docgen/generators/html.py index 7728eb5..4b0c511 100644 --- a/src/docgen/generators/html.py +++ b/src/docgen/generators/html.py @@ -1,10 +1,12 @@ -"""HTML documentation generator.""" +{"""HTML documentation generator.""" import re from pathlib import Path + from jinja2 import Environment, FileSystemLoader, select_autoescape -from docgen.models import DocConfig, Endpoint + from docgen.generators import BaseGenerator +from docgen.models import DocConfig, Endpoint def slugify(text: str) -> str: @@ -16,6 +18,7 @@ class HTMLGenerator(BaseGenerator): """Generator for Stripe-like interactive HTML documentation.""" def __init__(self, config: DocConfig = None): + """Initialize the HTML generator.""" super().__init__(config) template_dir = Path(__file__).parent.parent / "templates" self.env = Environment( @@ -58,7 +61,7 @@ class HTMLGenerator(BaseGenerator): else: parts = endpoint.path.strip("/").split("/") tag = parts[0] if parts else "Other" - + if tag not in grouped: grouped[tag] = [] grouped[tag].append(endpoint) @@ -69,7 +72,7 @@ class HTMLGenerator(BaseGenerator): static_dir = Path(__file__).parent.parent / "static" output_static = output_dir / "static" output_static.mkdir(exist_ok=True) - + for file in static_dir.iterdir(): if file.is_file(): (output_static / file.name).write_text(file.read_text())