Render {{main|...}} template

1 files changed, 10 insertions, 0 deletions

diff --git a/wikimini/__init__.py b/wikimini/__init__.py
index 032639a..790170f 100644
--- a/wikimini/__init__.py
+++ b/wikimini/__init__.py
@@ -86,6 +86,14 @@ def render_cite_book(obj: mwp.nodes.template.Template) -> str:
     return "{} ({})".format(title, "; ".join(names))
 
 
+def render_main(obj: mwp.nodes.template.Template) -> str:
+    """Renders the {{main|...}} template."""
+    links = [
+        "=> {} {}".format(page_url(str(t.value)), t.value) for t in obj.params
+    ]
+    return "Main articles:\n{}\n".format("\n".join(links))
+
+
 def _convert(obj: Union[mwp.nodes.Node, mwp.wikicode.Wikicode]) -> str:
     default = lambda obj: mwp.wikicode.Wikicode([obj]).strip_code(collapse=False)
     # This does the actual conversion
@@ -160,6 +168,8 @@ def _convert(obj: Union[mwp.nodes.Node, mwp.wikicode.Wikicode]) -> str:
             return render_reign(obj)
         elif name in {"cite book", "cite journal", "cite news"}:
             return render_cite_book(obj)
+        elif name == "main":
+            return render_main(obj)
         else:
             return default(obj)
     elif isinstance(obj, mwp.nodes.wikilink.Wikilink):