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 getGeneratorById(self, generator_id):
dao = GeneratorDAO()
row = dao.getGeneratorById(generator_id)
if not row:
return jsonify(Error="Generator Not Found"), 404
else:
genearator = self.build_generator_dict(row)
return jsonify(Generator=genearator)
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 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)