lab_weather/geocoder/uscensus.py

122 lines
3.3 KiB
Python

#!/usr/bin/python
# coding: utf8
from __future__ import absolute_import
import re
import logging
from geocoder.base import OneResult, MultipleResultsQuery
class USCensusResult(OneResult):
def __init__(self, json_content):
# create safe shortcuts
self._coordinates = json_content.get('coordinates', {})
self._address_components = json_content.get('addressComponents', {})
# proceed with super.__init__
super(USCensusResult, self).__init__(json_content)
@property
def lat(self):
return self._coordinates.get('y')
@property
def lng(self):
return self._coordinates.get('x')
@property
def address(self):
return self.raw.get('matchedAddress')
@property
def housenumber(self):
if self.address:
match = re.search('^\d+', self.address, re.UNICODE)
if match:
return match.group(0)
@property
def fromhousenumber(self):
return self._address_components.get('fromAddress')
@property
def tohousenumber(self):
return self._address_components.get('toAddress')
@property
def streetname(self):
return self._address_components.get('streetName')
@property
def prequalifier(self):
return self._address_components.get('preQualifier')
@property
def predirection(self):
return self._address_components.get('preDirection')
@property
def pretype(self):
return self._address_components.get('preType')
@property
def suffixtype(self):
return self._address_components.get('suffixType')
@property
def suffixdirection(self):
return self._address_components.get('suffixDirection')
@property
def suffixqualifier(self):
return self._address_components.get('suffixQualifier')
@property
def city(self):
return self._address_components.get('city')
@property
def state(self):
return self._address_components.get('state')
@property
def postal(self):
return self._address_components.get('zip')
class USCensusQuery(MultipleResultsQuery):
"""
US Census Geocoder REST Services
=======================
The Census Geocoder is an address look-up tool that converts your address to an approximate coordinate (latitude/longitude) and returns information about the address range that includes the address and the census geography the address is within. The geocoder is available as a web interface and as an API (Representational State Transfer - REST - web-based service).
API Reference
-------------
https://geocoding.geo.census.gov/geocoder/Geocoding_Services_API.html
"""
provider = 'uscensus'
method = 'geocode'
_URL = 'https://geocoding.geo.census.gov/geocoder/locations/onelineaddress'
_RESULT_CLASS = USCensusResult
_KEY_MANDATORY = False
def _build_params(self, location, provider_key, **kwargs):
return {
'address': location,
'benchmark': kwargs.get('benchmark', '4'),
'format': 'json'
}
def _adapt_results(self, json_response):
return json_response['result']['addressMatches']
if __name__ == '__main__':
logging.basicConfig(level=logging.INFO)
g = USCensusQuery('4600 Silver Hill Road, Suitland, MD 20746', benchmark=9)
g.debug()