1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
# coding: utf8
from __future__ import unicode_literals
from flask._compat import text_type
from flask_api import exceptions
from werkzeug.formparser import MultiPartParser as WerkzeugMultiPartParser
from werkzeug.formparser import default_stream_factory
from werkzeug.urls import url_decode_stream
import json
class BaseParser(object):
media_type = None
handles_file_uploads = False # If set then 'request.files' will be populated.
handles_form_data = False # If set then 'request.form' will be populated.
def parse(self, stream, media_type, **options):
msg = '`parse()` method must be implemented for class "%s"'
raise NotImplementedError(msg % self.__class__.__name__)
class JSONParser(BaseParser):
media_type = 'application/json'
def parse(self, stream, media_type, **options):
data = stream.read().decode('utf-8')
try:
return json.loads(data)
except ValueError as exc:
msg = 'JSON parse error - %s' % text_type(exc)
raise exceptions.ParseError(msg)
class MultiPartParser(BaseParser):
media_type = 'multipart/form-data'
handles_file_uploads = True
handles_form_data = True
def parse(self, stream, media_type, **options):
multipart_parser = WerkzeugMultiPartParser(default_stream_factory)
boundary = media_type.params.get('boundary')
if boundary is None:
msg = 'Multipart message missing boundary in Content-Type header'
raise exceptions.ParseError(msg)
boundary = boundary.encode('ascii')
content_length = options.get('content_length')
assert content_length is not None, 'MultiPartParser.parse() requires `content_length` argument'
try:
return multipart_parser.parse(stream, boundary, content_length)
except ValueError as exc:
msg = 'Multipart parse error - %s' % text_type(exc)
raise exceptions.ParseError(msg)
class URLEncodedParser(BaseParser):
media_type = 'application/x-www-form-urlencoded'
handles_form_data = True
def parse(self, stream, media_type, **options):
return url_decode_stream(stream)
|
