From d59ab827016c6a4a3cd64eb712da4044e6584bad Mon Sep 17 00:00:00 2001 From: Al Date: Sat, 6 Aug 2016 19:33:21 -0400 Subject: [PATCH] [metro stations] Adding metro station phrase generator --- scripts/geodata/addresses/metro_stations.py | 24 +++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 scripts/geodata/addresses/metro_stations.py diff --git a/scripts/geodata/addresses/metro_stations.py b/scripts/geodata/addresses/metro_stations.py new file mode 100644 index 00000000..c0d4c0a4 --- /dev/null +++ b/scripts/geodata/addresses/metro_stations.py @@ -0,0 +1,24 @@ +from geodata.addresses.config import address_config + +import random + +from geodata.addresses.config import address_config +from geodata.addresses.numbering import NumericPhrase +from geodata.encoding import safe_decode + + +class MetroStationPhrase(NumericPhrase): + key = 'metro_stations.alphanumeric' + dictionaries = ['qualifiers'] + + +class MetroStation(object): + @classmethod + def phrase(cls, station, language, country=None): + if station is None: + return None + phrase_prob = address_config.get_property('metro_stations.alphanumeric_phrase_probability', language, country=country, default=0.0) + if random.random() < phrase_prob: + return MetroStationPhrase.phrase(station, language, country=country) + + return None