109 lines
2.9 KiB
Python
109 lines
2.9 KiB
Python
#!/usr/local/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import json
|
|
import requests
|
|
|
|
from heremaps.heremapsexceptions import BadGeocodingParams
|
|
from heremaps.heremapsexceptions import EmptyGeocoderResponse
|
|
from heremaps.heremapsexceptions import NoGeocodingParams
|
|
from heremaps.heremapsexceptions import MalformedResult
|
|
|
|
|
|
class Geocoder:
|
|
'A Here Maps Geocoder wrapper for python'
|
|
|
|
PRODUCTION_GEOCODE_JSON_URL = 'https://geocoder.api.here.com/6.2/geocode.json'
|
|
STAGING_GEOCODE_JSON_URL = 'https://geocoder.cit.api.here.com/6.2/geocode.json'
|
|
DEFAULT_MAXRESULTS = 1
|
|
DEFAULT_GEN = 9
|
|
|
|
ADDRESS_PARAMS = [
|
|
'city',
|
|
'country',
|
|
'county',
|
|
'district',
|
|
'housenumber',
|
|
'postalcode',
|
|
'searchtext',
|
|
'state',
|
|
'street'
|
|
]
|
|
|
|
ADMITTED_PARAMS = [
|
|
'additionaldata',
|
|
'app_id',
|
|
'app_code',
|
|
'bbox',
|
|
'countryfocus',
|
|
'gen',
|
|
'jsonattributes',
|
|
'jsoncallback',
|
|
'language',
|
|
'locationattributes',
|
|
'locationid',
|
|
'mapview',
|
|
'maxresults',
|
|
'pageinformation',
|
|
'politicalview',
|
|
'prox',
|
|
'strictlanguagemode'
|
|
] + ADDRESS_PARAMS
|
|
|
|
app_id = ''
|
|
app_code = ''
|
|
maxresults = ''
|
|
|
|
def __init__(self, app_id, app_code, maxresults=DEFAULT_MAXRESULTS,
|
|
gen=DEFAULT_GEN, host=PRODUCTION_GEOCODE_JSON_URL):
|
|
self.app_id = app_id
|
|
self.app_code = app_code
|
|
self.maxresults = maxresults
|
|
self.gen = gen
|
|
self.host = host
|
|
|
|
def geocode(self, params):
|
|
if not set(params.keys()).issubset(set(self.ADDRESS_PARAMS)):
|
|
raise BadGeocodingParams(params)
|
|
response = self.perform_request(params)
|
|
try:
|
|
results = response['Response']['View'][0]['Result']
|
|
except IndexError:
|
|
raise EmptyGeocoderResponse()
|
|
|
|
return results
|
|
|
|
def perform_request(self, params):
|
|
request_params = {
|
|
'app_id': self.app_id,
|
|
'app_code': self.app_code,
|
|
'maxresults': self.maxresults,
|
|
'gen': self.gen
|
|
}
|
|
request_params.update(params)
|
|
response = requests.get(self.host, params=request_params)
|
|
if response.status_code == requests.codes.ok:
|
|
return json.loads(response.text)
|
|
else:
|
|
response.raise_for_status()
|
|
|
|
def geocode_address(self, **kwargs):
|
|
params = {}
|
|
for key, value in kwargs.iteritems():
|
|
if value:
|
|
params[key] = value
|
|
if not params:
|
|
raise NoGeocodingParams()
|
|
return self.geocode(params)
|
|
|
|
def extract_lng_lat_from_result(self, result):
|
|
try:
|
|
location = result['Location']
|
|
except KeyError:
|
|
raise MalformedResult()
|
|
|
|
longitude = location['DisplayPosition']['Longitude']
|
|
latitude = location['DisplayPosition']['Latitude']
|
|
|
|
return [longitude, latitude]
|