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
