import os
from dotenv import load_dotenv
import googlemaps


class GoogleMapsAPI:
    def __init__(self):
        load_dotenv()
        self.client = googlemaps.Client(key=os.getenv("GOOGLE_MAPS_API_KEY"))

    def address_validation(self, address):
        addressvalidation_result = self.client.addressvalidation([address], regionCode='DE')
        address_info = addressvalidation_result.get("result", {}).get("address", {})

        # Stelle sicher, dass eine formatierte Adresse vorhanden ist
        formatted_address = address_info.get("formattedAddress", None)
        if not formatted_address:
            return None  # Keine gültige Adresse gefunden

        # Überprüfe, ob "locality" oder "postal_town" vorhanden ist (steht für Stadt)
        components = address_info.get("addressComponents", [])
        has_city = any(c.get("componentType") in ["locality", "postal_town"] for c in components)

        return formatted_address if has_city else None