def searchFuel(self, args): if len(args) > 4: return jsonify(Error = "Malformed search string."), 400 else: rid = args.get("rid") ftype = args.get("ftype") price = args.get("price") csize = args.get("csize") brand = args.get("brand") dao = FuelDAO() fuel_list = [] if (len(args) == 1) and rid: fuel_list = dao.getFuelById(rid) elif (len(args) == 1) and ftype: fuel_list = dao.getFuelByType(ftype) elif (len(args) == 1) and price: fuel_list = dao.getFuelByPrice(price) elif (len(args) == 1) and csize: fuel_list = dao.getFuelByContainerSize(csize) elif (len(args) == 1) and brand: fuel_list = dao.getFuelByBrand(brand) else: return jsonify(Error = "Malformed query string"), 400 result_list = [] for row in fuel_list: result = self.build_fuel_dict(row) result_list.append(result) return jsonify(Fuel = result_list)
def getFuelByBrand(self, brand): dao = FuelDAO() fuel_list = dao.getFuelByBrand(brand) if not fuel_list: return jsonify(Error = "No Fuel found"), 404 else: result_list = [] for row in fuel_list: result = self.build_fuel_dict(row) result_list.append(result) return jsonify(Fuel = result_list)