1 files changed, 33 insertions, 0 deletions

diff --git a/foursquare/query_poi.py b/foursquare/query_poi.py
new file mode 100644
index 0000000..efdd1b1
--- /dev/null
+++ b/foursquare/query_poi.py
@@ -0,0 +1,33 @@
+import requests
+import json
+from config import FSQ_API_KEY
+from dbstore.dbm_store import get_loc, store_loc
+
+POI_API_ENDPOINT = "https://api.foursquare.com/v3/places/nearby?ll={}%2C{}"
+OSM_ENDPOINT = "https://www.openstreetmap.org/?mlat={}&mlon={}&zoom=15&layers=M"
+
+def query_poi(latitude, longitude):
+    locations = list()
+
+    url = POI_API_ENDPOINT.format(latitude, longitude)
+    headers = {
+        "accept": "application/json",
+        "Authorization": FSQ_API_KEY
+    }
+
+    response = requests.get(url, headers=headers)
+
+    for poi in json.loads(response.text)["results"]:
+        loc = {
+            "fsq_id": poi["fsq_id"],
+            "name": poi["name"],
+            "locality": poi["location"]["locality"],
+            "region": poi["location"]["region"],
+            "latitude": poi["geocodes"]["main"]["latitude"],
+            "longitude": poi["geocodes"]["main"]["longitude"],
+            "osm_url": OSM_ENDPOINT.format(poi["geocodes"]["main"]["latitude"], poi["geocodes"]["main"]["longitude"])
+        }
+        locations.append(loc)
+        store_loc(loc)
+
+    return locations