Update api.py

Author: devlinwaluja

pipeline/api.py

@@ -1,4 +1,3 @@
-
 import requests 
 
 API_URL = "https://api.data.gov.sg/v1/transport/traffic-images"
@@ -9,6 +8,7 @@ def __init__(self, url="No API URL"):
         self.timestamp = None
         self.api_status = "Unverified"
         self.metadata = None
+        self.camera_id_array = []
 
         # Get API response
         response = requests.get(self.url)
@@ -24,6 +24,10 @@ def __init__(self, url="No API URL"):
         print(f"The API status is: {self.api_status}")
         print(f"The API was called at: {self.timestamp}")
 
+        for item in self.metadata["items"]:
+            for camera in item["cameras"]:
+                self.camera_id_array.append(camera["camera_id"])
+
     def extract_image(self, camera_id):
         # Loop through the items and cameras to find the correct camera_id
         for item in self.metadata["items"]:
@@ -33,6 +37,17 @@ def extract_image(self, camera_id):
         # If camera ID is not found
         return f"Camera ID {camera_id} not found."
 
+    def extract_latlon(self, camera_id):
+        for item in self.metadata["items"]:
+            for camera in item["cameras"]:
+                if camera["camera_id"] == str(camera_id):
+                    longitude = camera["location"]["longitude"]
+                    latitude = camera["location"]["latitude"]
+                    return longitude, latitude  # Return both longitude and latitude as a tuple
+        # If camera ID is not found
+        return f"Camera ID {camera_id} not found."
+
+