30 lines
893 B
Python
30 lines
893 B
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 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)
|