Viewing file:  parsers.py (2.13 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
# 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)
|