Wikipedia package

bridges/python/Pipfile

@@ -8,5 +8,6 @@ requests = "==2.21.0"
 pytube = "==9.5.0"
 tinydb = "==3.9.0"
 beautifulsoup4 = "==4.7.1"
+wikipedia = "*"
 
 [dev-packages]

packages/wikipedia/README.md

@@ -0,0 +1,30 @@
+# Wikipedia Package
+
+The Wikipedia package allows your personal assistant to gather pieces of information from Wikipedia.
+
+#### Requirements
+- wikipedia
+
+#### Usage
+
+1. Duplicate the file `packages/wikipedia/config/config.sample.json` and rename it `config.json`.
+2. Set your language and other settings.
+3. This package uses wikipedia. To install it: from inside leon directory `cd bridges/python`
+4. `pipenv install wikipedia`
+5. Done!
+
+```
+(en-US)
+- "Search on Wikipedia GitHub"
+
+```
+
+#### Options
+- `lang`: Choose your language. Default: `en`
+- `sentences`: Number of sentences to report. No greater than 10.
+
+#### Links
+
+- [MediaWiki API](https://www.mediawiki.org/wiki/API:Main_page)
+- [Wikipedia GitHub](https://github.com/goldsmith/Wikipedia)
+- [Wikipedia Docs](https://wikipedia.readthedocs.io/en/latest/)

packages/wikipedia/config/config.sample.json

@@ -0,0 +1,7 @@
+{
+  "wikipedia": {
+    "lang": "en",
+    "sentences": 5,
+    "options": {}
+  }
+}

packages/wikipedia/data/answers/en.json

@@ -0,0 +1,25 @@
+{
+  "wikipedia": {
+    "acquiring": [
+      "Right away!",
+      "One second please.",
+      "Just a second...",
+      "Browsing Wikipedia right now..."
+    ],
+    "summary": [
+      "%summary%"
+    ],
+    "disambiguation_error": [
+      "Disambiguation error."
+    ],
+    "page_error": [
+      "Your request does not match any page."
+    ],
+    "no_entities_error": [
+      "Your request was not well formatted. Please, retry."
+    ],
+    "connection_error": [
+      "I'm having issuse reaching the server."
+    ]
+  }
+}

packages/wikipedia/data/expressions/en.json

@@ -0,0 +1,56 @@
+{
+  "wikipedia": {
+    "summary": {
+      "expressions": [
+        "Search on Wikipedia",
+        "Search on Wikipedia for",
+        "Search for ",
+        "Search",
+        "Look for",
+        "On wikipedia look for"
+      ],
+      "entities": [
+        {
+          "type": "trim",
+          "name": "page",
+          "conditions": [
+            {
+              "type": "after_last",
+              "from": "wikipedia"
+            },
+            {
+              "type": "between",
+              "from": "wikipedia",
+              "to": "?"
+            },
+            {
+              "type": "after_last",
+              "from": "for"
+            },
+            {
+              "type": "between",
+              "from": "for",
+              "to": "?"
+            },
+            {
+              "type": "between",
+              "from": "search",
+              "to": "on"
+            },
+            {
+              "type": "after_last",
+              "from": "search"
+            }
+          ]
+        }
+      ]
+    },
+    "random": {
+      "expressions": [
+        "Search a random page on Wikipedia",
+        "Search a random Wikipedia page",
+        "Give me a random page on Wikipedia"
+      ]
+    }
+  }
+}

packages/wikipedia/version.txt

@@ -0,0 +1 @@
+0.1.0

packages/wikipedia/wikipedia.py

@@ -0,0 +1,80 @@
+#!/usr/bin/env python
+# -*- coding:utf-8 -*-
+
+import wikipedia
+from wikipedia.exceptions import DisambiguationError, PageError
+
+import utils
+
+wikipedia.set_lang(utils.config("lang"))
+
+
+def summary(string, entities):
+    """
+    Get the summary of the requested Wikipedia page.
+    """
+
+    for entity in entities:
+        if entity["entity"] == "page":
+            utils.output(
+                "inter",
+                "acquiring",
+                utils.translate("acquiring")
+            )
+
+            try:
+                summary = wikipedia.summary(
+                    entity["sourceText"],
+                    sentences=utils.config("sentences")
+                )
+                return utils.output(
+                    "end",
+                    "summary",
+                    utils.translate(
+                        "summary",
+                        {"summary": summary}
+                    )
+                )
+
+            except DisambiguationError as pages:
+                # TODO: ask which page to visit
+                return utils.output(
+                    "end",
+                    "disambiguation_error",
+                    utils.translate("disambiguation_error")
+                )
+
+            except PageError:
+                return utils.output(
+                    "end",
+                    "page_error",
+                    utils.translate("page_error")
+                )
+
+    return utils.output(
+        "end",
+        "no_entities_error",
+        utils.translate("no_entities_error")
+    )
+
+
+def random(string, entities):
+    """
+    Get the summary of a random Wikipedia page.
+    """
+    utils.output(
+        "inter",
+        "acquiring",
+        utils.translate("acquiring")
+    )
+
+    summary = wikipedia.summary(title=wikipedia.random())
+
+    return utils.output(
+        "end",
+        "summary",
+        utils.translate(
+            "summary",
+            {"summary": summary}
+        )
+    )