style: minor code style improvements in core modules
This commit is contained in:
@@ -2,8 +2,9 @@
|
|||||||
|
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from collections.abc import Callable
|
||||||
from dataclasses import dataclass, field
|
from dataclasses import dataclass, field
|
||||||
from typing import TYPE_CHECKING, Any, Callable
|
from typing import TYPE_CHECKING, Any
|
||||||
|
|
||||||
from fastapi_traffic.core.algorithms import Algorithm
|
from fastapi_traffic.core.algorithms import Algorithm
|
||||||
|
|
||||||
|
|||||||
@@ -3,19 +3,26 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import functools
|
import functools
|
||||||
from typing import TYPE_CHECKING, Any, Callable, TypeVar, overload
|
from collections.abc import Callable
|
||||||
|
from typing import TYPE_CHECKING, Any, TypeVar, overload
|
||||||
|
|
||||||
from fastapi_traffic.core.algorithms import Algorithm
|
from fastapi_traffic.core.algorithms import Algorithm
|
||||||
from fastapi_traffic.core.config import KeyExtractor, RateLimitConfig, default_key_extractor
|
from fastapi_traffic.core.config import (
|
||||||
|
KeyExtractor,
|
||||||
|
RateLimitConfig,
|
||||||
|
default_key_extractor,
|
||||||
|
)
|
||||||
from fastapi_traffic.core.limiter import get_limiter
|
from fastapi_traffic.core.limiter import get_limiter
|
||||||
from fastapi_traffic.exceptions import RateLimitExceeded
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from starlette.requests import Request
|
from starlette.requests import Request
|
||||||
from starlette.responses import Response
|
from starlette.responses import Response
|
||||||
|
|
||||||
|
from fastapi_traffic.exceptions import RateLimitExceeded
|
||||||
|
|
||||||
F = TypeVar("F", bound=Callable[..., Any])
|
F = TypeVar("F", bound=Callable[..., Any])
|
||||||
|
|
||||||
|
# Note: Config loader from secrets .env
|
||||||
|
|
||||||
@overload
|
@overload
|
||||||
def rate_limit(
|
def rate_limit(
|
||||||
|
|||||||
@@ -3,18 +3,24 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
from typing import TYPE_CHECKING, Awaitable, Callable
|
from typing import TYPE_CHECKING
|
||||||
|
|
||||||
from starlette.middleware.base import BaseHTTPMiddleware
|
from starlette.middleware.base import BaseHTTPMiddleware
|
||||||
from starlette.responses import JSONResponse
|
from starlette.responses import JSONResponse
|
||||||
|
|
||||||
from fastapi_traffic.backends.memory import MemoryBackend
|
from fastapi_traffic.backends.memory import MemoryBackend
|
||||||
from fastapi_traffic.core.algorithms import Algorithm
|
from fastapi_traffic.core.algorithms import Algorithm
|
||||||
from fastapi_traffic.core.config import GlobalConfig, RateLimitConfig, default_key_extractor
|
from fastapi_traffic.core.config import (
|
||||||
|
GlobalConfig,
|
||||||
|
RateLimitConfig,
|
||||||
|
default_key_extractor,
|
||||||
|
)
|
||||||
from fastapi_traffic.core.limiter import RateLimiter
|
from fastapi_traffic.core.limiter import RateLimiter
|
||||||
from fastapi_traffic.exceptions import RateLimitExceeded
|
from fastapi_traffic.exceptions import RateLimitExceeded
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
|
from collections.abc import Awaitable, Callable
|
||||||
|
|
||||||
from starlette.requests import Request
|
from starlette.requests import Request
|
||||||
from starlette.responses import Response
|
from starlette.responses import Response
|
||||||
from starlette.types import ASGIApp
|
from starlette.types import ASGIApp
|
||||||
|
|||||||
Reference in New Issue
Block a user