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)