uvicorn 0.34.2__py3-none-any.whl → 0.35.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
uvicorn/__init__.py CHANGED
@@ -1,5 +1,5 @@
1
1
  from uvicorn.config import Config
2
2
  from uvicorn.main import Server, main, run
3
3
 
4
- __version__ = "0.34.2"
4
+ __version__ = "0.35.0"
5
5
  __all__ = ["main", "run", "Config", "Server"]
uvicorn/config.py CHANGED
@@ -25,7 +25,7 @@ from uvicorn.middleware.proxy_headers import ProxyHeadersMiddleware
25
25
  from uvicorn.middleware.wsgi import WSGIMiddleware
26
26
 
27
27
  HTTPProtocolType = Literal["auto", "h11", "httptools"]
28
- WSProtocolType = Literal["auto", "none", "websockets", "wsproto"]
28
+ WSProtocolType = Literal["auto", "none", "websockets", "websockets-sansio", "wsproto"]
29
29
  LifespanType = Literal["auto", "on", "off"]
30
30
  LoopSetupType = Literal["none", "auto", "asyncio", "uvloop"]
31
31
  InterfaceType = Literal["auto", "asgi3", "asgi2", "wsgi"]
@@ -47,6 +47,7 @@ WS_PROTOCOLS: dict[WSProtocolType, str | None] = {
47
47
  "auto": "uvicorn.protocols.websockets.auto:AutoWebSocketsProtocol",
48
48
  "none": None,
49
49
  "websockets": "uvicorn.protocols.websockets.websockets_impl:WebSocketProtocol",
50
+ "websockets-sansio": "uvicorn.protocols.websockets.websockets_sansio_impl:WebSocketsSansIOProtocol",
50
51
  "wsproto": "uvicorn.protocols.websockets.wsproto_impl:WSProtocol",
51
52
  }
52
53
  LIFESPAN: dict[LifespanType, str] = {
@@ -313,7 +314,7 @@ class Config:
313
314
  + "directories, watching current working directory.",
314
315
  reload_dirs,
315
316
  )
316
- self.reload_dirs = [Path(os.getcwd())]
317
+ self.reload_dirs = [Path.cwd()]
317
318
 
318
319
  logger.info(
319
320
  "Will watch for changes in these directories: %s",
uvicorn/main.py CHANGED
@@ -223,7 +223,7 @@ def print_version(ctx: click.Context, param: click.Parameter, value: bool) -> No
223
223
  "--proxy-headers/--no-proxy-headers",
224
224
  is_flag=True,
225
225
  default=True,
226
- help="Enable/Disable X-Forwarded-Proto, X-Forwarded-For, X-Forwarded-Port to populate remote address info.",
226
+ help="Enable/Disable X-Forwarded-Proto, X-Forwarded-For to populate url scheme and remote address info.",
227
227
  )
228
228
  @click.option(
229
229
  "--server-header/--no-server-header",
@@ -1,9 +1,9 @@
1
1
  from __future__ import annotations
2
2
 
3
3
  import asyncio
4
- import typing
4
+ from typing import Callable
5
5
 
6
- AutoWebSocketsProtocol: typing.Callable[..., asyncio.Protocol] | None
6
+ AutoWebSocketsProtocol: Callable[..., asyncio.Protocol] | None
7
7
  try:
8
8
  import websockets # noqa
9
9
  except ImportError: # pragma: no cover
@@ -34,6 +34,7 @@ from uvicorn.config import Config
34
34
  from uvicorn.logging import TRACE_LOG_LEVEL
35
35
  from uvicorn.protocols.utils import (
36
36
  ClientDisconnected,
37
+ get_client_addr,
37
38
  get_local_addr,
38
39
  get_path_with_query_string,
39
40
  get_remote_addr,
@@ -271,7 +272,7 @@ class WebSocketProtocol(WebSocketServerProtocol):
271
272
  message = cast("WebSocketAcceptEvent", message)
272
273
  self.logger.info(
273
274
  '%s - "WebSocket %s" [accepted]',
274
- self.scope["client"],
275
+ get_client_addr(self.scope),
275
276
  get_path_with_query_string(self.scope),
276
277
  )
277
278
  self.initial_response = None
@@ -289,7 +290,7 @@ class WebSocketProtocol(WebSocketServerProtocol):
289
290
  message = cast("WebSocketCloseEvent", message)
290
291
  self.logger.info(
291
292
  '%s - "WebSocket %s" 403',
292
- self.scope["client"],
293
+ get_client_addr(self.scope),
293
294
  get_path_with_query_string(self.scope),
294
295
  )
295
296
  self.initial_response = (http.HTTPStatus.FORBIDDEN, [], b"")
@@ -300,7 +301,7 @@ class WebSocketProtocol(WebSocketServerProtocol):
300
301
  message = cast("WebSocketResponseStartEvent", message)
301
302
  self.logger.info(
302
303
  '%s - "WebSocket %s" %d',
303
- self.scope["client"],
304
+ get_client_addr(self.scope),
304
305
  get_path_with_query_string(self.scope),
305
306
  message["status"],
306
307
  )
@@ -0,0 +1,417 @@
1
+ from __future__ import annotations
2
+
3
+ import asyncio
4
+ import logging
5
+ import sys
6
+ from asyncio.transports import BaseTransport, Transport
7
+ from http import HTTPStatus
8
+ from typing import Any, Literal, cast
9
+ from urllib.parse import unquote
10
+
11
+ from websockets.exceptions import InvalidState
12
+ from websockets.extensions.permessage_deflate import ServerPerMessageDeflateFactory
13
+ from websockets.frames import Frame, Opcode
14
+ from websockets.http11 import Request
15
+ from websockets.server import ServerProtocol
16
+
17
+ from uvicorn._types import (
18
+ ASGIReceiveEvent,
19
+ ASGISendEvent,
20
+ WebSocketAcceptEvent,
21
+ WebSocketCloseEvent,
22
+ WebSocketResponseBodyEvent,
23
+ WebSocketResponseStartEvent,
24
+ WebSocketScope,
25
+ WebSocketSendEvent,
26
+ )
27
+ from uvicorn.config import Config
28
+ from uvicorn.logging import TRACE_LOG_LEVEL
29
+ from uvicorn.protocols.utils import (
30
+ ClientDisconnected,
31
+ get_local_addr,
32
+ get_path_with_query_string,
33
+ get_remote_addr,
34
+ is_ssl,
35
+ )
36
+ from uvicorn.server import ServerState
37
+
38
+ if sys.version_info >= (3, 11): # pragma: no cover
39
+ from typing import assert_never
40
+ else: # pragma: no cover
41
+ from typing_extensions import assert_never
42
+
43
+
44
+ class WebSocketsSansIOProtocol(asyncio.Protocol):
45
+ def __init__(
46
+ self,
47
+ config: Config,
48
+ server_state: ServerState,
49
+ app_state: dict[str, Any],
50
+ _loop: asyncio.AbstractEventLoop | None = None,
51
+ ) -> None:
52
+ if not config.loaded:
53
+ config.load() # pragma: no cover
54
+
55
+ self.config = config
56
+ self.app = config.loaded_app
57
+ self.loop = _loop or asyncio.get_event_loop()
58
+ self.logger = logging.getLogger("uvicorn.error")
59
+ self.root_path = config.root_path
60
+ self.app_state = app_state
61
+
62
+ # Shared server state
63
+ self.connections = server_state.connections
64
+ self.tasks = server_state.tasks
65
+ self.default_headers = server_state.default_headers
66
+
67
+ # Connection state
68
+ self.transport: asyncio.Transport = None # type: ignore[assignment]
69
+ self.server: tuple[str, int] | None = None
70
+ self.client: tuple[str, int] | None = None
71
+ self.scheme: Literal["wss", "ws"] = None # type: ignore[assignment]
72
+
73
+ # WebSocket state
74
+ self.queue: asyncio.Queue[ASGIReceiveEvent] = asyncio.Queue()
75
+ self.handshake_initiated = False
76
+ self.handshake_complete = False
77
+ self.close_sent = False
78
+ self.initial_response: tuple[int, list[tuple[str, str]], bytes] | None = None
79
+
80
+ extensions = []
81
+ if self.config.ws_per_message_deflate:
82
+ extensions = [
83
+ ServerPerMessageDeflateFactory(
84
+ server_max_window_bits=12,
85
+ client_max_window_bits=12,
86
+ compress_settings={"memLevel": 5},
87
+ )
88
+ ]
89
+ self.conn = ServerProtocol(
90
+ extensions=extensions,
91
+ max_size=self.config.ws_max_size,
92
+ logger=logging.getLogger("uvicorn.error"),
93
+ )
94
+
95
+ self.read_paused = False
96
+ self.writable = asyncio.Event()
97
+ self.writable.set()
98
+
99
+ # Buffers
100
+ self.bytes = b""
101
+
102
+ def connection_made(self, transport: BaseTransport) -> None:
103
+ """Called when a connection is made."""
104
+ transport = cast(Transport, transport)
105
+ self.connections.add(self)
106
+ self.transport = transport
107
+ self.server = get_local_addr(transport)
108
+ self.client = get_remote_addr(transport)
109
+ self.scheme = "wss" if is_ssl(transport) else "ws"
110
+
111
+ if self.logger.level <= TRACE_LOG_LEVEL:
112
+ prefix = "%s:%d - " % self.client if self.client else ""
113
+ self.logger.log(TRACE_LOG_LEVEL, "%sWebSocket connection made", prefix)
114
+
115
+ def connection_lost(self, exc: Exception | None) -> None:
116
+ code = 1005 if self.handshake_complete else 1006
117
+ self.queue.put_nowait({"type": "websocket.disconnect", "code": code})
118
+ self.connections.remove(self)
119
+
120
+ if self.logger.level <= TRACE_LOG_LEVEL:
121
+ prefix = "%s:%d - " % self.client if self.client else ""
122
+ self.logger.log(TRACE_LOG_LEVEL, "%sWebSocket connection lost", prefix)
123
+
124
+ self.handshake_complete = True
125
+ if exc is None:
126
+ self.transport.close()
127
+
128
+ def eof_received(self) -> None:
129
+ pass
130
+
131
+ def shutdown(self) -> None:
132
+ if self.handshake_complete:
133
+ self.queue.put_nowait({"type": "websocket.disconnect", "code": 1012})
134
+ self.conn.send_close(1012)
135
+ output = self.conn.data_to_send()
136
+ self.transport.write(b"".join(output))
137
+ else:
138
+ self.send_500_response()
139
+ self.transport.close()
140
+
141
+ def data_received(self, data: bytes) -> None:
142
+ self.conn.receive_data(data)
143
+ if self.conn.parser_exc is not None: # pragma: no cover
144
+ self.handle_parser_exception()
145
+ return
146
+ self.handle_events()
147
+
148
+ def handle_events(self) -> None:
149
+ for event in self.conn.events_received():
150
+ if isinstance(event, Request):
151
+ self.handle_connect(event)
152
+ if isinstance(event, Frame):
153
+ if event.opcode == Opcode.CONT:
154
+ self.handle_cont(event) # pragma: no cover
155
+ elif event.opcode == Opcode.TEXT:
156
+ self.handle_text(event)
157
+ elif event.opcode == Opcode.BINARY:
158
+ self.handle_bytes(event)
159
+ elif event.opcode == Opcode.PING:
160
+ self.handle_ping()
161
+ elif event.opcode == Opcode.PONG:
162
+ pass # pragma: no cover
163
+ elif event.opcode == Opcode.CLOSE:
164
+ self.handle_close(event)
165
+ else:
166
+ assert_never(event.opcode) # pragma: no cover
167
+
168
+ # Event handlers
169
+
170
+ def handle_connect(self, event: Request) -> None:
171
+ self.request = event
172
+ self.response = self.conn.accept(event)
173
+ self.handshake_initiated = True
174
+ if self.response.status_code != 101:
175
+ self.handshake_complete = True
176
+ self.close_sent = True
177
+ self.conn.send_response(self.response)
178
+ output = self.conn.data_to_send()
179
+ self.transport.write(b"".join(output))
180
+ self.transport.close()
181
+ return
182
+
183
+ headers = [
184
+ (key.encode("ascii"), value.encode("ascii", errors="surrogateescape"))
185
+ for key, value in event.headers.raw_items()
186
+ ]
187
+ raw_path, _, query_string = event.path.partition("?")
188
+ self.scope: WebSocketScope = {
189
+ "type": "websocket",
190
+ "asgi": {"version": self.config.asgi_version, "spec_version": "2.3"},
191
+ "http_version": "1.1",
192
+ "scheme": self.scheme,
193
+ "server": self.server,
194
+ "client": self.client,
195
+ "root_path": self.root_path,
196
+ "path": unquote(raw_path),
197
+ "raw_path": raw_path.encode("ascii"),
198
+ "query_string": query_string.encode("ascii"),
199
+ "headers": headers,
200
+ "subprotocols": event.headers.get_all("Sec-WebSocket-Protocol"),
201
+ "state": self.app_state.copy(),
202
+ "extensions": {"websocket.http.response": {}},
203
+ }
204
+ self.queue.put_nowait({"type": "websocket.connect"})
205
+ task = self.loop.create_task(self.run_asgi())
206
+ task.add_done_callback(self.on_task_complete)
207
+ self.tasks.add(task)
208
+
209
+ def handle_cont(self, event: Frame) -> None: # pragma: no cover
210
+ self.bytes += event.data
211
+ if event.fin:
212
+ self.send_receive_event_to_app()
213
+
214
+ def handle_text(self, event: Frame) -> None:
215
+ self.bytes = event.data
216
+ self.curr_msg_data_type: Literal["text", "bytes"] = "text"
217
+ if event.fin:
218
+ self.send_receive_event_to_app()
219
+
220
+ def handle_bytes(self, event: Frame) -> None:
221
+ self.bytes = event.data
222
+ self.curr_msg_data_type = "bytes"
223
+ if event.fin:
224
+ self.send_receive_event_to_app()
225
+
226
+ def send_receive_event_to_app(self) -> None:
227
+ if self.curr_msg_data_type == "text":
228
+ try:
229
+ self.queue.put_nowait({"type": "websocket.receive", "text": self.bytes.decode()})
230
+ except UnicodeDecodeError: # pragma: no cover
231
+ self.logger.exception("Invalid UTF-8 sequence received from client.")
232
+ self.conn.send_close(1007)
233
+ self.handle_parser_exception()
234
+ return
235
+ else:
236
+ self.queue.put_nowait({"type": "websocket.receive", "bytes": self.bytes})
237
+ if not self.read_paused:
238
+ self.read_paused = True
239
+ self.transport.pause_reading()
240
+
241
+ def handle_ping(self) -> None:
242
+ output = self.conn.data_to_send()
243
+ self.transport.write(b"".join(output))
244
+
245
+ def handle_close(self, event: Frame) -> None:
246
+ if not self.close_sent and not self.transport.is_closing():
247
+ assert self.conn.close_rcvd is not None
248
+ code = self.conn.close_rcvd.code
249
+ reason = self.conn.close_rcvd.reason
250
+ self.queue.put_nowait({"type": "websocket.disconnect", "code": code, "reason": reason})
251
+
252
+ output = self.conn.data_to_send()
253
+ self.transport.write(b"".join(output))
254
+ self.transport.close()
255
+
256
+ def handle_parser_exception(self) -> None: # pragma: no cover
257
+ assert self.conn.close_sent is not None
258
+ code = self.conn.close_sent.code
259
+ reason = self.conn.close_sent.reason
260
+ self.queue.put_nowait({"type": "websocket.disconnect", "code": code, "reason": reason})
261
+
262
+ output = self.conn.data_to_send()
263
+ self.transport.write(b"".join(output))
264
+ self.close_sent = True
265
+ self.transport.close()
266
+
267
+ def on_task_complete(self, task: asyncio.Task[None]) -> None:
268
+ self.tasks.discard(task)
269
+
270
+ async def run_asgi(self) -> None:
271
+ try:
272
+ result = await self.app(self.scope, self.receive, self.send)
273
+ except ClientDisconnected:
274
+ self.transport.close() # pragma: no cover
275
+ except BaseException:
276
+ self.logger.exception("Exception in ASGI application\n")
277
+ self.send_500_response()
278
+ self.transport.close()
279
+ else:
280
+ if not self.handshake_complete:
281
+ self.logger.error("ASGI callable returned without completing handshake.")
282
+ self.send_500_response()
283
+ self.transport.close()
284
+ elif result is not None:
285
+ self.logger.error("ASGI callable should return None, but returned '%s'.", result)
286
+ self.transport.close()
287
+
288
+ def send_500_response(self) -> None:
289
+ if self.initial_response or self.handshake_complete:
290
+ return
291
+ response = self.conn.reject(500, "Internal Server Error")
292
+ self.conn.send_response(response)
293
+ output = self.conn.data_to_send()
294
+ self.transport.write(b"".join(output))
295
+
296
+ async def send(self, message: ASGISendEvent) -> None:
297
+ await self.writable.wait()
298
+
299
+ message_type = message["type"]
300
+
301
+ if not self.handshake_complete and self.initial_response is None:
302
+ if message_type == "websocket.accept":
303
+ message = cast(WebSocketAcceptEvent, message)
304
+ self.logger.info(
305
+ '%s - "WebSocket %s" [accepted]',
306
+ self.scope["client"],
307
+ get_path_with_query_string(self.scope),
308
+ )
309
+ headers = [
310
+ (name.decode("latin-1").lower(), value.decode("latin-1").lower())
311
+ for name, value in (self.default_headers + list(message.get("headers", [])))
312
+ ]
313
+ accepted_subprotocol = message.get("subprotocol")
314
+ if accepted_subprotocol:
315
+ headers.append(("Sec-WebSocket-Protocol", accepted_subprotocol))
316
+ self.response.headers.update(headers)
317
+
318
+ if not self.transport.is_closing():
319
+ self.handshake_complete = True
320
+ self.conn.send_response(self.response)
321
+ output = self.conn.data_to_send()
322
+ self.transport.write(b"".join(output))
323
+
324
+ elif message_type == "websocket.close":
325
+ message = cast(WebSocketCloseEvent, message)
326
+ self.queue.put_nowait({"type": "websocket.disconnect", "code": 1006})
327
+ self.logger.info(
328
+ '%s - "WebSocket %s" 403',
329
+ self.scope["client"],
330
+ get_path_with_query_string(self.scope),
331
+ )
332
+ response = self.conn.reject(HTTPStatus.FORBIDDEN, "")
333
+ self.conn.send_response(response)
334
+ output = self.conn.data_to_send()
335
+ self.close_sent = True
336
+ self.handshake_complete = True
337
+ self.transport.write(b"".join(output))
338
+ self.transport.close()
339
+ elif message_type == "websocket.http.response.start" and self.initial_response is None:
340
+ message = cast(WebSocketResponseStartEvent, message)
341
+ if not (100 <= message["status"] < 600):
342
+ raise RuntimeError("Invalid HTTP status code '%d' in response." % message["status"])
343
+ self.logger.info(
344
+ '%s - "WebSocket %s" %d',
345
+ self.scope["client"],
346
+ get_path_with_query_string(self.scope),
347
+ message["status"],
348
+ )
349
+ headers = [
350
+ (name.decode("latin-1"), value.decode("latin-1"))
351
+ for name, value in list(message.get("headers", []))
352
+ ]
353
+ self.initial_response = (message["status"], headers, b"")
354
+ else:
355
+ msg = (
356
+ "Expected ASGI message 'websocket.accept', 'websocket.close' "
357
+ "or 'websocket.http.response.start' "
358
+ "but got '%s'."
359
+ )
360
+ raise RuntimeError(msg % message_type)
361
+
362
+ elif not self.close_sent and self.initial_response is None:
363
+ try:
364
+ if message_type == "websocket.send":
365
+ message = cast(WebSocketSendEvent, message)
366
+ bytes_data = message.get("bytes")
367
+ text_data = message.get("text")
368
+ if text_data:
369
+ self.conn.send_text(text_data.encode())
370
+ elif bytes_data:
371
+ self.conn.send_binary(bytes_data)
372
+ output = self.conn.data_to_send()
373
+ self.transport.write(b"".join(output))
374
+
375
+ elif message_type == "websocket.close" and not self.transport.is_closing():
376
+ message = cast(WebSocketCloseEvent, message)
377
+ code = message.get("code", 1000)
378
+ reason = message.get("reason", "") or ""
379
+ self.queue.put_nowait({"type": "websocket.disconnect", "code": code, "reason": reason})
380
+ self.conn.send_close(code, reason)
381
+ output = self.conn.data_to_send()
382
+ self.transport.write(b"".join(output))
383
+ self.close_sent = True
384
+ self.transport.close()
385
+ else:
386
+ msg = "Expected ASGI message 'websocket.send' or 'websocket.close', but got '%s'."
387
+ raise RuntimeError(msg % message_type)
388
+ except InvalidState:
389
+ raise ClientDisconnected()
390
+ elif self.initial_response is not None:
391
+ if message_type == "websocket.http.response.body":
392
+ message = cast(WebSocketResponseBodyEvent, message)
393
+ body = self.initial_response[2] + message["body"]
394
+ self.initial_response = self.initial_response[:2] + (body,)
395
+ if not message.get("more_body", False):
396
+ response = self.conn.reject(self.initial_response[0], body.decode())
397
+ response.headers.update(self.initial_response[1])
398
+ self.queue.put_nowait({"type": "websocket.disconnect", "code": 1006})
399
+ self.conn.send_response(response)
400
+ output = self.conn.data_to_send()
401
+ self.close_sent = True
402
+ self.transport.write(b"".join(output))
403
+ self.transport.close()
404
+ else: # pragma: no cover
405
+ msg = "Expected ASGI message 'websocket.http.response.body' but got '%s'."
406
+ raise RuntimeError(msg % message_type)
407
+
408
+ else:
409
+ msg = "Unexpected ASGI message '%s', after sending 'websocket.close'."
410
+ raise RuntimeError(msg % message_type)
411
+
412
+ async def receive(self) -> ASGIReceiveEvent:
413
+ message = await self.queue.get()
414
+ if self.read_paused and self.queue.empty():
415
+ self.read_paused = False
416
+ self.transport.resume_reading()
417
+ return message
@@ -2,8 +2,7 @@ from __future__ import annotations
2
2
 
3
3
  import asyncio
4
4
  import logging
5
- import typing
6
- from typing import Literal, cast
5
+ from typing import Any, Literal, cast
7
6
  from urllib.parse import unquote
8
7
 
9
8
  import wsproto
@@ -27,6 +26,7 @@ from uvicorn.config import Config
27
26
  from uvicorn.logging import TRACE_LOG_LEVEL
28
27
  from uvicorn.protocols.utils import (
29
28
  ClientDisconnected,
29
+ get_client_addr,
30
30
  get_local_addr,
31
31
  get_path_with_query_string,
32
32
  get_remote_addr,
@@ -40,7 +40,7 @@ class WSProtocol(asyncio.Protocol):
40
40
  self,
41
41
  config: Config,
42
42
  server_state: ServerState,
43
- app_state: dict[str, typing.Any],
43
+ app_state: dict[str, Any],
44
44
  _loop: asyncio.AbstractEventLoop | None = None,
45
45
  ) -> None:
46
46
  if not config.loaded:
@@ -255,10 +255,10 @@ class WSProtocol(asyncio.Protocol):
255
255
 
256
256
  if not self.handshake_complete:
257
257
  if message_type == "websocket.accept":
258
- message = typing.cast(WebSocketAcceptEvent, message)
258
+ message = cast(WebSocketAcceptEvent, message)
259
259
  self.logger.info(
260
260
  '%s - "WebSocket %s" [accepted]',
261
- self.scope["client"],
261
+ get_client_addr(self.scope),
262
262
  get_path_with_query_string(self.scope),
263
263
  )
264
264
  subprotocol = message.get("subprotocol")
@@ -281,7 +281,7 @@ class WSProtocol(asyncio.Protocol):
281
281
  self.queue.put_nowait({"type": "websocket.disconnect", "code": 1006})
282
282
  self.logger.info(
283
283
  '%s - "WebSocket %s" 403',
284
- self.scope["client"],
284
+ get_client_addr(self.scope),
285
285
  get_path_with_query_string(self.scope),
286
286
  )
287
287
  self.handshake_complete = True
@@ -292,14 +292,14 @@ class WSProtocol(asyncio.Protocol):
292
292
  self.transport.close()
293
293
 
294
294
  elif message_type == "websocket.http.response.start":
295
- message = typing.cast(WebSocketResponseStartEvent, message)
295
+ message = cast(WebSocketResponseStartEvent, message)
296
296
  # ensure status code is in the valid range
297
297
  if not (100 <= message["status"] < 600):
298
298
  msg = "Invalid HTTP status code '%d' in response."
299
299
  raise RuntimeError(msg % message["status"])
300
300
  self.logger.info(
301
301
  '%s - "WebSocket %s" %d',
302
- self.scope["client"],
302
+ get_client_addr(self.scope),
303
303
  get_path_with_query_string(self.scope),
304
304
  message["status"],
305
305
  )
@@ -324,7 +324,7 @@ class WSProtocol(asyncio.Protocol):
324
324
  elif not self.close_sent and not self.response_started:
325
325
  try:
326
326
  if message_type == "websocket.send":
327
- message = typing.cast(WebSocketSendEvent, message)
327
+ message = cast(WebSocketSendEvent, message)
328
328
  bytes_data = message.get("bytes")
329
329
  text_data = message.get("text")
330
330
  data = text_data if bytes_data is None else bytes_data
@@ -333,7 +333,7 @@ class WSProtocol(asyncio.Protocol):
333
333
  self.transport.write(output)
334
334
 
335
335
  elif message_type == "websocket.close":
336
- message = typing.cast(WebSocketCloseEvent, message)
336
+ message = cast(WebSocketCloseEvent, message)
337
337
  self.close_sent = True
338
338
  code = message.get("code", 1000)
339
339
  reason = message.get("reason", "") or ""
@@ -350,7 +350,7 @@ class WSProtocol(asyncio.Protocol):
350
350
  raise ClientDisconnected from exc
351
351
  elif self.response_started:
352
352
  if message_type == "websocket.http.response.body":
353
- message = typing.cast("WebSocketResponseBodyEvent", message)
353
+ message = cast("WebSocketResponseBodyEvent", message)
354
354
  body_finished = not message.get("more_body", False)
355
355
  reject_data = events.RejectData(data=message["body"], body_finished=body_finished)
356
356
  output = self.conn.send(reject_data)
uvicorn/server.py CHANGED
@@ -23,9 +23,10 @@ if TYPE_CHECKING:
23
23
  from uvicorn.protocols.http.h11_impl import H11Protocol
24
24
  from uvicorn.protocols.http.httptools_impl import HttpToolsProtocol
25
25
  from uvicorn.protocols.websockets.websockets_impl import WebSocketProtocol
26
+ from uvicorn.protocols.websockets.websockets_sansio_impl import WebSocketsSansIOProtocol
26
27
  from uvicorn.protocols.websockets.wsproto_impl import WSProtocol
27
28
 
28
- Protocols = Union[H11Protocol, HttpToolsProtocol, WSProtocol, WebSocketProtocol]
29
+ Protocols = Union[H11Protocol, HttpToolsProtocol, WSProtocol, WebSocketProtocol, WebSocketsSansIOProtocol]
29
30
 
30
31
  HANDLED_SIGNALS = (
31
32
  signal.SIGINT, # Unix signal 2. Sent by Ctrl+C.
@@ -63,10 +63,7 @@ class WatchFilesReload(BaseReload):
63
63
  self.reloader_name = "WatchFiles"
64
64
  self.reload_dirs = []
65
65
  for directory in config.reload_dirs:
66
- if Path.cwd() not in directory.parents:
67
- self.reload_dirs.append(directory)
68
- if Path.cwd() not in self.reload_dirs:
69
- self.reload_dirs.append(Path.cwd())
66
+ self.reload_dirs.append(directory)
70
67
 
71
68
  self.watch_filter = FileFilter(config)
72
69
  self.watcher = watch(
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: uvicorn
3
- Version: 0.34.2
3
+ Version: 0.35.0
4
4
  Summary: The lightning-fast ASGI server.
5
5
  Project-URL: Changelog, https://www.uvicorn.org/release-notes
6
6
  Project-URL: Funding, https://github.com/sponsors/encode
@@ -32,7 +32,7 @@ Requires-Dist: colorama>=0.4; (sys_platform == 'win32') and extra == 'standard'
32
32
  Requires-Dist: httptools>=0.6.3; extra == 'standard'
33
33
  Requires-Dist: python-dotenv>=0.13; extra == 'standard'
34
34
  Requires-Dist: pyyaml>=5.1; extra == 'standard'
35
- Requires-Dist: uvloop!=0.15.0,!=0.15.1,>=0.14.0; (sys_platform != 'win32' and (sys_platform != 'cygwin' and platform_python_implementation != 'PyPy')) and extra == 'standard'
35
+ Requires-Dist: uvloop>=0.15.1; (sys_platform != 'win32' and (sys_platform != 'cygwin' and platform_python_implementation != 'PyPy')) and extra == 'standard'
36
36
  Requires-Dist: watchfiles>=0.13; extra == 'standard'
37
37
  Requires-Dist: websockets>=10.4; extra == 'standard'
38
38
  Description-Content-Type: text/markdown
@@ -1,13 +1,13 @@
1
- uvicorn/__init__.py,sha256=jcZ4XeMQu13Rqvp-TeOhLhZn26yKMbfQ9lr6heDAsO8,147
1
+ uvicorn/__init__.py,sha256=pMIkABPLJ9eS0OuDO-bP506aVdLfWlHDtk9uzSTgE5E,147
2
2
  uvicorn/__main__.py,sha256=DQizy6nKP0ywhPpnCHgmRDYIMfcqZKVEzNIWQZjqtVQ,62
3
3
  uvicorn/_subprocess.py,sha256=HbfRnsCkXyg7xCWVAWWzXQTeWlvLKfTlIF5wevFBkR4,2766
4
4
  uvicorn/_types.py,sha256=5FcPvvIfeKsJDjGhTrceDv8TmwzYI8yPF7mXsXTWOUM,7775
5
- uvicorn/config.py,sha256=zg-UX2vu3Zy0e7jXOJKoY1mPhsbIuq-3IwwKy9yERkg,20894
5
+ uvicorn/config.py,sha256=OHgnEBBuk7XMhwKByt1hlBJsXk8EGWRYGpCQ_G1deaA,21013
6
6
  uvicorn/importer.py,sha256=nRt0QQ3qpi264-n_mR0l55C2ddM8nowTNzT1jsWaam8,1128
7
7
  uvicorn/logging.py,sha256=-eCE4nOJmFbtB9qfNJuEVNF0Y13LGUHqvFzemYT0PaQ,4235
8
- uvicorn/main.py,sha256=5TFzub2UbSk4LOQ_UeQ3PZLiTRG51KIQ0sSY9SZUMN8,17234
8
+ uvicorn/main.py,sha256=83-MEMVc6ErEdvUfW6x-gELYJ78GiWRNMsmuing9DUI,17231
9
9
  uvicorn/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
10
- uvicorn/server.py,sha256=Kd2S0UarzwJyt9YHZzTghMPxxRVa1zIVA8Cqy6AuT9A,12879
10
+ uvicorn/server.py,sha256=3AKFM50WbZkwnqjZFxSd8wLBVIkcxZjmvP8gPZf95_s,12998
11
11
  uvicorn/workers.py,sha256=7KGgGAapxGkGeXUcBGunOjSNdI9R44KCoWTGEAqAdfs,3895
12
12
  uvicorn/lifespan/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
13
13
  uvicorn/lifespan/off.py,sha256=nfI6qHAUo_8-BEXMBKoHQ9wUbsXrPaXLCbDSS0vKSr8,332
@@ -29,16 +29,17 @@ uvicorn/protocols/http/flow_control.py,sha256=050WVg31EvPOkHwynCoMP1zXFl_vO3U4du
29
29
  uvicorn/protocols/http/h11_impl.py,sha256=4b-KswK57FBaRPeHXlK_oy8pKM6vG1haALCIeRH-1bA,20694
30
30
  uvicorn/protocols/http/httptools_impl.py,sha256=tuQBCiD6rf5DQeyQwHVc45rxH9ouojMpkXi9KG6NRBk,21805
31
31
  uvicorn/protocols/websockets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
32
- uvicorn/protocols/websockets/auto.py,sha256=kNP-h07ZzjA9dKRUd7MNO0J7xhRJ5xVBfit7wCbdB0A,574
33
- uvicorn/protocols/websockets/websockets_impl.py,sha256=E0e7aX4ICmSIuytfc4d5RhWvjH37Ed8ZgXwAH2f8oog,15504
34
- uvicorn/protocols/websockets/wsproto_impl.py,sha256=2OB4E6OsQ1KPiETkHYi2UQDxY9sUzTr0LLZQJxlXPxo,15375
32
+ uvicorn/protocols/websockets/auto.py,sha256=SH_KV_3vwR8_oGda2GrHFt38VG-IwY0ufjvHOu4VA0o,581
33
+ uvicorn/protocols/websockets/websockets_impl.py,sha256=qonQJxz9wMKMwB2RnYZezeP-XfmAsxCvNeGgu4sC3Lc,15546
34
+ uvicorn/protocols/websockets/websockets_sansio_impl.py,sha256=TxpjkbuhaTd0UXYc1VMe4UgBr6kn_JwJV-hqegHHmxs,17145
35
+ uvicorn/protocols/websockets/wsproto_impl.py,sha256=u2TKyzRUCmQpS1e4E_X6Uou9QIuoKZNNNmHU1IzkWPU,15366
35
36
  uvicorn/supervisors/__init__.py,sha256=wT8eOEIqT1yWQgytZtv5taWMul7xoTIY0xm1m4oyPTw,507
36
37
  uvicorn/supervisors/basereload.py,sha256=MAXSQ3ckZPwqzJ8Un9yDhk3W0yyAArQtZLuOE0OInSc,4036
37
38
  uvicorn/supervisors/multiprocess.py,sha256=Opt0XvOUj1DIMXYwb4OlkJZxeh_RjweFnTmDPYItONw,7507
38
39
  uvicorn/supervisors/statreload.py,sha256=uYblmoxM3IbPbvMDzr5Abw2-WykQl8NxTTzeLfVyvnU,1566
39
- uvicorn/supervisors/watchfilesreload.py,sha256=41FGNMXPKrKvPr-5O8yRWg43l6OCBtapt39M-gpdk0E,3010
40
- uvicorn-0.34.2.dist-info/METADATA,sha256=kUM_FZJJlmpGa7hrum4L1EHIkhbs-vhpPCPRMSBD8QY,6549
41
- uvicorn-0.34.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
42
- uvicorn-0.34.2.dist-info/entry_points.txt,sha256=FW1w-hkc9QgwaGoovMvm0ZY73w_NcycWdGAUfDsNGxw,46
43
- uvicorn-0.34.2.dist-info/licenses/LICENSE.md,sha256=7-Gs8-YvuZwoiw7HPlp3O3Jo70Mg_nV-qZQhTktjw3E,1526
44
- uvicorn-0.34.2.dist-info/RECORD,,
40
+ uvicorn/supervisors/watchfilesreload.py,sha256=W86Ybb0E5SdMYYuWHJ3bpAFXdw5ZurvLRdFcvLnYEIA,2859
41
+ uvicorn-0.35.0.dist-info/METADATA,sha256=a1sVo25u3fbfeGXSMmFdYOuPPnBXF8WjHlYq1oc2qTo,6531
42
+ uvicorn-0.35.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
43
+ uvicorn-0.35.0.dist-info/entry_points.txt,sha256=FW1w-hkc9QgwaGoovMvm0ZY73w_NcycWdGAUfDsNGxw,46
44
+ uvicorn-0.35.0.dist-info/licenses/LICENSE.md,sha256=7-Gs8-YvuZwoiw7HPlp3O3Jo70Mg_nV-qZQhTktjw3E,1526
45
+ uvicorn-0.35.0.dist-info/RECORD,,