From 34e07f6b7e83dc78d9bf4723273f9c49a8f7f731 Mon Sep 17 00:00:00 2001 From: zanewalker Date: Wed, 4 Feb 2026 01:28:45 +0000 Subject: [PATCH] style: apply ruff formatting --- fastapi_traffic/core/config_loader.py | 22 ++++++++++++++-------- fastapi_traffic/core/decorator.py | 1 + tests/conftest.py | 1 + tests/test_config_loader.py | 4 +--- 4 files changed, 17 insertions(+), 11 deletions(-) diff --git a/fastapi_traffic/core/config_loader.py b/fastapi_traffic/core/config_loader.py index afe4ef6..59447f1 100644 --- a/fastapi_traffic/core/config_loader.py +++ b/fastapi_traffic/core/config_loader.py @@ -49,12 +49,14 @@ _GLOBAL_FIELD_TYPES: dict[str, type[Any]] = { } # Fields that cannot be loaded from config files (callables, complex objects) -_NON_LOADABLE_FIELDS: frozenset[str] = frozenset({ - "key_extractor", - "exempt_when", - "on_blocked", - "backend", -}) +_NON_LOADABLE_FIELDS: frozenset[str] = frozenset( + { + "key_extractor", + "exempt_when", + "on_blocked", + "backend", + } +) class ConfigLoader: @@ -200,7 +202,11 @@ class ConfigLoader: value = value.strip() # 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] env_vars[key] = value @@ -261,7 +267,7 @@ class ConfigLoader: for key, value in source.items(): if key.startswith(full_prefix): - field_name = key[len(full_prefix):].lower() + field_name = key[len(full_prefix) :].lower() if field_name in field_types: result[field_name] = value diff --git a/fastapi_traffic/core/decorator.py b/fastapi_traffic/core/decorator.py index 4f99d3c..4fd8bd7 100644 --- a/fastapi_traffic/core/decorator.py +++ b/fastapi_traffic/core/decorator.py @@ -24,6 +24,7 @@ F = TypeVar("F", bound=Callable[..., Any]) # Note: Config loader from secrets .env + @overload def rate_limit( limit: int, diff --git a/tests/conftest.py b/tests/conftest.py index 7d05c09..34afc1b 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -24,6 +24,7 @@ from fastapi_traffic.middleware import RateLimitMiddleware if TYPE_CHECKING: from collections.abc import AsyncGenerator, Generator + pass diff --git a/tests/test_config_loader.py b/tests/test_config_loader.py index bacd9ae..0d21651 100644 --- a/tests/test_config_loader.py +++ b/tests/test_config_loader.py @@ -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_paths == {"/health", "/metrics"} - def test_load_global_config_from_env_empty_sets( - self, loader: ConfigLoader - ) -> None: + def test_load_global_config_from_env_empty_sets(self, loader: ConfigLoader) -> None: """Test loading GlobalConfig with empty set fields.""" env_vars = { "FASTAPI_TRAFFIC_GLOBAL_EXEMPT_IPS": "",