29 lines
778 B
Python
29 lines
778 B
Python
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)
|