Cloned DMOJ

dmoj/throttle_mail.py

@@ -0,0 +1,29 @@
+import traceback
+
+from django.conf import settings
+from django.core.cache import cache
+from django.utils.log import AdminEmailHandler
+
+DEFAULT_THROTTLE = (10, 60)
+
+
+def new_email():
+    cache.add('error_email_throttle', 0, settings.DMOJ_EMAIL_THROTTLING[1])
+    return cache.incr('error_email_throttle')
+
+
+class ThrottledEmailHandler(AdminEmailHandler):
+    def __init__(self, *args, **kwargs):
+        super(ThrottledEmailHandler, self).__init__(*args, **kwargs)
+
+        self.throttle = settings.DMOJ_EMAIL_THROTTLING[0]
+
+    def emit(self, record):
+        try:
+            count = new_email()
+        except Exception:
+            traceback.print_exc()
+        else:
+            if count >= self.throttle:
+                return
+        AdminEmailHandler.emit(self, record)