warningstracker.py 410 B

12345678910111213
  1. class WarningsTracker(object):
  2. def __init__(self, warning_level):
  3. self.warning_level = warning_level
  4. def is_warning_active(self, warning):
  5. if self.warning_level > 0 and not warning.canceled:
  6. self.warning_level -= 1
  7. return True
  8. else:
  9. return False
  10. def is_warning_expired(self, warning):
  11. return not self.is_warning_active(warning)