aboutsummaryrefslogtreecommitdiff
path: root/discord/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'discord/utils.py')
-rw-r--r--discord/utils.py13
1 files changed, 9 insertions, 4 deletions
diff --git a/discord/utils.py b/discord/utils.py
index 8090d4a7..a3162a8a 100644
--- a/discord/utils.py
+++ b/discord/utils.py
@@ -302,10 +302,15 @@ def _bytes_to_base64_data(data):
def to_json(obj):
return json.dumps(obj, separators=(',', ':'), ensure_ascii=True)
-def _parse_ratelimit_header(request):
- now = parsedate_to_datetime(request.headers['Date'])
- reset = datetime.datetime.fromtimestamp(float(request.headers['X-Ratelimit-Reset']), datetime.timezone.utc)
- return (reset - now).total_seconds()
+def _parse_ratelimit_header(request, *, use_clock=False):
+ reset_after = request.headers.get('X-Ratelimit-Reset-After')
+ if use_clock or not reset_after:
+ utc = datetime.timezone.utc
+ now = datetime.datetime.now(utc)
+ reset = datetime.datetime.fromtimestamp(float(request.headers['X-Ratelimit-Reset']), utc)
+ return (reset - now).total_seconds()
+ else:
+ return float(reset_after)
async def maybe_coroutine(f, *args, **kwargs):
value = f(*args, **kwargs)