NDOJ/dmoj/throttle_mail.py

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)