Add config option for minimum number of pictures

This commit is contained in:
Maël Le Garrec$ 2017-10-24 17:21:48 +02:00
parent c96b2a3f4c
commit e7c81d31ac
2 changed files with 11 additions and 0 deletions

View File

@ -34,6 +34,7 @@ DEFAULT_CONFIG = {
"cost": (None, None), # (min, max) in currency unit
"rooms": (None, None), # (min, max)
"bedrooms": (None, None), # (min, max)
"minimum_pictures": None,
"time_to": {} # Dict mapping names to {"gps": [lat, lng],
# "time": (min, max) }
# Time is in seconds
@ -136,6 +137,10 @@ def validate_config(config, check_with_data):
assert "type" in constraint
assert isinstance(constraint["type"], str)
assert constraint["type"].upper() in ["RENT", "SALE", "SHARING"]
assert "minimum_pictures" in constraint
assert isinstance(constraint["minimum_pictures"], int)
assert constraint["minimum_pictures"] >= 0
assert "house_types" in constraint
assert constraint["house_types"]

View File

@ -67,6 +67,12 @@ def refine_with_housing_criteria(flats_list, constraint):
field.capitalize(), flat["id"])
is_ok[i] = is_ok[i] and is_within_interval
# Check number of pictures
print(constraint['minimum_pictures'], len(flat['photos']))
if not (constraint['minimum_pictures'] <= len(flat['photos'])):
is_ok[i] = False
return (
[
flat