django/daphne
1
1
Fork 0
mirror of https://github.com/django/daphne.git synced 2025-04-22 09:42:07 +03:00
Code Issues Packages Projects Releases Wiki Activity

Make session decorator work right with ?session_key=

Browse Source
This commit is contained in:
Andrew Godwin 2015-09-03 00:06:58 -07:00
parent a0830f88f7
commit 29eb75326f
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
channels/decorators.py
View File

@ -47,12 +47,14 @@ def http_session(func):
@functools.wraps(func)
def inner(*args, **kwargs):
if "COOKIES" not in kwargs and "GET" not in kwargs:
print kwargs
raise ValueError("No COOKIES or GET sent to consumer; this decorator can only be used on messages containing at least one.")
# Make sure there's a session key
session_key = None
if "GET" in kwargs:
session_key = kwargs['GET'].get("session_key")
try:
session_key = kwargs['GET'].get("session_key", [])[0]
except IndexError:
pass
if "COOKIES" in kwargs and session_key is None:
session_key = kwargs['COOKIES'].get(settings.SESSION_COOKIE_NAME)
# Make a session storage