mirror of
https://github.com/github/octodns.git
synced 2024-05-11 05:55:00 +00:00
82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
#
|
|
#
|
|
#
|
|
|
|
from logging import getLogger
|
|
|
|
from .geo_data import geo_data
|
|
|
|
|
|
class GeoCodes(object):
|
|
log = getLogger('GeoCodes')
|
|
|
|
@classmethod
|
|
def validate(cls, code, prefix):
|
|
'''
|
|
Validates an octoDNS geo code making sure that it is a valid and
|
|
corresponding:
|
|
* continent
|
|
* continent & country
|
|
* continent, country, & province
|
|
'''
|
|
reasons = []
|
|
|
|
pieces = code.split('-')
|
|
n = len(pieces)
|
|
if n > 3:
|
|
reasons.append(f'{prefix}invalid geo code "{code}"')
|
|
elif n > 0 and pieces[0] not in geo_data:
|
|
reasons.append(f'{prefix}unknown continent code "{code}"')
|
|
elif n > 1 and pieces[1] not in geo_data[pieces[0]]:
|
|
reasons.append(f'{prefix}unknown country code "{code}"')
|
|
elif (
|
|
n > 2
|
|
and pieces[2] not in geo_data[pieces[0]][pieces[1]]['provinces']
|
|
):
|
|
reasons.append(f'{prefix}unknown province code "{code}"')
|
|
|
|
return reasons
|
|
|
|
@classmethod
|
|
def parse(cls, code):
|
|
pieces = code.split('-')
|
|
try:
|
|
country_code = pieces[1]
|
|
except IndexError:
|
|
country_code = None
|
|
try:
|
|
province_code = pieces[2]
|
|
except IndexError:
|
|
province_code = None
|
|
return {
|
|
'continent_code': pieces[0],
|
|
'country_code': country_code,
|
|
'province_code': province_code,
|
|
}
|
|
|
|
@classmethod
|
|
def country_to_code(cls, country):
|
|
for continent, countries in geo_data.items():
|
|
if country in countries:
|
|
return f'{continent}-{country}'
|
|
cls.log.warning('country_to_code: unrecognized country "%s"', country)
|
|
return
|
|
|
|
@classmethod
|
|
def province_to_code(cls, province):
|
|
# We cheat on this one a little since we only support provinces in
|
|
# NA-US, NA-CA
|
|
if (
|
|
province not in geo_data['NA']['US']['provinces']
|
|
and province not in geo_data['NA']['CA']['provinces']
|
|
):
|
|
cls.log.warning(
|
|
'country_to_code: unrecognized province "%s"', province
|
|
)
|
|
return
|
|
if province in geo_data['NA']['US']['provinces']:
|
|
country = 'US'
|
|
if province in geo_data['NA']['CA']['provinces']:
|
|
country = 'CA'
|
|
return f'NA-{country}-{province}'
|