def updateGenerator(self, generator_id, json):
generator_dao = GeneratorDAO()
if not generator_dao.getGeneratorById(generator_id):
return jsonify(Error="Generator not found."), 404
else:
supplier_id = json['supplier_id']
category_id = json['category_id']
generator_name = json['generator_name']
generator_brand = json['generator_brand']
generator_quantity = json['generator_quantity']
generator_price = json['generator_price']
power_capacity = json['power_capacity']
power_condition = json['power_condition']
generator_fuel = json['generator_fuel']
if supplier_id and category_id and generator_name and generator_brand and generator_quantity and (
generator_price >= 0
) and power_capacity and power_condition and generator_fuel:
resource_id = generator_dao.update(generator_id,
power_capacity,
power_condition,
generator_fuel)
res_dao = ResourceDAO()
res_dao.update(resource_id, supplier_id, category_id,
generator_name, generator_brand,
generator_quantity, generator_price)
result = self.build_generator_attributes(
supplier_id, resource_id, generator_id, category_id,
generator_name, generator_brand, generator_quantity,
generator_price, power_capacity, power_condition,
generator_fuel)
return jsonify(Generator=result), 200
else:
return jsonify(
Error="Unexpected attributes in update request"), 400
def insertGenerator(self, json):
supplier_id = json['supplier_id']
category_id = json['category_id']
generator_name = json['generator_name']
generator_brand = json['generator_brand']
generator_quantity = json['generator_quantity']
generator_price = json['generator_price']
power_capacity = json['power_capacity']
power_condition = json['power_condition']
generator_fuel = json['generator_fuel']
if supplier_id and category_id and generator_name and generator_brand and generator_quantity and (
generator_price >=
0) and power_capacity and power_condition and generator_fuel:
res_dao = ResourceDAO()
resource_id = res_dao.insert(supplier_id, category_id,
generator_name, generator_brand,
generator_quantity, generator_price)
generator_dao = GeneratorDAO()
generator_id = generator_dao.insert(resource_id, power_capacity,
power_condition,
generator_fuel)
result = self.build_generator_attributes(
supplier_id, resource_id, generator_id, category_id,
generator_name, generator_brand, generator_quantity,
generator_price, power_capacity, power_condition,
generator_fuel)
return jsonify(Generator=result), 201
else:
return jsonify(Error="Unexpected attributes in post request"), 400
def getGeneratorByResourceId(self, resource_id):
dao = GeneratorDAO()
row = dao.getGeneratorsByResourceId(resource_id)
if not row:
return jsonify(Error="Generator Not Found"), 404
else:
genearator = self.build_generator_dict(row)
return jsonify(Generator=genearator)
def getAllReservedGenerators(self):
dao = GeneratorDAO()
result = dao.getAllReservedGenerators()
result_list = []
for row in result:
result = self.build_generator_dict(row)
result_list.append(result)
return jsonify(Generators=result_list)
def deleteGenerator(self, generator_id):
generator_dao = GeneratorDAO()
if not generator_dao.getGeneratorById(generator_id):
return jsonify(Error="Generator not found."), 404
else:
resource_id = generator_dao.delete(generator_id)
res_dao = ResourceDAO()
res_dao.delete(resource_id)
return jsonify(DeleteStatus="OK"), 200
def getGeneratorsBySupplierId(self, supplier_id):
supplier_dao = SupplierDAO()
if not supplier_dao.getSupplierById(supplier_id):
return jsonify(Error="Supplier Not Found"), 404
else:
generator_dao = GeneratorDAO()
result_list = []
generator_list = generator_dao.getGeneratorsBySupplierId(
supplier_id)
for row in generator_list:
result = self.build_generator_dict(row)
result_list.append(result)
return jsonify(Generators=result_list)
def getGeneratorAddress(self, generator_id):
generator_dao = GeneratorDAO()
try:
supplier_id = generator_dao.getGeneratorById(generator_id)[5]
except Exception:
return jsonify(Error="Generator not found."), 404
supplier_dao = SupplierDAO()
if not supplier_dao.getSupplierById(supplier_id):
return jsonify(Error="Supplier not found."), 404
else:
row = generator_dao.getGeneratorAddress(supplier_id)
if not row:
return jsonify(Error="Address not found."), 404
else:
address = self.build_address_dic(row)
return jsonify(Address=address)
def searchGenerators(self, args):
generator_power_capacity = args.get('power_capacity')
generator_power_condition = args.get('power_condition')
generator_fuel = args.get('generator_fuel')
dao = GeneratorDAO()
generator_list = []
if (len(args) == 1) and generator_power_capacity:
generator_list = dao.getGeneratorsByPowerCapacity(
generator_power_capacity)
elif (len(args) == 1) and generator_power_condition:
generator_list = dao.getGeneratorsByPowerCondition(
generator_power_condition)
elif (len(args) == 1) and generator_fuel:
generator_list = dao.getGeneratorsByFuel(generator_fuel)
else:
return jsonify(Error="Malformed query string"), 400
result_list = []
for row in generator_list:
result = self.build_generator_dict(row)
result_list.append(result)
return jsonify(Generators=result_list)