zanewalker/fastapi-traffic
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: apply ruff formatting

Browse Source
This commit is contained in:
Zane Walker
2026-02-04 01:28:45 +00:00
parent 2900fca30a
commit 34e07f6b7e
4 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
fastapi_traffic/core/config_loader.py
View File

@@ -49,12 +49,14 @@ _GLOBAL_FIELD_TYPES: dict[str, type[Any]] = {
} }
# Fields that cannot be loaded from config files (callables, complex objects) # Fields that cannot be loaded from config files (callables, complex objects)
_NON_LOADABLE_FIELDS: frozenset[str] = frozenset({ _NON_LOADABLE_FIELDS: frozenset[str] = frozenset(
{
"key_extractor", "key_extractor",
"exempt_when", "exempt_when",
"on_blocked", "on_blocked",
"backend", "backend",
}) }
)
class ConfigLoader: class ConfigLoader:
@@ -200,7 +202,11 @@ class ConfigLoader:
value = value.strip() value = value.strip()
# Remove surrounding quotes if present # Remove surrounding quotes if present
if len(value) >= 2 and value[0] == value[-1] and value[0] in ('"', "'"): if (
len(value) >= 2
and value[0] == value[-1]
and value[0] in ('"', "'")
):
value = value[1:-1] value = value[1:-1]
env_vars[key] = value env_vars[key] = value

1
fastapi_traffic/core/decorator.py
View File

@@ -24,6 +24,7 @@ F = TypeVar("F", bound=Callable[..., Any])
# Note: Config loader from secrets .env # Note: Config loader from secrets .env
@overload @overload
def rate_limit( def rate_limit(
limit: int, limit: int,

1
tests/conftest.py
View File

@@ -24,6 +24,7 @@ from fastapi_traffic.middleware import RateLimitMiddleware
if TYPE_CHECKING: if TYPE_CHECKING:
from collections.abc import AsyncGenerator, Generator from collections.abc import AsyncGenerator, Generator
pass pass

4
tests/test_config_loader.py
View File

@@ -143,9 +143,7 @@ class TestConfigLoaderEnv:
assert config.exempt_ips == {"127.0.0.1", "192.168.1.1", "10.0.0.1"} assert config.exempt_ips == {"127.0.0.1", "192.168.1.1", "10.0.0.1"}
assert config.exempt_paths == {"/health", "/metrics"} assert config.exempt_paths == {"/health", "/metrics"}
def test_load_global_config_from_env_empty_sets( def test_load_global_config_from_env_empty_sets(self, loader: ConfigLoader) -> None:
self, loader: ConfigLoader
) -> None:
"""Test loading GlobalConfig with empty set fields.""" """Test loading GlobalConfig with empty set fields."""
env_vars = { env_vars = {
"FASTAPI_TRAFFIC_GLOBAL_EXEMPT_IPS": "", "FASTAPI_TRAFFIC_GLOBAL_EXEMPT_IPS": "",