Back to Freecodecamp

Step 33

curriculum/challenges/english/blocks/workshop-salary-tracker/68caaf1e590806f0b87f5922.md

latest3.4 KB
Original Source

--description--

At the beginning of your setter, create an if statement that raises a TypeError with the message 'salary' must be a number. when new_salary is not either an integer or a float.

--hints--

You should have an if statement in your salary setter.

js
({ test: () => assert(runPython(`_Node(_code).find_class("Employee").find_functions("salary")[1].find_ifs()[0]`)) })

You should raise a TypeError with the message 'salary' must be a number. when new_salary is not either an integer or a float.

js
({ test: () => runPython(`
  emp = Employee('Frank', 'trainee')
  built_in_print = print
  print = lambda x: x
  try:
    emp.salary = "100"
  except TypeError as e:
    print = built_in_print
    assert str(e) == "'salary' must be a number."
  else:
    assert False, "Expected to raise TypeError with invalid new_salary"
`) })

You should not raise any exception when new_salary is either an integer or a float.

js
({ test: () => runPython(`
  emp = Employee('Frank', 'trainee')
  built_in_print = print
  print = lambda x: x
  try:
    emp.salary = 100
  except Exception:
    assert False, "Expected not to raise exception with valid new_salary"
  try:
    emp.salary = 100.00
  except Exception:
    assert False, "Expected not to raise exception with valid new_salary"
`) })

--seed--

--seed-contents--

py
class Employee:
    _base_salaries = {
        'trainee': 1000,
        'junior': 2000,
        'mid-level': 3000,
        'senior': 4000,
    }

    def __init__(self, name, level):
        if not (isinstance(name, str) and isinstance(level, str)):
            raise TypeError("'name' and 'level' attribute must be of type 'str'.")
        if level not in Employee._base_salaries:
            raise ValueError(f"Invalid value '{level}' for 'level' attribute.")
        self._name = name
        self._level = level
        self._salary = Employee._base_salaries[level]

    def __str__(self):
        return f'{self.name}: {self.level}'

    def __repr__(self):
        return f"Employee('{self.name}', '{self.level}')"

    @property
    def name(self):
        return self._name

    @name.setter
    def name(self, new_name):
        if not isinstance(new_name, str):
            raise TypeError("'name' must be a string.")
        self._name = new_name
        print(f"'name' updated to '{self.name}'.")

    @property
    def level(self):
        return self._level

    @level.setter
    def level(self, new_level):
        if new_level not in Employee._base_salaries:
            raise ValueError(f"Invalid value '{new_level}' for 'level' attribute.")
        if new_level == self.level:
            raise ValueError(f"'{self.level}' is already the selected level.")
        if Employee._base_salaries[new_level] < Employee._base_salaries[self.level]:
            raise ValueError(f"Cannot change to lower level.")
        print(f"'{self.name}' promoted to '{new_level}'.")
        self._salary = Employee._base_salaries[new_level]
        self._level = new_level

    @property
    def salary(self):
        return self._salary
--fcc-editable-region--
    @salary.setter
    def salary(self, new_salary):
        
        self._salary = new_salary
        print(f'Salary updated to ${self.salary}.')
--fcc-editable-region--

charlie_brown = Employee('Charlie Brown', 'trainee')
print(charlie_brown)
print(f'Base salary: ${charlie_brown.salary}')

charlie_brown.level = 'junior'