def deleteCompany(self, compid):
dao = CompanyDAO()
if not dao.getCompanyById(compid):
return jsonify(Error="Company not found."), 404
else:
dao.delete(compid)
return jsonify(DeleteStatus="OK"), 200
def getCompanyById(self, compid):
dao = CompanyDAO()
row = dao.getCompanyById(compid)
if not row:
return jsonify(Error="Company Not Found"), 404
else:
company = self.build_company_dict(row)
return jsonify(Company=company)
def getResourcesByCompanyId(self, compid):
dao = CompanyDAO()
if not dao.getCompanyById(compid):
return jsonify(Error="Company Not Found"), 404
resources_list = dao.getResourcesByCompanyId(compid)
result_list = []
for row in resources_list:
result = self.build_resource_dict(row)
result_list.append(result)
return jsonify(Company=result_list)
def getConsumerByCompanyId(self, compid):
dao = CompanyDAO()
if not dao.getCompanyById(compid):
return jsonify(Error="Company Not Found"), 404
consumers_list = dao.getConsumerByCompanyId(compid)
result_list = []
for row in consumers_list:
result = self.build_consumer_dict(row)
result_list.append(result)
return jsonify(Company=result_list)
def getSuppliersByCompanyId(self, company_id):
company_dao = CompanyDAO()
if not company_dao.getCompanyById(company_id):
return jsonify(Error = "Company Not Found"), 404
else:
supplier_dao = SupplierDAO()
result_list = []
supplier_list = supplier_dao.getSuppliersByCompanyId(company_id)
for row in supplier_list:
result = self.build_supplier_dict(row)
result_list.append(result)
return jsonify(Suppliers = result_list)
def updateCompany(self, compid, form):
dao = CompanyDAO()
if not dao.getCompanyById(compid):
return jsonify(Error="Company not found."), 404
else:
if len(form) != 1:
return jsonify(Error="Malformed update request"), 400
else:
compname = form['compname']
if compname:
dao.update(compid, compname)
result = self.build_company_attributes(compid, compname)
return jsonify(Company=result), 200
else:
return jsonify(
Error="Unexpected attributes in update request"), 400
def updateCompany(self, company_id, json):
dao = CompanyDAO()
if not dao.getCompanyById(company_id):
return jsonify(Error="Company not found."), 404
else:
company_id = json['company_id']
company_name = json['company_name']
company_address = json['company_address']
company_phone = json['company_phone']
if company_id and company_name and company_address and company_phone:
dao.update(company_id, company_name, company_address,
company_phone)
result = self.build_company_attributes(company_id,
company_name,
company_address,
company_phone)
return jsonify(Company=result), 200
else:
return jsonify(
Error="Unexpected attributes in update request"), 400