[states] Moving state abbreviations config to YAML
This commit is contained in:
Al
@@ -1,93 +1,53 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from __future__ import unicode_literals
|
||||
import os
|
||||
import six
|
||||
import sys
|
||||
import yaml
|
||||
|
||||
from collections import defaultdict
|
||||
|
||||
STATE_ABBREVIATIONS = {
|
||||
'US': {
|
||||
'AL': {'en': 'Alabama'},
|
||||
'AK': {'en': 'Alaska'},
|
||||
'AZ': {'en': 'Arizona'},
|
||||
'AR': {'en': 'Arkansas'},
|
||||
'CA': {'en': 'California'},
|
||||
'CO': {'en': 'Colorado'},
|
||||
'CT': {'en': 'Connecticut'},
|
||||
'DE': {'en': 'Delaware'},
|
||||
'DC': {'en': 'District of Columbia'},
|
||||
'FL': {'en': 'Florida'},
|
||||
'GA': {'en': 'Georgia'},
|
||||
'HI': {'en': 'Hawaii'},
|
||||
'ID': {'en': 'Idaho'},
|
||||
'IL': {'en': 'Illinois'},
|
||||
'IN': {'en': 'Indiana'},
|
||||
'IA': {'en': 'Iowa'},
|
||||
'KS': {'en': 'Kansas'},
|
||||
'LA': {'en': 'Louisiana'},
|
||||
'KY': {'en': 'Kentucky'},
|
||||
'ME': {'en': 'Maine'},
|
||||
'MD': {'en': 'Maryland'},
|
||||
'MA': {'en': 'Massachusetts'},
|
||||
'MI': {'en': 'Michigan'},
|
||||
'MN': {'en': 'Minnesota'},
|
||||
'MS': {'en': 'Mississippi'},
|
||||
'MO': {'en': 'Missouri'},
|
||||
'MT': {'en': 'Montana'},
|
||||
'NE': {'en': 'Nebraska'},
|
||||
'NV': {'en': 'Nevada'},
|
||||
'NH': {'en': 'New Hampshire'},
|
||||
'NJ': {'en': 'New Jersey'},
|
||||
'NM': {'en': 'New Mexico'},
|
||||
'NY': {'en': 'New York'},
|
||||
'NC': {'en': 'North Carolina'},
|
||||
'ND': {'en': 'North Dakota'},
|
||||
'OH': {'en': 'Ohio'},
|
||||
'OK': {'en': 'Oklahoma'},
|
||||
'OR': {'en': 'Oregon'},
|
||||
'PA': {'en': 'Pennsylvania'},
|
||||
'RI': {'en': 'Rhode Island'},
|
||||
'SC': {'en': 'South Carolina'},
|
||||
'SD': {'en': 'South Dakota'},
|
||||
'TN': {'en': 'Tennessee'},
|
||||
'TX': {'en': 'Texas'},
|
||||
'UT': {'en': 'Utah'},
|
||||
'VT': {'en': 'Vermont'},
|
||||
'VA': {'en': 'Virginia'},
|
||||
'WA': {'en': 'Washington'},
|
||||
'WV': {'en': 'West Virginia'},
|
||||
'WI': {'en': 'Wisconsin'},
|
||||
'WY': {'en': 'Wyoming'}
|
||||
},
|
||||
'CA': {
|
||||
'AB': {'en': 'Alberta', 'fr': 'Alberta'},
|
||||
'BC': {'en': 'British Columbia', 'fr': 'Colombie-Britannique'},
|
||||
'MB': {'en': 'Manitoba', 'fr': 'Manitoba'},
|
||||
'NB': {'en': 'New Brunswick', 'fr': 'Nouveau-Brunswick'},
|
||||
'NL': {'en': 'Newfoundland and Labrador', 'fr': 'Terre-Neuve-et-Labrador'},
|
||||
'NS': {'en': 'Nova Scotia', 'fr': 'Nouvelle-Écosse'},
|
||||
'NT': {'en': 'Northwest Territories', 'fr': 'Territoires du Nord-Ouest'},
|
||||
'NU': {'en': 'Nunavut', 'fr': 'Nunavut'},
|
||||
'ON': {'en': 'Ontario', 'fr': 'Ontario'},
|
||||
'PE': {'en': 'Prince Edward Island', 'fr': 'Île-du-Prince-Édouard'},
|
||||
'QC': {'en': 'Quebec', 'fr': 'Québec'},
|
||||
'SK': {'en': 'Saskatchewan', 'fr': 'Saskatchewan'},
|
||||
'YT': {'en': 'Yukon', 'fr': 'Yukon'},
|
||||
},
|
||||
'AU': {
|
||||
'AAT': {'en': 'Australian Antarctic Territory'},
|
||||
'ACT': {'en': 'Australian Capital Territory'},
|
||||
'JBT': {'en': 'Jervis Bay Territory'},
|
||||
'NSW': {'en': 'New South Wales'},
|
||||
'NT': {'en': 'Northern Territory'},
|
||||
'QLD': {'en': 'Queensland'},
|
||||
'SA': {'en': 'South Australia'},
|
||||
'TAS': {'en': 'Tasmania'},
|
||||
'VIC': {'en': 'Victoria'},
|
||||
'WA': {'en': 'Western Australia'},
|
||||
},
|
||||
}
|
||||
this_dir = os.path.realpath(os.path.dirname(__file__))
|
||||
|
||||
STATE_EXPANSIONS = defaultdict(dict)
|
||||
from geodata.configs.utils import nested_get, DoesNotExist
|
||||
from geodata.encoding import safe_decode
|
||||
|
||||
for country, values in STATE_ABBREVIATIONS.iteritems():
|
||||
for abbrev, expansions in values.iteritems():
|
||||
for lang, expansion in expansions.iteritems():
|
||||
STATE_EXPANSIONS[country][expansion] = abbrev
|
||||
|
||||
STATE_DIR = os.path.join(this_dir, os.pardir, os.pardir, os.pardir,
|
||||
'resources', 'states')
|
||||
|
||||
|
||||
class StateAbbreviations(object):
|
||||
def __init__(self, base_dir=STATE_DIR):
|
||||
self.full_names = {}
|
||||
self.abbreviations = {}
|
||||
|
||||
for filename in os.listdir(base_dir):
|
||||
country = filename.split('.yaml')[0]
|
||||
country_config = yaml.load(open(os.path.join(base_dir, filename)))
|
||||
|
||||
country_abbreviations = defaultdict(dict)
|
||||
country_full_names = defaultdict(dict)
|
||||
|
||||
for abbreviation, vals in six.iteritems(country_config):
|
||||
for language, full_name in six.iteritems(vals):
|
||||
full_name = safe_decode(full_name)
|
||||
abbreviation = safe_decode(abbreviation)
|
||||
country_abbreviations[full_name.lower()][language] = abbreviation
|
||||
country_full_names[abbreviation][language] = full_name
|
||||
|
||||
self.abbreviations[country] = dict(country_abbreviations)
|
||||
self.full_names[country] = dict(country_full_names)
|
||||
|
||||
def get_abbreviation(self, country, language, state, default=None):
|
||||
value = nested_get(self.abbreviations, (country.lower(), state.lower(), language.lower()))
|
||||
if value is DoesNotExist:
|
||||
return None
|
||||
return value
|
||||
|
||||
def get_full_name(self, country, language, state, default=None):
|
||||
value = nested_get(self.full_names, (country.lower(), state.lower(), language.lower()))
|
||||
if value is DoesNotExist:
|
||||
return None
|
||||
return value
|
||||
|
||||
|
||||
state_abbreivations = StateAbbreviations()
|
||||
|
||||
Reference in New Issue
Block a user