Skip to content

Commit

Permalink
Fix code formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
victor-torres committed Sep 30, 2024
1 parent f439d0e commit 41f4638
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 4 deletions.
2 changes: 1 addition & 1 deletion tests/middleware/test_proxy_headers.py
Original file line number Diff line number Diff line change
Expand Up @@ -432,7 +432,7 @@ async def test_proxy_headers_multiple_proxies(trusted_hosts: str | list[str], ex
(True, "1234", 1234),
(False, "1234", 0),
(False, "443", 0),
]
],
)
async def test_proxy_headers_with_forwarded_port(forwarded_port, port, expected) -> None:
async with make_httpx_client("*", forwarded_port=forwarded_port) as client:
Expand Down
4 changes: 3 additions & 1 deletion uvicorn/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -470,7 +470,9 @@ def load(self) -> None:
if logger.getEffectiveLevel() <= TRACE_LOG_LEVEL:
self.loaded_app = MessageLoggerMiddleware(self.loaded_app)
if self.proxy_headers:
self.loaded_app = ProxyHeadersMiddleware(self.loaded_app, trusted_hosts=self.forwarded_allow_ips, forwarded_port=self.forwarded_port)
self.loaded_app = ProxyHeadersMiddleware(
self.loaded_app, trusted_hosts=self.forwarded_allow_ips, forwarded_port=self.forwarded_port
)

self.loaded = True

Expand Down
5 changes: 3 additions & 2 deletions uvicorn/middleware/proxy_headers.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,9 @@ class ProxyHeadersMiddleware:
- <https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/X-Forwarded-For>
"""

def __init__(self, app: ASGI3Application, trusted_hosts: list[str] | str = "127.0.0.1",
forwarded_port: bool = False) -> None:
def __init__(
self, app: ASGI3Application, trusted_hosts: list[str] | str = "127.0.0.1", forwarded_port: bool = False
) -> None:
self.app = app
self.trusted_hosts = _TrustedHosts(trusted_hosts)
self.forwarded_port = forwarded_port
Expand Down

0 comments on commit 41f4638

Please sign in to comment.