From a6e20a8bf3998685a5a29e89be8ffd05f7237464 Mon Sep 17 00:00:00 2001 From: Dave Tapley Date: Mon, 29 Jan 2024 21:41:35 +0000 Subject: [PATCH 1/2] Add type to Request stream --- falcon/request.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/falcon/request.py b/falcon/request.py index cb037369d..565e43dc7 100644 --- a/falcon/request.py +++ b/falcon/request.py @@ -12,10 +12,15 @@ """Request class.""" + from datetime import datetime from io import BytesIO +import sys from uuid import UUID +if sys.version_info >= (3, 11): + from wsgiref.types import InputStream + from falcon import errors from falcon import request_helpers as helpers from falcon import util @@ -454,6 +459,9 @@ class Request: # Child classes may override this context_type = structures.Context + if sys.version_info >= (3, 11): + stream: InputStream + _wsgi_input_type_known = False def __init__(self, env, options=None): From b6f62c318039d5ca6774de865f3cf7edb1f97d77 Mon Sep 17 00:00:00 2001 From: Dave Tapley Date: Mon, 29 Jan 2024 21:55:29 +0000 Subject: [PATCH 2/2] Add type to Response text and _data --- falcon/request.py | 1 - falcon/response.py | 5 ++++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/falcon/request.py b/falcon/request.py index 565e43dc7..4bcbf1c06 100644 --- a/falcon/request.py +++ b/falcon/request.py @@ -12,7 +12,6 @@ """Request class.""" - from datetime import datetime from io import BytesIO import sys diff --git a/falcon/response.py b/falcon/response.py index cad529d3c..7377938be 100644 --- a/falcon/response.py +++ b/falcon/response.py @@ -174,6 +174,9 @@ class Response: # Child classes may override this context_type = structures.Context + text: Optional[str] + _data: Optional[bytes] + def __init__(self, options=None): self.status = '200 OK' self._headers = {} @@ -227,7 +230,7 @@ def data(self): return self._data @data.setter - def data(self, value): + def data(self, value: bytes): self._data = value @property