maya-chen/flask
1
0
Fork 0
forked from orbit-oss/flask
Code Pull requests Activity

werkzeug 2.3.3 compatibility

Browse source
This commit is contained in:
David Lord 2023-05-02 07:05:51 -07:00
parent 726d3f4fa9
commit 3fbfbad79f
No known key found for this signature in database
GPG key ID: 7A1C87E3F5BC42A8
2 changed files with 16 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
CHANGES.rst
View file

@ -3,6 +3,8 @@ Version 2.2.5
Unreleased Unreleased
- Update for compatibility with Werkzeug 2.3.3.
Version 2.2.4 Version 2.2.4
------------- -------------

17
src/flask/testing.py
View file

@ -168,10 +168,21 @@ class FlaskClient(Client):
app.session_interface.save_session(app, sess, resp) app.session_interface.save_session(app, sess, resp)
if hasattr(self, "_update_cookies_from_response"): if hasattr(self, "_update_cookies_from_response"):
self._update_cookies_from_response( try:
ctx.request.host.partition(":")[0], resp.headers.getlist("Set-Cookie") # Werkzeug>=2.3.3
) self._update_cookies_from_response(
ctx.request.host.partition(":")[0],
ctx.request.path,
resp.headers.getlist("Set-Cookie"),
)
except TypeError:
# Werkzeug>=2.3.0,<2.3.3
self._update_cookies_from_response( # type: ignore[call-arg]
ctx.request.host.partition(":")[0],
resp.headers.getlist("Set-Cookie"), # type: ignore[arg-type]
)
else: else:
# Werkzeug<2.3.0
self.cookie_jar.extract_wsgi( # type: ignore[union-attr] self.cookie_jar.extract_wsgi( # type: ignore[union-attr]
ctx.request.environ, resp.headers ctx.request.environ, resp.headers
) )