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)