def add_new_employee(self, data): """ :param data: :return: """ title = data['title'] age = data['age'] employee = Employee(**data) if title == 'CEO': employee.associates = map( lambda x: x.name, filter(lambda x: x.title.lower() == 'senior partner', self.employees)) employee.compensation_package = [ 'free money', 'cafeteria', '+1 free day' ] car = 'Bentley Coupe' if age > 50: car = 'Rolls Royce' employee.car = car employee.salary = self.base_salary * 5 * max( [len(self.employees), 1]) elif title == 'Senior partner': employee.salary = self.base_salary * 5 * max( [len(self.employees), 1]) * 0.80 employee.compensation_package = ['cafeteria', '+1 free day'] car = 'VW Passat' if employee.salary < 20: car = 'VW Jetta' employee.car = car employee.associates = map( lambda x: x.name, filter(lambda x: x.title.lower() == 'junior partner', self.employees)) elif title == 'Junior partner': employee.salary = self.base_salary * 3 employee.compensation_package = ['+1 free day'] employee.car = 'VW Jetta' employee.associates = map( lambda x: x.name, filter(lambda x: x.title.lower() == 'minion', self.employees)) else: employee.salary = self.base_salary self.employees.append(employee)
def add_new_employee(self, data): """ :param data: :return: """ title = data['title'] age = data['age'] employee = Employee(**data) associates = [] compensation_package = None car = None if title == 'CEO': associates, car, salary, compensation_package = self.ceo(age) elif title == 'Senior partner': associates, car, salary, compensation_package = self.senior() elif title == 'Junior partner': associates, car, salary, compensation_package = self.junior() else: salary = self.base_salary employee.salary = salary employee.car = car employee.compensation_package = compensation_package employee.associates = associates self.employees.append(employee)