41 lines
962 B
Python
41 lines
962 B
Python
import json
|
|
import re
|
|
|
|
from rest_framework.throttling import UserRateThrottle
|
|
|
|
|
|
class FailSafeJSONEncoder(json.JSONEncoder):
|
|
def default(self, obj):
|
|
try:
|
|
return super(FailSafeJSONEncoder, self).default(obj)
|
|
except Exception:
|
|
return str(obj)
|
|
|
|
|
|
def safe_json_dumps(data, **kwargs):
|
|
return json.dumps(data, cls=FailSafeJSONEncoder, **kwargs)
|
|
|
|
|
|
class HourUserRateThrottle(UserRateThrottle):
|
|
scope = "hour-throttle"
|
|
|
|
|
|
class DayUserRateThrottle(UserRateThrottle):
|
|
scope = "day-throttle"
|
|
|
|
|
|
def find_first(iterable, pred=None, default=None):
|
|
return next(filter(pred, iterable), default)
|
|
|
|
|
|
def find_first_index(iterable, pred, default=None):
|
|
return next((i for i, x in enumerate(iterable) if pred(x)), default)
|
|
|
|
|
|
def replace_whitespace(text, replacement=" "):
|
|
return re.sub(r"\s+", replacement, text).strip()
|
|
|
|
|
|
def get_django_content_type(obj):
|
|
return obj._meta.app_label + "." + type(obj).__name__
|