prefiltering of items

This commit is contained in:
Adrian Manteza
2022-03-23 16:32:33 +00:00
parent 09d3fce02d
commit ad3019e478
3 changed files with 9 additions and 10 deletions

View File

@ -5,7 +5,7 @@ from tokenize import String
from models.item import Item
def items_csv_to_dict(items_csv_reader, solver_run):
def csv_to_item(items_csv_reader, solver_run):
items = []
headers = []
@ -46,9 +46,8 @@ def items_csv_to_dict(items_csv_reader, solver_run):
valid_item = True
item = Item.parse_obj(item)
for constraint in solver_run.constraints:
attribute = item.get_attribute(constraint.reference_attribute)
if attribute and constraint.minimum == 100 and int(attribute.value) == 0:
valid_item = False
if item.attribute_exists(constraint.reference_attribute) == False and constraint.minimum == 100:
valid_item = False
if valid_item: items.append(item)