Add geographic search query URL generator

1 files changed, 20 insertions, 0 deletions

diff --git a/src/downloader/__init__.py b/src/downloader/__init__.py
index 3a02ebe..f256ddc 100644
--- a/src/downloader/__init__.py
+++ b/src/downloader/__init__.py
@@ -106,3 +106,23 @@ class WikimediaAPI():
             "list": "recentchanges",
             "rcnamespace": WikimediaAPI.RCNAMESPACES_CODES[namespace],
         })
+
+    def get_geo_pages(self, lat, long, radius, limit=500, namespace="(Main)"):
+        """
+        Generate the query URL performing a geographic search of articles
+        located near the given coordinates.
+
+        :param lat: Latitude
+        :param long: Longitude
+        :param radius: radius (in m) to look inside
+        :param limit: maximum number of results (max 500)
+        :param namespace: namespace to restrict the search in
+        :return:
+        """
+        return self.gen_query_url({
+            "list": "geosearch",
+            "gscoord": "%f|%f" % (lat, long),
+            "gsradius": "%f" % radius,
+            "gsnamespace": WikimediaAPI.RCNAMESPACES_CODES[namespace],
+            "gslimit": limit,
+        })