1 files changed, 21 insertions, 4 deletions

diff --git a/foursquare/query_poi.py b/foursquare/poi.py
index e2d5625..722dbb7 100644
--- a/foursquare/query_poi.py
+++ b/foursquare/poi.py
@@ -6,17 +6,18 @@ from config import FSQ_API_KEY
 from dbstore.dbm_store import store_loc
 
 POI_API_ENDPOINT = "https://api.foursquare.com/v3/places/nearby?ll={}%2C{}"
+POI_PHOTO_ENDPOINT = "https://api.foursquare.com/v3/places/{}/photos?sort=POPULAR&limit=10"
 OSM_ENDPOINT = "https://www.openstreetmap.org/?mlat={}&mlon={}&zoom=15&layers=M"
+headers = {
+    "accept": "application/json",
+    "Authorization": FSQ_API_KEY
+}
 
 
 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)
 
@@ -34,3 +35,19 @@ def query_poi(latitude, longitude):
         store_loc(loc)
 
     return locations
+
+
+def get_poi_top_photo(fsq_id):
+    url = POI_PHOTO_ENDPOINT.format(fsq_id)
+    response = requests.get(url, headers=headers)
+
+    poi_photo_urls = []
+    for poi in json.loads(response.text):
+        prefix = poi["prefix"]
+        suffix = poi["suffix"]
+        poi_photo_urls.append(prefix + "original" + suffix)
+
+    if len(poi_photo_urls) == 0:
+        return None
+
+    return poi_photo_urls[0]