final

input-to-route/libpostal_service.py

@@ -0,0 +1,38 @@
+from dotenv import load_dotenv
+import os
+import requests
+
+class LibpostalService:
+
+    def __init__(self):
+        load_dotenv()
+        self.libpostal_url = os.getenv("LIBPOSTAL_URL")
+
+    def parse_address(self, address_string):
+        if not self.libpostal_url:
+            raise ValueError("LIBPOSTAL_URL is not set in environment variables.")
+
+        url = f"{self.libpostal_url}/parse"
+        print(url)
+        params = {"address": address_string}
+        try:
+            response = requests.get(url, params=params)
+            response.raise_for_status()  # Raise an exception for HTTP errors
+            return response.json()
+        except requests.exceptions.RequestException as e:
+            print(f"Error parsing address: {e}")
+            return None
+
+    def expand_address(self, address_string):
+        if not self.libpostal_url:
+            raise ValueError("LIBPOSTAL_URL is not set in environment variables.")
+
+        url = f"{self.libpostal_url}/expand"
+        params = {"address": address_string}
+        try:
+            response = requests.get(url, params=params)
+            response.raise_for_status()  # Raise an exception for HTTP errors
+            return response.json()
+        except requests.exceptions.RequestException as e:
+            print(f"Error expanding address: {e}")
+            return None