Files
libpostal/scripts/geodata/addresses/directions.py

38 lines
1.0 KiB
Python

from geodata.addresses.config import address_config
from geodata.addresses.numbering import NumericPhrase
from geodata.math.sampling import weighted_choice
class RelativeDirection(NumericPhrase):
key = 'directions'
dictionaries = ['unit_directions']
class AnteroposteriorDirection(RelativeDirection):
key = 'directions.anteroposterior'
class LateralDirection(RelativeDirection):
key = 'directions.lateral'
class CardinalDirection(NumericPhrase):
key = 'cardinal_directions'
dictionaries = ['cardinal_directions']
class Direction(object):
CARDINAL = 'cardinal'
RELATIVE = 'relative'
@classmethod
def random(cls, language, country=None, cardinal_proability=0.5):
values = [cls.CARDINAL, cls.RELATIVE]
probs_cdf = [cardinal_proability, 1.0]
choice = weighted_choice(values, probs_cdf)
if choice == cls.CARDINAL:
return CardinalDirection.phrase(None, language, country=country)
else:
return RelativeDirection.phrase(None, language, country=country)