Index: third_party/google-endpoints/future/backports/email/charset.py |
diff --git a/third_party/google-endpoints/future/backports/email/charset.py b/third_party/google-endpoints/future/backports/email/charset.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..2385ce68f3351024bee03316a7dec9c72cb73ba6 |
--- /dev/null |
+++ b/third_party/google-endpoints/future/backports/email/charset.py |
@@ -0,0 +1,409 @@ |
+from __future__ import unicode_literals |
+from __future__ import division |
+from __future__ import absolute_import |
+from future.builtins import str |
+from future.builtins import next |
+ |
+# Copyright (C) 2001-2007 Python Software Foundation |
+# Author: Ben Gertzfield, Barry Warsaw |
+# Contact: email-sig@python.org |
+ |
+__all__ = [ |
+ 'Charset', |
+ 'add_alias', |
+ 'add_charset', |
+ 'add_codec', |
+ ] |
+ |
+from functools import partial |
+ |
+from future.backports import email |
+from future.backports.email import errors |
+from future.backports.email.encoders import encode_7or8bit |
+ |
+ |
+# Flags for types of header encodings |
+QP = 1 # Quoted-Printable |
+BASE64 = 2 # Base64 |
+SHORTEST = 3 # the shorter of QP and base64, but only for headers |
+ |
+# In "=?charset?q?hello_world?=", the =?, ?q?, and ?= add up to 7 |
+RFC2047_CHROME_LEN = 7 |
+ |
+DEFAULT_CHARSET = 'us-ascii' |
+UNKNOWN8BIT = 'unknown-8bit' |
+EMPTYSTRING = '' |
+ |
+ |
+# Defaults |
+CHARSETS = { |
+ # input header enc body enc output conv |
+ 'iso-8859-1': (QP, QP, None), |
+ 'iso-8859-2': (QP, QP, None), |
+ 'iso-8859-3': (QP, QP, None), |
+ 'iso-8859-4': (QP, QP, None), |
+ # iso-8859-5 is Cyrillic, and not especially used |
+ # iso-8859-6 is Arabic, also not particularly used |
+ # iso-8859-7 is Greek, QP will not make it readable |
+ # iso-8859-8 is Hebrew, QP will not make it readable |
+ 'iso-8859-9': (QP, QP, None), |
+ 'iso-8859-10': (QP, QP, None), |
+ # iso-8859-11 is Thai, QP will not make it readable |
+ 'iso-8859-13': (QP, QP, None), |
+ 'iso-8859-14': (QP, QP, None), |
+ 'iso-8859-15': (QP, QP, None), |
+ 'iso-8859-16': (QP, QP, None), |
+ 'windows-1252':(QP, QP, None), |
+ 'viscii': (QP, QP, None), |
+ 'us-ascii': (None, None, None), |
+ 'big5': (BASE64, BASE64, None), |
+ 'gb2312': (BASE64, BASE64, None), |
+ 'euc-jp': (BASE64, None, 'iso-2022-jp'), |
+ 'shift_jis': (BASE64, None, 'iso-2022-jp'), |
+ 'iso-2022-jp': (BASE64, None, None), |
+ 'koi8-r': (BASE64, BASE64, None), |
+ 'utf-8': (SHORTEST, BASE64, 'utf-8'), |
+ } |
+ |
+# Aliases for other commonly-used names for character sets. Map |
+# them to the real ones used in email. |
+ALIASES = { |
+ 'latin_1': 'iso-8859-1', |
+ 'latin-1': 'iso-8859-1', |
+ 'latin_2': 'iso-8859-2', |
+ 'latin-2': 'iso-8859-2', |
+ 'latin_3': 'iso-8859-3', |
+ 'latin-3': 'iso-8859-3', |
+ 'latin_4': 'iso-8859-4', |
+ 'latin-4': 'iso-8859-4', |
+ 'latin_5': 'iso-8859-9', |
+ 'latin-5': 'iso-8859-9', |
+ 'latin_6': 'iso-8859-10', |
+ 'latin-6': 'iso-8859-10', |
+ 'latin_7': 'iso-8859-13', |
+ 'latin-7': 'iso-8859-13', |
+ 'latin_8': 'iso-8859-14', |
+ 'latin-8': 'iso-8859-14', |
+ 'latin_9': 'iso-8859-15', |
+ 'latin-9': 'iso-8859-15', |
+ 'latin_10':'iso-8859-16', |
+ 'latin-10':'iso-8859-16', |
+ 'cp949': 'ks_c_5601-1987', |
+ 'euc_jp': 'euc-jp', |
+ 'euc_kr': 'euc-kr', |
+ 'ascii': 'us-ascii', |
+ } |
+ |
+ |
+# Map charsets to their Unicode codec strings. |
+CODEC_MAP = { |
+ 'gb2312': 'eucgb2312_cn', |
+ 'big5': 'big5_tw', |
+ # Hack: We don't want *any* conversion for stuff marked us-ascii, as all |
+ # sorts of garbage might be sent to us in the guise of 7-bit us-ascii. |
+ # Let that stuff pass through without conversion to/from Unicode. |
+ 'us-ascii': None, |
+ } |
+ |
+ |
+# Convenience functions for extending the above mappings |
+def add_charset(charset, header_enc=None, body_enc=None, output_charset=None): |
+ """Add character set properties to the global registry. |
+ |
+ charset is the input character set, and must be the canonical name of a |
+ character set. |
+ |
+ Optional header_enc and body_enc is either Charset.QP for |
+ quoted-printable, Charset.BASE64 for base64 encoding, Charset.SHORTEST for |
+ the shortest of qp or base64 encoding, or None for no encoding. SHORTEST |
+ is only valid for header_enc. It describes how message headers and |
+ message bodies in the input charset are to be encoded. Default is no |
+ encoding. |
+ |
+ Optional output_charset is the character set that the output should be |
+ in. Conversions will proceed from input charset, to Unicode, to the |
+ output charset when the method Charset.convert() is called. The default |
+ is to output in the same character set as the input. |
+ |
+ Both input_charset and output_charset must have Unicode codec entries in |
+ the module's charset-to-codec mapping; use add_codec(charset, codecname) |
+ to add codecs the module does not know about. See the codecs module's |
+ documentation for more information. |
+ """ |
+ if body_enc == SHORTEST: |
+ raise ValueError('SHORTEST not allowed for body_enc') |
+ CHARSETS[charset] = (header_enc, body_enc, output_charset) |
+ |
+ |
+def add_alias(alias, canonical): |
+ """Add a character set alias. |
+ |
+ alias is the alias name, e.g. latin-1 |
+ canonical is the character set's canonical name, e.g. iso-8859-1 |
+ """ |
+ ALIASES[alias] = canonical |
+ |
+ |
+def add_codec(charset, codecname): |
+ """Add a codec that map characters in the given charset to/from Unicode. |
+ |
+ charset is the canonical name of a character set. codecname is the name |
+ of a Python codec, as appropriate for the second argument to the unicode() |
+ built-in, or to the encode() method of a Unicode string. |
+ """ |
+ CODEC_MAP[charset] = codecname |
+ |
+ |
+# Convenience function for encoding strings, taking into account |
+# that they might be unknown-8bit (ie: have surrogate-escaped bytes) |
+def _encode(string, codec): |
+ string = str(string) |
+ if codec == UNKNOWN8BIT: |
+ return string.encode('ascii', 'surrogateescape') |
+ else: |
+ return string.encode(codec) |
+ |
+ |
+class Charset(object): |
+ """Map character sets to their email properties. |
+ |
+ This class provides information about the requirements imposed on email |
+ for a specific character set. It also provides convenience routines for |
+ converting between character sets, given the availability of the |
+ applicable codecs. Given a character set, it will do its best to provide |
+ information on how to use that character set in an email in an |
+ RFC-compliant way. |
+ |
+ Certain character sets must be encoded with quoted-printable or base64 |
+ when used in email headers or bodies. Certain character sets must be |
+ converted outright, and are not allowed in email. Instances of this |
+ module expose the following information about a character set: |
+ |
+ input_charset: The initial character set specified. Common aliases |
+ are converted to their `official' email names (e.g. latin_1 |
+ is converted to iso-8859-1). Defaults to 7-bit us-ascii. |
+ |
+ header_encoding: If the character set must be encoded before it can be |
+ used in an email header, this attribute will be set to |
+ Charset.QP (for quoted-printable), Charset.BASE64 (for |
+ base64 encoding), or Charset.SHORTEST for the shortest of |
+ QP or BASE64 encoding. Otherwise, it will be None. |
+ |
+ body_encoding: Same as header_encoding, but describes the encoding for the |
+ mail message's body, which indeed may be different than the |
+ header encoding. Charset.SHORTEST is not allowed for |
+ body_encoding. |
+ |
+ output_charset: Some character sets must be converted before they can be |
+ used in email headers or bodies. If the input_charset is |
+ one of them, this attribute will contain the name of the |
+ charset output will be converted to. Otherwise, it will |
+ be None. |
+ |
+ input_codec: The name of the Python codec used to convert the |
+ input_charset to Unicode. If no conversion codec is |
+ necessary, this attribute will be None. |
+ |
+ output_codec: The name of the Python codec used to convert Unicode |
+ to the output_charset. If no conversion codec is necessary, |
+ this attribute will have the same value as the input_codec. |
+ """ |
+ def __init__(self, input_charset=DEFAULT_CHARSET): |
+ # RFC 2046, $4.1.2 says charsets are not case sensitive. We coerce to |
+ # unicode because its .lower() is locale insensitive. If the argument |
+ # is already a unicode, we leave it at that, but ensure that the |
+ # charset is ASCII, as the standard (RFC XXX) requires. |
+ try: |
+ if isinstance(input_charset, str): |
+ input_charset.encode('ascii') |
+ else: |
+ input_charset = str(input_charset, 'ascii') |
+ except UnicodeError: |
+ raise errors.CharsetError(input_charset) |
+ input_charset = input_charset.lower() |
+ # Set the input charset after filtering through the aliases |
+ self.input_charset = ALIASES.get(input_charset, input_charset) |
+ # We can try to guess which encoding and conversion to use by the |
+ # charset_map dictionary. Try that first, but let the user override |
+ # it. |
+ henc, benc, conv = CHARSETS.get(self.input_charset, |
+ (SHORTEST, BASE64, None)) |
+ if not conv: |
+ conv = self.input_charset |
+ # Set the attributes, allowing the arguments to override the default. |
+ self.header_encoding = henc |
+ self.body_encoding = benc |
+ self.output_charset = ALIASES.get(conv, conv) |
+ # Now set the codecs. If one isn't defined for input_charset, |
+ # guess and try a Unicode codec with the same name as input_codec. |
+ self.input_codec = CODEC_MAP.get(self.input_charset, |
+ self.input_charset) |
+ self.output_codec = CODEC_MAP.get(self.output_charset, |
+ self.output_charset) |
+ |
+ def __str__(self): |
+ return self.input_charset.lower() |
+ |
+ __repr__ = __str__ |
+ |
+ def __eq__(self, other): |
+ return str(self) == str(other).lower() |
+ |
+ def __ne__(self, other): |
+ return not self.__eq__(other) |
+ |
+ def get_body_encoding(self): |
+ """Return the content-transfer-encoding used for body encoding. |
+ |
+ This is either the string `quoted-printable' or `base64' depending on |
+ the encoding used, or it is a function in which case you should call |
+ the function with a single argument, the Message object being |
+ encoded. The function should then set the Content-Transfer-Encoding |
+ header itself to whatever is appropriate. |
+ |
+ Returns "quoted-printable" if self.body_encoding is QP. |
+ Returns "base64" if self.body_encoding is BASE64. |
+ Returns conversion function otherwise. |
+ """ |
+ assert self.body_encoding != SHORTEST |
+ if self.body_encoding == QP: |
+ return 'quoted-printable' |
+ elif self.body_encoding == BASE64: |
+ return 'base64' |
+ else: |
+ return encode_7or8bit |
+ |
+ def get_output_charset(self): |
+ """Return the output character set. |
+ |
+ This is self.output_charset if that is not None, otherwise it is |
+ self.input_charset. |
+ """ |
+ return self.output_charset or self.input_charset |
+ |
+ def header_encode(self, string): |
+ """Header-encode a string by converting it first to bytes. |
+ |
+ The type of encoding (base64 or quoted-printable) will be based on |
+ this charset's `header_encoding`. |
+ |
+ :param string: A unicode string for the header. It must be possible |
+ to encode this string to bytes using the character set's |
+ output codec. |
+ :return: The encoded string, with RFC 2047 chrome. |
+ """ |
+ codec = self.output_codec or 'us-ascii' |
+ header_bytes = _encode(string, codec) |
+ # 7bit/8bit encodings return the string unchanged (modulo conversions) |
+ encoder_module = self._get_encoder(header_bytes) |
+ if encoder_module is None: |
+ return string |
+ return encoder_module.header_encode(header_bytes, codec) |
+ |
+ def header_encode_lines(self, string, maxlengths): |
+ """Header-encode a string by converting it first to bytes. |
+ |
+ This is similar to `header_encode()` except that the string is fit |
+ into maximum line lengths as given by the argument. |
+ |
+ :param string: A unicode string for the header. It must be possible |
+ to encode this string to bytes using the character set's |
+ output codec. |
+ :param maxlengths: Maximum line length iterator. Each element |
+ returned from this iterator will provide the next maximum line |
+ length. This parameter is used as an argument to built-in next() |
+ and should never be exhausted. The maximum line lengths should |
+ not count the RFC 2047 chrome. These line lengths are only a |
+ hint; the splitter does the best it can. |
+ :return: Lines of encoded strings, each with RFC 2047 chrome. |
+ """ |
+ # See which encoding we should use. |
+ codec = self.output_codec or 'us-ascii' |
+ header_bytes = _encode(string, codec) |
+ encoder_module = self._get_encoder(header_bytes) |
+ encoder = partial(encoder_module.header_encode, charset=codec) |
+ # Calculate the number of characters that the RFC 2047 chrome will |
+ # contribute to each line. |
+ charset = self.get_output_charset() |
+ extra = len(charset) + RFC2047_CHROME_LEN |
+ # Now comes the hard part. We must encode bytes but we can't split on |
+ # bytes because some character sets are variable length and each |
+ # encoded word must stand on its own. So the problem is you have to |
+ # encode to bytes to figure out this word's length, but you must split |
+ # on characters. This causes two problems: first, we don't know how |
+ # many octets a specific substring of unicode characters will get |
+ # encoded to, and second, we don't know how many ASCII characters |
+ # those octets will get encoded to. Unless we try it. Which seems |
+ # inefficient. In the interest of being correct rather than fast (and |
+ # in the hope that there will be few encoded headers in any such |
+ # message), brute force it. :( |
+ lines = [] |
+ current_line = [] |
+ maxlen = next(maxlengths) - extra |
+ for character in string: |
+ current_line.append(character) |
+ this_line = EMPTYSTRING.join(current_line) |
+ length = encoder_module.header_length(_encode(this_line, charset)) |
+ if length > maxlen: |
+ # This last character doesn't fit so pop it off. |
+ current_line.pop() |
+ # Does nothing fit on the first line? |
+ if not lines and not current_line: |
+ lines.append(None) |
+ else: |
+ separator = (' ' if lines else '') |
+ joined_line = EMPTYSTRING.join(current_line) |
+ header_bytes = _encode(joined_line, codec) |
+ lines.append(encoder(header_bytes)) |
+ current_line = [character] |
+ maxlen = next(maxlengths) - extra |
+ joined_line = EMPTYSTRING.join(current_line) |
+ header_bytes = _encode(joined_line, codec) |
+ lines.append(encoder(header_bytes)) |
+ return lines |
+ |
+ def _get_encoder(self, header_bytes): |
+ if self.header_encoding == BASE64: |
+ return email.base64mime |
+ elif self.header_encoding == QP: |
+ return email.quoprimime |
+ elif self.header_encoding == SHORTEST: |
+ len64 = email.base64mime.header_length(header_bytes) |
+ lenqp = email.quoprimime.header_length(header_bytes) |
+ if len64 < lenqp: |
+ return email.base64mime |
+ else: |
+ return email.quoprimime |
+ else: |
+ return None |
+ |
+ def body_encode(self, string): |
+ """Body-encode a string by converting it first to bytes. |
+ |
+ The type of encoding (base64 or quoted-printable) will be based on |
+ self.body_encoding. If body_encoding is None, we assume the |
+ output charset is a 7bit encoding, so re-encoding the decoded |
+ string using the ascii codec produces the correct string version |
+ of the content. |
+ """ |
+ if not string: |
+ return string |
+ if self.body_encoding is BASE64: |
+ if isinstance(string, str): |
+ string = string.encode(self.output_charset) |
+ return email.base64mime.body_encode(string) |
+ elif self.body_encoding is QP: |
+ # quopromime.body_encode takes a string, but operates on it as if |
+ # it were a list of byte codes. For a (minimal) history on why |
+ # this is so, see changeset 0cf700464177. To correctly encode a |
+ # character set, then, we must turn it into pseudo bytes via the |
+ # latin1 charset, which will encode any byte as a single code point |
+ # between 0 and 255, which is what body_encode is expecting. |
+ if isinstance(string, str): |
+ string = string.encode(self.output_charset) |
+ string = string.decode('latin1') |
+ return email.quoprimime.body_encode(string) |
+ else: |
+ if isinstance(string, str): |
+ string = string.encode(self.output_charset).decode('ascii') |
+ return string |