import collections.abc import json from functools import partial from urllib.parse import urlencode from geopy.adapters import AdapterHTTPError from geopy.exc import GeocoderQueryError, GeocoderQuotaExceeded from geopy.geocoders.base import DEFAULT_SENTINEL, NONE_RESULT, Geocoder from geopy.location import Location from geopy.util import logger __all__ = ("Geocodio", ) class Geocodio(Geocoder): """Geocoder using the Geocod.io API. Documentation at: https://www.geocod.io/docs/ Pricing details: https://www.geocod.io/pricing/ .. versionadded:: 2.2 """ structured_query_params = { 'street', 'city', 'state', 'postal_code', 'country', } domain = 'api.geocod.io' geocode_path = '/v1.6/geocode' reverse_path = '/v1.6/reverse' def __init__( self, api_key, *, scheme=None, timeout=DEFAULT_SENTINEL, proxies=DEFAULT_SENTINEL, user_agent=None, ssl_context=DEFAULT_SENTINEL, adapter_factory=None ): """ :param str api_key: A valid Geocod.io API key. (https://dash.geocod.io/apikey/create) :param str scheme: See :attr:`geopy.geocoders.options.default_scheme`. :param int timeout: See :attr:`geopy.geocoders.options.default_timeout`. :param dict proxies: See :attr:`geopy.geocoders.options.default_proxies`. :param str user_agent: See :attr:`geopy.geocoders.options.default_user_agent`. :type ssl_context: :class:`ssl.SSLContext` :param ssl_context: See :attr:`geopy.geocoders.options.default_ssl_context`. :param callable adapter_factory: See :attr:`geopy.geocoders.options.default_adapter_factory`. """ super().__init__( scheme=scheme, timeout=timeout, proxies=proxies, user_agent=user_agent, ssl_context=ssl_context, adapter_factory=adapter_factory, ) self.api_key = api_key def geocode( self, query, *, limit=None, exactly_one=True, timeout=DEFAULT_SENTINEL ): """ Return a location point by address. :param query: The address, query or a structured query you wish to geocode. For a structured query, provide a dictionary whose keys are one of: `street`, `city`, `state`, `postal_code` or `country`. :type query: dict or str :param int limit: The maximum number of matches to return. This will be reset to 1 if ``exactly_one`` is ``True``. :param bool exactly_one: Return one result or a list of results, if available. :param int timeout: Time, in seconds, to wait for the geocoding service to respond before raising a :class:`geopy.exc.GeocoderTimedOut` exception. Set this only if you wish to override, on this call only, the value set during the geocoder's initialization. :rtype: ``None``, :class:`geopy.location.Location` or a list of them, if ``exactly_one=False``. """ if isinstance(query, collections.abc.Mapping): params = { key: val for key, val in query.items() if key in self.structured_query_params } else: params = {'q': query} params['api_key'] = self.api_key if limit: params['limit'] = limit if exactly_one: params['limit'] = 1 api = '%s://%s%s' % (self.scheme, self.domain, self.geocode_path) url = "?".join((api, urlencode(params))) logger.debug("%s.geocode: %s", self.__class__.__name__, url) callback = partial(self._parse_json, exactly_one=exactly_one) return self._call_geocoder(url, callback, timeout=timeout) def reverse( self, query, *, exactly_one=True, timeout=DEFAULT_SENTINEL, limit=None ): """Return an address by location point. :param str query: The coordinates for which you wish to obtain the closest human-readable addresses :param bool exactly_one: Return one result or a list of results, if available. :param int timeout: Time, in seconds, to wait for the geocoding service to respond before raising a :class:`geopy.exc.GeocoderTimedOut` exception. Set this only if you wish to override, on this call only, the value set during the geocoder's initialization. :param int limit: The maximum number of matches to return. This will be reset to 1 if ``exactly_one`` is ``True``. :rtype: ``None``, :class:`geopy.location.Location` or a list of them, if ``exactly_one=False``. """ params = { 'q': self._coerce_point_to_string(query), 'api_key': self.api_key } if exactly_one: limit = 1 if limit is not None: params['limit'] = limit api = '%s://%s%s' % (self.scheme, self.domain, self.reverse_path) url = "?".join((api, urlencode(params))) logger.debug("%s.reverse: %s", self.__class__.__name__, url) callback = partial(self._parse_json, exactly_one=exactly_one) return self._call_geocoder(url, callback, timeout=timeout) def _parse_json(self, page, exactly_one=True): """Returns location, (latitude, longitude) from json feed.""" places = page.get('results', []) if not places: return None def parse_place(place): """Get the location, lat, lng from a single json place.""" location = place.get('formatted_address') latitude = place['location']['lat'] longitude = place['location']['lng'] return Location(location, (latitude, longitude), place) if exactly_one: return parse_place(places[0]) else: return [parse_place(place) for place in places] def _geocoder_exception_handler(self, error): """Custom exception handling for invalid queries and exceeded quotas. Geocod.io returns a ``422`` status code for invalid queries, which is not mapped in :const:`~geopy.geocoders.base.ERROR_CODE_MAP`. The service also returns a ``403`` status code for exceeded quotas instead of the ``429`` code mapped in :const:`~geopy.geocoders.base.ERROR_CODE_MAP` """ if not isinstance(error, AdapterHTTPError): return if error.status_code is None or error.text is None: return if error.status_code == 422: error_message = self._get_error_message(error) if ( 'could not geocode address' in error_message.lower() and 'postal code or city required' in error_message.lower() ): return NONE_RESULT raise GeocoderQueryError(error_message) from error if error.status_code == 403: error_message = self._get_error_message(error) quota_exceeded_snippet = "You can't make this request as it is " \ "above your daily maximum." if quota_exceeded_snippet in error_message: raise GeocoderQuotaExceeded(error_message) from error def _get_error_message(self, error): """Try to extract an error message from the 'error' property of a JSON response. """ try: error_message = json.loads(error.text).get('error') except ValueError: error_message = None return error_message or error.text