feat(renderer): narrow notes and warnings sections

Author: machow

quartodoc/renderers.py

@@ -30,16 +30,28 @@ def tuple_to_data(el: "tuple[ds.DocstringSectionKind, str]"):
 
 def docstring_section_narrow(el: ds.DocstringSection) -> ds.DocstringSection:
     # attempt to narrow down text sections
+    to_narrow = [
+        DocstringSectionSeeAlso,
+        DocstringSectionNotes,
+        DocstringSectionWarnings,
+    ]
     prefix = "See Also\n---"
-    if isinstance(el, ds.DocstringSectionText) and el.value.startswith(prefix):
-        stripped = el.value.replace(prefix, "", 1).lstrip("-\n")
-        return DocstringSectionSeeAlso(stripped, el.title)
+
+    if isinstance(el, ds.DocstringSectionText):
+        for cls in to_narrow:
+            prefix = cls.kind.value.title() + "\n---"
+            if el.value.lstrip("\n").startswith(prefix):
+
+                stripped = el.value.replace(prefix, "", 1).lstrip("-\n")
+                return cls(stripped, el.title)
 
     return el
 
 
 class DocstringSectionKindPatched(Enum):
     see_also = "see also"
+    notes = "notes"
+    warnings = "warnings"
 
 
 class DocstringSectionSeeAlso(ds.DocstringSection):
@@ -50,6 +62,22 @@ def __init__(self, value: str, title: "str | None"):
         super().__init__(title)
 
 
+class DocstringSectionNotes(ds.DocstringSection):
+    kind = DocstringSectionKindPatched.notes
+
+    def __init__(self, value: str, title: "str | None"):
+        self.value = value
+        super().__init__(title)
+
+
+class DocstringSectionWarnings(ds.DocstringSection):
+    kind = DocstringSectionKindPatched.warnings
+
+    def __init__(self, value: str, title: "str | None"):
+        self.value = value
+        super().__init__(title)
+
+
 @dataclass
 class ExampleCode:
     value: str