def random_employee(employee_type):
if employee_type == 'developer':
return Developer(**Developer.random_attrs())
if employee_type == 'sales':
return Sales(**Sales.random_attrs())
if employee_type == 'project_manager':
return ProjectManager(**ProjectManager.random_attrs())
if employee_type == 'support':
return Support(**Support.random_attrs())
err_msg = (
'{} employee type is not supported.\n'.format(type),
'Allowed values are: \'developer\', \'project_manager\', \'sales\' or \'support\'',
)
raise ValueError(err_msg)
