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