1
- from datetime import timedelta , datetime , timezone
2
1
from types import TracebackType
3
2
from contextvars import ContextVar
3
+ from datetime import datetime , timezone , timedelta
4
4
from contextlib import contextmanager , asynccontextmanager
5
5
from typing import (
6
6
Any ,
23
23
from .response import Response
24
24
from .utils import obj_to_jsonable
25
25
from .config import Config , get_config
26
- from .exception import RequestError , RequestFailed , RequestTimeout , RateLimitExceededError
27
26
from .auth import BaseAuthStrategy , TokenAuthStrategy , UnauthAuthStrategy
27
+ from .exception import (
28
+ RequestError ,
29
+ RequestFailed ,
30
+ RequestTimeout ,
31
+ RateLimitExceededError ,
32
+ )
28
33
from .typing import (
29
34
URLTypes ,
30
35
CookieTypes ,
@@ -319,7 +324,10 @@ def _check(
319
324
status_code = str (response .status_code )
320
325
321
326
if response .status_code == 403 and "retry-after" in response .headers :
322
- raise RateLimitExceededError (timedelta (seconds = int (response .headers ["retry-after" ])), original_message = response .content )
327
+ raise RateLimitExceededError (
328
+ timedelta (seconds = int (response .headers ["retry-after" ])),
329
+ original_message = response .content ,
330
+ )
323
331
324
332
if (
325
333
response .status_code in [403 , 429 ]
@@ -330,7 +338,9 @@ def _check(
330
338
int (response .headers ["x-ratelimit-reset" ]), tz = timezone .utc
331
339
) - datetime .now (tz = timezone .utc )
332
340
retry_after = max (retry_after , timedelta ())
333
- raise RateLimitExceededError (retry_after , original_message = response .content )
341
+ raise RateLimitExceededError (
342
+ retry_after , original_message = response .content
343
+ )
334
344
335
345
error_model = error_models .get (
336
346
status_code ,
0 commit comments