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

Correct body/POST handling

Browse Source
This commit is contained in:
Andrew Godwin 2016-01-02 18:10:25 -08:00
parent 3d5c399a41
commit d452486524
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
channels/handler.py
View File

@ -46,9 +46,9 @@ class AsgiRequest(http.HttpRequest):
self.META['SERVER_PORT'] = self.message['server'][1] self.META['SERVER_PORT'] = self.message['server'][1]
# Headers go into META # Headers go into META
for name, value in self.message.get('headers', {}).items(): for name, value in self.message.get('headers', {}).items():
if name == "content_length": if name == "content-length":
corrected_name = "CONTENT_LENGTH" corrected_name = "CONTENT_LENGTH"
elif name == "content_type": elif name == "content-type":
corrected_name = "CONTENT_TYPE" corrected_name = "CONTENT_TYPE"
else: else:
corrected_name = 'HTTP_%s' % name.upper().replace("-", "_") corrected_name = 'HTTP_%s' % name.upper().replace("-", "_")
@ -59,8 +59,22 @@ class AsgiRequest(http.HttpRequest):
self._content_length = int(self.META['CONTENT_LENGTH']) self._content_length = int(self.META['CONTENT_LENGTH'])
except (ValueError, TypeError): except (ValueError, TypeError):
pass pass
# TODO: body handling # Body handling
self._body = "" self._body = message.get("body", "")
if message.get("body_channel", None):
while True:
# Get the next chunk from the request body channel
chunk = None
while chunk is None:
_, chunk = message.channel_layer.receive_many(
[message['body_channel']],
block=True,
)
# Add content to body
self._body += chunk.get("content", "")
# Exit loop if this was the last
if not chunk.get("more_content", False):
break
# Other bits # Other bits
self.resolver_match = None self.resolver_match = None
@ -73,6 +87,7 @@ class AsgiRequest(http.HttpRequest):
def _get_post(self): def _get_post(self):
if not hasattr(self, '_post'): if not hasattr(self, '_post'):
self._read_started = False
self._load_post_and_files() self._load_post_and_files()
return self._post return self._post