def getEquipByCondition(self, condition):
dao = HeavyEquipDAO()
equip_list = dao.getEquipByCondition(condition)
if not equip_list:
return jsonify(Error="No equipment found"), 404
else:
result_list = []
for row in equip_list:
result = self.build_equip_dict(row)
result_list.append(result)
return jsonify(HeavyEquip=result_list)
def searchEquip(self, args):
if len(args) > 5:
return jsonify(Error = "Malformed search string."), 400
else:
rid = args.get("rid")
price = args.get("price")
make = args.get("make")
condition = args.get("condition")
equipfunction = args.get("function")
dao = HeavyEquipDAO()
equip_list = []
if (len(args) == 4) and price and make and condition and equipfunction:
equip_list = dao.getEquipByPriceMakeConditionAndFunction(price, make, condition, equipfunction)
elif (len(args) == 3) and price and make and condition:
equip_list = dao.getEquipByPriceMakeAndCondition(price, make, condition)
elif (len(args) == 3) and price and make and equipfunction:
equip_list = dao.getEquipByPriceMakeAndFunction(price, make, equipfunction)
elif (len(args) == 3) and price and equipfunction and condition:
equip_list = dao.getEquipByPriceFunctionAndCondition(price, equipfunction, condition)
elif (len(args) == 3) and equipfunction and make and condition:
equip_list = dao.getEquipByFunctionMakeAndCondition(equipfunction, make, condition)
elif (len(args) == 2) and price and make:
equip_list = dao.getEquipByPriceAndMake(price, make)
elif (len(args) == 2) and condition and make:
equip_list = dao.getEquipByConditionAndMake(condition, make)
elif (len(args) == 2) and price and condition:
equip_list = dao.getEquipByPriceAndCondition(price, condition)
elif (len(args) == 2) and price and equipfunction:
equip_list = dao.getEquipByPriceAndFunction(price, equipfunction)
elif (len(args) == 2) and equipfunction and condition:
equip_list = dao.getEquipByFunctionAndCondition(equipfunction, condition)
elif (len(args) == 2) and make and equipfunction:
equip_list = dao.getEquipByMakeAndFunction(make, equipfunction)
elif (len(args) == 1) and rid:
equip_list = dao.getEquipById(rid)
elif (len(args) == 1) and price:
equip_list = dao.getEquipByPrice(price)
elif (len(args) == 1) and make:
equip_list = dao.getEquipByMake(make)
elif (len(args) == 1) and condition:
equip_list = dao.getEquipByCondition(condition)
elif (len(args) == 1) and equipfunction:
equip_list = dao.getEquipByFunction(equipfunction)
else:
return jsonify(Error = "Malformed query string"), 400
result_list = []
for row in equip_list:
result = self.build_equip_dict(row)
result_list.append(result)
return jsonify(HeavyEquip=result_list)