voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Improve technique for finding cookie set by function.

Browse source 
Instead of iterating through all the outgoing headers.
This commit is contained in:
voussoir 2018-01-15 18:56:41 -08:00
parent be0fca729f
commit 55e1281774
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
frontends/etiquette_flask/etiquette_flask/sessions.py
View file

@ -70,10 +70,8 @@ class SessionManager:
# Send the token back to the client # Send the token back to the client
# but only if the endpoint didn't manually set the cookie. # but only if the endpoint didn't manually set the cookie.
for (headerkey, value) in response.headers: function_cookies = response.headers.get_all('Set-Cookie')
if headerkey == 'Set-Cookie' and value.startswith('etiquette_session='): if not any('etiquette_session=' in cookie for cookie in function_cookies):
break
else:
response.set_cookie( response.set_cookie(
'etiquette_session', 'etiquette_session',
value=session.token, value=session.token,