Browse Source

Add config option for minimum number of pictures

updated_weboob
Maël Le Garrec$ 5 years ago
parent
commit
e7c81d31ac
  1. 5
      flatisfy/config.py
  2. 6
      flatisfy/filters/__init__.py

5
flatisfy/config.py

@ -34,6 +34,7 @@ DEFAULT_CONFIG = { @@ -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): @@ -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"]

6
flatisfy/filters/__init__.py

@ -67,6 +67,12 @@ def refine_with_housing_criteria(flats_list, constraint): @@ -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

Loading…
Cancel
Save