def getEquipById(self, rid):
dao = HeavyEquipDAO()
row = dao.getEquipById(rid)
if not row:
return jsonify(Error="Equipment Not Found"), 404
else:
equip = self.build_equip_dict(row)
return jsonify(HeavyEquip=equip)
def getAllEquip(self):
dao = HeavyEquipDAO()
equip_list = dao.getAllEquip()
result_list = []
for row in equip_list:
result = self.build_equip_dict(row)
result_list.append(result)
return jsonify(HeavyEquip=result_list)
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 getEquipByMake(self, make):
dao = HeavyEquipDAO()
make_list = dao.getEquipByMake(make)
if not make_list:
return jsonify(Error = "No Make found"), 404
else:
result_list = []
for row in make_list:
result = self.build_equip_dict(row)
result_list.append(result)
return jsonify(HeavyEquip=result_list)
def getEquipSuppliersByRegion(self, region):
dao = HeavyEquipDAO()
suppliers_list = dao.getEquipSuppliersByRegion(region)
if not suppliers_list:
return jsonify(Error = "No Suppliers found"), 404
else:
result_list = []
for row in suppliers_list:
result = self.build_supplierequip_dict(row)
result_list.append(result)
return jsonify(Suppliers = result_list)
def getEquipByPrice(self, price):
dao = HeavyEquipDAO()
price_list = dao.getEquipByPrice(price)
if not price_list:
return jsonify(Error = "No Equipment found"), 404
else:
result_list = []
for row in price_list:
result = self.build_equip_dict(row)
result_list.append(result)
return jsonify(HeavyEquip=result_list)
def insertEquip(self, form):
if len(form) != 6:
return jsonify(Error="Malformed POST request"), 400
else:
sid = form["sid"]
quantity = form["quantity"]
price = form['price']
make = form['make']
condition = form['condition']
equipfunction = form["function"]
if sid and quantity and price and make and condition and equipfunction:
dao = HeavyEquipDAO()
rid = ResourcesDAO().insert(sid, quantity)
rid = dao.insert(rid, price, make, condition, equipfunction)
result = self.build_equip_attributes(rid, price, make, condition, equipfunction)
return jsonify(HeavyEquip=result), 201
else:
return jsonify(Error="Unexpected attributes in POST request"), 400
def searchEquipRequests(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.getEquipRequestsByPriceMakeConditionAndFunction(price, make, condition, equipfunction)
elif (len(args) == 3) and price and make and condition:
equip_list = dao.getEquipRequestsByPriceMakeAndCondition(price, make, condition)
elif (len(args) == 3) and price and make and equipfunction:
equip_list = dao.getEquipRequestsByPriceMakeAndFunction(price, make, equipfunction)
elif (len(args) == 3) and price and equipfunction and condition:
equip_list = dao.getEquipRequestsByPriceFunctionAndCondition(price, equipfunction, condition)
elif (len(args) == 3) and equipfunction and make and condition:
equip_list = dao.getEquipRequestsByFunctionMakeAndCondition(equipfunction, make, condition)
elif (len(args) == 2) and price and make:
equip_list = dao.getEquipRequestsByPriceAndMake(price, make)
elif (len(args) == 2) and condition and make:
equip_list = dao.getEquipRequestsByConditionAndMake(condition, make)
elif (len(args) == 2) and price and condition:
equip_list = dao.getEquipRequestsByPriceAndCondition(price, condition)
elif (len(args) == 2) and price and equipfunction:
equip_list = dao.getEquipRequestsByPriceAndFunction(price, equipfunction)
elif (len(args) == 2) and equipfunction and condition:
equip_list = dao.getEquipRequestsByFunctionAndCondition(equipfunction, condition)
elif (len(args) == 2) and make and equipfunction:
equip_list = dao.getEquipRequestsByMakeAndFunction(make, equipfunction)
elif (len(args) == 1) and rid:
equip_list = dao.getEquipRequestsById(rid)
elif (len(args) == 1) and price:
equip_list = dao.getEquipRequestsByPrice(price)
elif (len(args) == 1) and make:
equip_list = dao.getEquipRequestsByMake(make)
elif (len(args) == 1) and condition:
equip_list = dao.getEquipRequestsByCondition(condition)
elif (len(args) == 1) and equipfunction:
equip_list = dao.getEquipRequestsByFunction(equipfunction)
else:
return jsonify(Error = "Malformed query string"), 400
result_list = []
for row in equip_list:
result = self.build_requestequip_dict(row)
result_list.append(result)
return jsonify(HeavyEquip=result_list)