Exemplo n.º 1
0
 def testByteString_turnsIntoUnicodeString(self):
   self.assertEqual(json_util.Cleanse(b'\xc2\xa3'), u'\u00a3')  # is # sterling
Exemplo n.º 2
0
 def testSet_turnsIntoSortedList(self):
   self.assertEqual(json_util.Cleanse(set(['b', 'a'])), ['a', 'b'])
Exemplo n.º 3
0
 def _assertWrapsAs(self, to_wrap, expected):
   """Asserts that |to_wrap| becomes |expected| when wrapped."""
   actual = json_util.Cleanse(to_wrap)
   for a, e in zip(actual, expected):
     self.assertEqual(e, a)
Exemplo n.º 4
0
 def testTuple_turnsIntoList(self):
   self.assertEqual(json_util.Cleanse(('a', 'b')), ['a', 'b'])
Exemplo n.º 5
0
def Respond(request,
            content,
            content_type,
            code=200,
            expires=0,
            content_encoding=None,
            encoding='utf-8'):
    """Construct a werkzeug Response.

  Responses are transmitted to the browser with compression if: a) the browser
  supports it; b) it's sane to compress the content_type in question; and c)
  the content isn't already compressed, as indicated by the content_encoding
  parameter.

  Browser and proxy caching is completely disabled by default. If the expires
  parameter is greater than zero then the response will be able to be cached by
  the browser for that many seconds; however, proxies are still forbidden from
  caching so that developers can bypass the cache with Ctrl+Shift+R.

  For textual content that isn't JSON, the encoding parameter is used as the
  transmission charset which is automatically appended to the Content-Type
  header. That is unless of course the content_type parameter contains a
  charset parameter. If the two disagree, the characters in content will be
  transcoded to the latter.

  If content_type declares a JSON media type, then content MAY be a dict, list,
  tuple, or set, in which case this function has an implicit composition with
  json_util.Cleanse and json.dumps. The encoding parameter is used to decode
  byte strings within the JSON object; therefore transmitting binary data
  within JSON is not permitted. JSON is transmitted as ASCII unless the
  content_type parameter explicitly defines a charset parameter, in which case
  the serialized JSON bytes will use that instead of escape sequences.

  Args:
    request: A werkzeug Request object. Used mostly to check the
      Accept-Encoding header.
    content: Payload data as byte string, unicode string, or maybe JSON.
    content_type: Media type and optionally an output charset.
    code: Numeric HTTP status code to use.
    expires: Second duration for browser caching.
    content_encoding: Encoding if content is already encoded, e.g. 'gzip'.
    encoding: Input charset if content parameter has byte strings.

  Returns:
    A werkzeug Response object (a WSGI application).
  """

    mimetype = _EXTRACT_MIMETYPE_PATTERN.search(content_type).group(0)
    charset_match = _EXTRACT_CHARSET_PATTERN.search(content_type)
    charset = charset_match.group(1) if charset_match else encoding
    textual = charset_match or mimetype in _TEXTUAL_MIMETYPES
    if mimetype in _JSON_MIMETYPES and (isinstance(content, dict)
                                        or isinstance(content, list)
                                        or isinstance(content, set)
                                        or isinstance(content, tuple)):
        content = json.dumps(json_util.Cleanse(content, encoding),
                             ensure_ascii=not charset_match)
    if charset != encoding:
        content = tf.compat.as_text(content, encoding)
    content = tf.compat.as_bytes(content, charset)
    if textual and not charset_match and mimetype not in _JSON_MIMETYPES:
        content_type += '; charset=' + charset
    if (not content_encoding and textual and _ALLOWS_GZIP_PATTERN.search(
            request.headers.get('Accept-Encoding', ''))):
        out = six.BytesIO()
        f = gzip.GzipFile(fileobj=out, mode='wb', compresslevel=3)
        f.write(content)
        f.close()
        content = out.getvalue()
        content_encoding = 'gzip'
    if request.method == 'HEAD':
        content = ''
    headers = []

    headers.append(('Content-Length', str(len(content))))
    if content_encoding:
        headers.append(('Content-Encoding', content_encoding))
    if expires > 0:
        e = wsgiref.handlers.format_date_time(time.time() + float(expires))
        headers.append(('Expires', e))
        headers.append(('Cache-Control', 'private, max-age=%d' % expires))
    else:
        headers.append(('Expires', '0'))
        headers.append(('Cache-Control', 'no-cache, must-revalidate'))

    return wrappers.Response(response=content,
                             status=code,
                             headers=headers,
                             content_type=content_type)