def delete(self, name): if not EmployeeModel.find_by_name(name): return {'message': f'Employee {name} does not exists'} employee = EmployeeModel.find_by_name(name) employee.delete_from_db() return {'message': f'Employee {name} deleted'}
def put(self, name): data = Employee.parser.parse_args() employee = EmployeeModel.find_by_name(name) if employee: employee.salary = data['salary'] else: employee = EmployeeModel(name, **data) employee.save_to_db() return employee.json()
def post(self, name): if EmployeeModel.find_by_name(name): return {'message': "An employee with name {} already exists".format(name)}, 400 data = Employee.parser.parse_args() employee = EmployeeModel(name, **data) try: employee.save_to_db() except: return {"message":"An error occure"},500 return employee.json(),201
def put(self, name): data = Employee.parser.parse_args() employee = EmployeeModel.find_by_name(name) if not employee: employee = EmployeeModel(name, **data) else: employee.role = data['role'] employee.email = data['email'] employee.store_id = data['store_id'] employee.save_to_db() return employee.json()
def post(self, name): if EmployeeModel.find_by_name(name): return {'message': f'Employee {name} already exists'}, 400 data = Employee.parser.parse_args() employee = EmployeeModel(name, **data) try: employee.save_to_db() except: return {'message': 'Error while inserting to DB'}, 500 return employee.json(), 201
def get(self, name): employee = EmployeeModel.find_by_name(name) if not employee: return {'message': f'Employee {name} not found'}, 404 return employee.json()
def delete(self, name): employee=EmployeeModel.find_by_name(name) if employee: employee.delete_from_db() return {'message': 'Employee deleted'} return {'message': 'Employee not found.'}, 404
def get(self, name): employee=EmployeeModel.find_by_name(name) if employee: return employee.json() return {'message': 'Employee not found'}, 404