def test_parse_bad_content_type(self): parser = FormDataParser() assert parser.parse("", "bad-mime-type", 0) == ( "", MultiDict([]), MultiDict([]), )
def decode_multipart(stream, size, boundary): """Decode multipart data. :param stream: object that has `read()` to restore data from :param size: number of bytes to read from `stream` :param boundary: boundary string that separates multipart portions :returns: iterator object that returns `(filename, stream)` for each multipart portion """ from werkzeug.formparser import FormDataParser if hasattr(stream, 'recv'): stream = SocketFile(stream) parser = FormDataParser( silent=False, stream_factory=lambda *args: tempfile.NamedTemporaryFile()) __, __, files = parser.parse(stream, 'multipart/form-data', size, {'boundary': boundary}) for __, chunk in files.iteritems(multi=True): yield chunk.filename, chunk.stream chunk.close()
def multi_part_parser(body: bytes, headers: dict = None) -> ImmutableMultiDict: mimetype, options = get_mimetype_and_options(headers) content_length = get_content_length(headers) parser = FormDataParser() _, form, files = parser.parse(BytesIO(body), mimetype, content_length, options) return ImmutableMultiDict(list(form.items()) + list(files.items()))
def parse(self, headers: http.Headers, stream: http.RequestStream) -> ImmutableMultiDict: mimetype, options = self.get_mimetype_and_options(headers) content_length = self.get_content_length(headers) parser = FormDataParser() stream, form, files = parser.parse(stream, mimetype, content_length, options) return ImmutableMultiDict(list(form.items()) + list(files.items()))
def decode(self, bytestring, headers, **options): try: content_length = max(0, int(headers['content-length'])) except (KeyError, ValueError, TypeError): content_length = None try: mime_type, mime_options = parse_options_header(headers['content-type']) except KeyError: mime_type, mime_options = '', {} body_file = BytesIO(bytestring) parser = FormDataParser() stream, form, files = parser.parse(body_file, mime_type, content_length, mime_options) return ImmutableMultiDict(chain(form.items(multi=True), files.items(multi=True)))
def TestOneInput(data): parser = FormDataParser() parser.parse(io.BytesIO(data), "multipart/form-data", 0) parser.parse(io.BytesIO(data), "application/x-url-encoded", 0) parser.parse(io.BytesIO(data), "application/x-www-form-urlencoded", 0) multiparser = MultiPartParser(len(data)) try: multiparser.parse(io.BytesIO(data), b"", len(data)) except ValueError as e: if "Invalid form-data" in str(e): return raise e
async def get_request_data(headers: http.Headers, message: UMIMessage, channels: UMIChannels): content_type = headers.get('Content-Type') if content_type: mimetype, options = parse_options_header(content_type) else: mimetype, options = None, {} if mimetype is None: value = None elif mimetype == 'application/json': body = await get_body(message, channels) value = json.loads(body.decode('utf-8')) elif mimetype in ('multipart/form-data', 'application/x-www-form-urlencoded'): body = await get_body(message, channels) stream = io.BytesIO(body) content_length = _get_content_length(headers) parser = FormDataParser() stream, form, files = parser.parse(stream, mimetype, content_length, options) value = ImmutableMultiDict(list(form.items()) + list(files.items())) else: raise exceptions.UnsupportedMediaType() return value
def test_parse_bad_content_type(self): parser = FormDataParser() assert parser.parse('', 'bad-mime-type', 0) == \ ('', MultiDict([]), MultiDict([]))
def test_parse_bad_content_type(self): parser = FormDataParser() assert parser.parse('', 'bad-mime-type', 0) == \ ('', MultiDict([]), MultiDict([]))