From fbd697535186a8390cfd7a281efb522fd23fb573 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Sat, 4 Aug 2018 12:48:38 +0100
Subject: Remove debug print, improve wikipedia.py

---
 modules/wikipedia.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

(limited to 'modules')

diff --git a/modules/wikipedia.py b/modules/wikipedia.py
index 0b1d3401..98f61714 100644
--- a/modules/wikipedia.py
+++ b/modules/wikipedia.py
@@ -11,13 +11,15 @@ class Module(object):
     def wikipedia(self, event):
         page = Utils.get_url(URL_WIKIPEDIA, get_params={
             "action": "query", "prop": "extracts",
-            "titles": event["args"], "exsentences": "2",
-            "explaintext": "", "formatversion": "2",
-            "format": "json"}, json=True)
+            "titles": event["args"], "exintro": "",
+            "explaintext": "", "exchars": "500",
+            "redirects": "", "format": "json"}, json=True)
         if page:
-            if not "missing" in page["query"]["pages"][0]:
-                article = page["query"]["pages"][0]
+            pages = page["query"]["pages"]
+            article = list(pages.items())[0][1]
+            if not "missing" in article:
                 title, info = article["title"], article["extract"]
+                info = info.replace("\n\n", " ").split("\n")[0]
                 event["stdout"].write("%s: %s" % (title, info))
             else:
                 event["stderr"].write("No results found")
-- 
cgit v1.3.1-10-gc9f91