def get(self, us_employer_id): company = CompanyModel.find_by_us_employer_id(us_employer_id) if not company: return { "message": "No such company registered with this employer ID!" }, 404 # returns a tuple: (body, status code), (body:dictionary, status code: integer - default 200) else: return format_company_to_json(company)
def delete(self, us_employer_id): company = CompanyModel.find_by_us_employer_id(us_employer_id) if not company: return { "message": "No such company associated with this employer ID!" }, 404 company.delete_from_db() return {"message": "company deleted!"}
def put(self, us_employer_id): company = CompanyModel.find_by_us_employer_id(us_employer_id) if not company: return { "message": "No such company registered with this employer ID !" }, 404 data = Company.parser.parse_args() company.company_name = data['company_name'] company.business_area = data['business_area'] company.number_of_employees = data['number_of_employees'] company.phone = data['phone'] company.address = data['address'] try: company.save_to_db() except: return { "message": "An error occurred while updating the company info!" }, 500 return format_company_to_json(company)
def post(self, us_employer_id): if CompanyModel.find_by_us_employer_id(us_employer_id): return { "message": "A company already exists with the same employer ID, use a different one!" }, 400 data = Company.parser.parse_args() is_valid, error_message = CompanyModel.check_if_data_has_valid_format( us_employer_id, **data) if is_valid: new_company = CompanyModel(us_employer_id, **data) try: format_company_to_json(new_company) new_company.save_to_db() return format_company_to_json(new_company), 201 except: return { "message": "An error occurred while creating the user!" }, 500 else: return {"message": error_message}, 400