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

Move give_theme_cookie to flasktools.

Browse source
This commit is contained in:
voussoir 2021-10-30 17:29:28 -07:00
parent f267d55f9d
commit bae4953097
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 22 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
voussoirkit/flasktools.py
View file

@ -105,6 +105,28 @@ def ensure_response_type(function):
return response return response
return wrapped return wrapped
def give_theme_cookie(function, *, cookie_name, default_theme):
@functools.wraps(function)
def wrapped(*args, **kwargs):
old_theme = request.cookies.get(cookie_name, None)
new_theme = request.args.get('theme', None)
theme = new_theme or old_theme or default_theme
request.cookies = werkzeug.datastructures.MultiDict(request.cookies)
request.cookies[cookie_name] = theme
response = function(*args, **kwargs)
if new_theme is None:
pass
elif new_theme == '':
response.set_cookie(cookie_name, value='', expires=0)
elif new_theme != old_theme:
response.set_cookie(cookie_name, value=new_theme, expires=2147483647)
return response
return wrapped
def gzip_response(request, response): def gzip_response(request, response):
if response.direct_passthrough: if response.direct_passthrough:
return response return response