tommy/libpostal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[addresses] Adding ability to determine unit numbers using a known floor number

Browse Source
This commit is contained in:
Al
2016-06-29 15:56:45 -04:00
parent 5b17a3a3ce
commit 171a2c9b2f
3 changed files with 30 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/geodata/addresses/units.py
View File

@@ -51,17 +51,18 @@ class Unit(NumberedComponent):
return six.u('{}{}').format(floor_number, safe_decode(unit).zfill(num_digits))
@classmethod
def random(cls, language, country=None, num_floors=None, num_basements=None):
def random(cls, language, country=None, num_floors=None, num_basements=None, floor=None):
num_type, num_type_props = cls.choose_alphanumeric_type('units.alphanumeric', language, country=country)
if num_type is None:
return None
use_floor_prob = address_config.get_property('units.alphanumeric.use_floor_probability', language, country=country, default=0.0)
if num_floors is None or random.random() >= use_floor_prob:
if (num_floors is None and floor is None) or random.random() >= use_floor_prob:
number = weighted_choice(cls.numbered_units, cls.unit_probs_cdf)
else:
floor = Floor.random(language, country=country, num_floors=num_floors, num_basements=num_basements)
if floor is None:
floor = Floor.random_int(language, country=country, num_floors=num_floors, num_basements=num_basements)
floor_numbering_starts_at = address_config.get_property('levels.numbering_starts_at', language, country=country, default=0)
ground_floor_starts_at = address_config.get_property('units.alphanumeric.use_floor_ground_starts_at', language, country=country, default=None)