Просмотр исходного кода

Small util for finding out if warnings are active or expired. #37

Rafał Pitoń 11 лет назад
Родитель
Сommit
b9bb2bc16f
1 измененных файлов с 13 добавлено и 0 удалено
  1. 13 0
      misago/apps/profiles/warnings/warningstracker.py

+ 13 - 0
misago/apps/profiles/warnings/warningstracker.py

@@ -0,0 +1,13 @@
+class WarningsTracker(object):
+    def __init__(self, warning_level):
+        self.warning_level = warning_level
+
+    def is_warning_active(self, warning):
+        if self.warning_level > 0 and not warning.canceled:
+            self.warning_level -= 1
+            return True
+        else:
+            return False
+
+    def is_warning_expired(self, warning):
+        return not self.is_warning_active(warning)