Initial commit

1 files changed, 42 insertions, 0 deletions

diff --git a/banapedia/wapi/WikipediaQuery.py b/banapedia/wapi/WikipediaQuery.py
new file mode 100644
index 0000000..d3d2f94
--- /dev/null
+++ b/banapedia/wapi/WikipediaQuery.py
@@ -0,0 +1,42 @@
+from ..api.Query import JSONQuery
+
+__author__ = 'pacien'
+
+WIKIPEDIA_QUERY_BASE_URL = "https://en.wikipedia.org/w/api.php"
+LIST_SEPARATOR = "|"
+DEFAULT_BKPROP = ["id", "user", "userid", "by", "byid", "timestamp", "expiry", "reason", "range", "flags"]
+DEFAULT_BKSHOW = ["account", "temp", "ip", "range"]
+
+
+class WikipediaQuery(JSONQuery):
+    def __init__(self, params={}):
+        params.update({
+            "action": "query",
+            "format": "json",
+        })
+        JSONQuery.__init__(self, base_url=WIKIPEDIA_QUERY_BASE_URL, params=params)
+
+
+class ListQuery(WikipediaQuery):
+    def __init__(self, list_name, params={}):
+        params.update({
+            "list": list_name,
+        })
+        WikipediaQuery.__init__(self, params)
+
+
+class BlockQuery(ListQuery):
+    def __init__(self, bkprop=DEFAULT_BKPROP, bkshow=DEFAULT_BKSHOW, bkdir="newer", limit=500, continue_token=None):
+        params = {
+            "bkprop": LIST_SEPARATOR.join(bkprop),
+            "bkshow": LIST_SEPARATOR.join(bkshow),
+            "bkdir": bkdir,
+            "bklimit": limit,
+        }
+
+        if continue_token is not None:
+            params.update({"bkcontinue": continue_token})
+
+        ListQuery.__init__(self, "blocks", params=params)
+
+