Retrieve title as well

1 files changed, 12 insertions, 3 deletions

diff --git a/wikimini/__init__.py b/wikimini/__init__.py
index 7dc573e..ca8bfab 100644
--- a/wikimini/__init__.py
+++ b/wikimini/__init__.py
@@ -17,7 +17,7 @@ def page_url(title):
     return title.replace(" ", "_")
 
 
-def retrieve(title: str) -> mwp.wikicode.Wikicode:
+def retrieve(title: str) -> (str, mwp.wikicode.Wikicode):
     params = {
         "action": "query",
         "prop": "revisions",
@@ -31,9 +31,11 @@ def retrieve(title: str) -> mwp.wikicode.Wikicode:
     headers = {"User-Agent": "Wikimini/1.0"}
     req = requests.get(API_URL, headers=headers, params=params)
     res = req.json()
-    revision = res["query"]["pages"][0]["revisions"][0]
+    page = res["query"]["pages"][0]
+    title = page["title"]
+    revision = page["revisions"][0]
     text = revision["slots"]["main"]["content"]
-    return mwp.parse(text)
+    return (title, mwp.parse(text))
 
 
 def render_convert(obj: mwp.nodes.template.Template) -> str:
@@ -178,10 +180,17 @@ def _convert(obj: Union[mwp.nodes.Node, mwp.wikicode.Wikicode]) -> str:
             return render_convert(obj)
         elif name in {"lang", "script"}:
             return _convert(obj.params[1].value)
+        elif name == "lang-ar":
+            return "Arabic {}/{}".format(
+                _convert(obj.params[0].value),
+                _convert(obj.params[1].value),
+            )
         elif name in {"reign", "r.", "ruled", "rexit"}:
             return render_reign(obj)
         elif name in {"cite book", "cite journal", "cite news"}:
             return render_cite_book(obj)
+        elif name.startswith("ipa"):
+            return "pronounced [{}]".format(_convert(obj.params[0].value))
         elif name == "main":
             return render_main(obj)
         else: