Behdad Esfahbod | 503179f | 2013-11-27 14:36:57 -0500 | [diff] [blame] | 1 | """Python 2/3 compat layer.""" |
| 2 | |
Behdad Esfahbod | 1ae2959 | 2014-01-14 15:07:50 +0800 | [diff] [blame^] | 3 | from __future__ import print_function, division, absolute_import |
Behdad Esfahbod | 30e691e | 2013-11-27 17:27:45 -0500 | [diff] [blame] | 4 | |
Behdad Esfahbod | 503179f | 2013-11-27 14:36:57 -0500 | [diff] [blame] | 5 | try: |
| 6 | basestring |
| 7 | except NameError: |
| 8 | basestring = str |
| 9 | |
| 10 | try: |
| 11 | unicode |
| 12 | except NameError: |
| 13 | unicode = str |
| 14 | |
| 15 | try: |
| 16 | unichr |
| 17 | bytechr = chr |
Behdad Esfahbod | 319c5fd | 2013-11-27 18:13:48 -0500 | [diff] [blame] | 18 | byteord = ord |
Behdad Esfahbod | 503179f | 2013-11-27 14:36:57 -0500 | [diff] [blame] | 19 | except: |
| 20 | unichr = chr |
| 21 | def bytechr(n): |
| 22 | return bytes([n]) |
Behdad Esfahbod | 319c5fd | 2013-11-27 18:13:48 -0500 | [diff] [blame] | 23 | def byteord(c): |
Behdad Esfahbod | d368939 | 2013-11-27 21:13:05 -0500 | [diff] [blame] | 24 | return c if isinstance(c, int) else ord(c) |
Behdad Esfahbod | 503179f | 2013-11-27 14:36:57 -0500 | [diff] [blame] | 25 | |
| 26 | try: |
Behdad Esfahbod | 94968f3 | 2013-12-04 04:11:06 -0500 | [diff] [blame] | 27 | from StringIO import StringIO |
Behdad Esfahbod | 503179f | 2013-11-27 14:36:57 -0500 | [diff] [blame] | 28 | except ImportError: |
Behdad Esfahbod | cc13b78 | 2013-12-04 04:12:57 -0500 | [diff] [blame] | 29 | from io import BytesIO as StringIO |
Behdad Esfahbod | ac4672e | 2013-11-27 16:44:53 -0500 | [diff] [blame] | 30 | |
Behdad Esfahbod | 18316aa | 2013-11-27 21:17:35 -0500 | [diff] [blame] | 31 | def strjoin(iterable): |
| 32 | return ''.join(iterable) |
Behdad Esfahbod | ac4672e | 2013-11-27 16:44:53 -0500 | [diff] [blame] | 33 | if str == bytes: |
| 34 | class Tag(str): |
| 35 | def tobytes(self): |
| 36 | if isinstance(self, bytes): |
| 37 | return self |
| 38 | else: |
Behdad Esfahbod | c076261 | 2013-11-28 06:46:59 -0500 | [diff] [blame] | 39 | return self.encode('latin1') |
Behdad Esfahbod | 5cf4008 | 2013-11-27 19:51:59 -0500 | [diff] [blame] | 40 | |
Behdad Esfahbod | 18316aa | 2013-11-27 21:17:35 -0500 | [diff] [blame] | 41 | def tostr(s, encoding='ascii'): |
Behdad Esfahbod | 5cf4008 | 2013-11-27 19:51:59 -0500 | [diff] [blame] | 42 | if not isinstance(s, str): |
Behdad Esfahbod | 18316aa | 2013-11-27 21:17:35 -0500 | [diff] [blame] | 43 | return s.encode(encoding) |
Behdad Esfahbod | 5cf4008 | 2013-11-27 19:51:59 -0500 | [diff] [blame] | 44 | else: |
| 45 | return s |
| 46 | tobytes = tostr |
Behdad Esfahbod | 821572c | 2013-11-27 21:09:03 -0500 | [diff] [blame] | 47 | |
Behdad Esfahbod | 18316aa | 2013-11-27 21:17:35 -0500 | [diff] [blame] | 48 | bytesjoin = strjoin |
Behdad Esfahbod | ac4672e | 2013-11-27 16:44:53 -0500 | [diff] [blame] | 49 | else: |
| 50 | class Tag(str): |
| 51 | |
| 52 | @staticmethod |
| 53 | def transcode(blob): |
| 54 | if not isinstance(blob, str): |
| 55 | blob = blob.decode('latin-1') |
| 56 | return blob |
| 57 | |
| 58 | def __new__(self, content): |
| 59 | return str.__new__(self, self.transcode(content)) |
Behdad Esfahbod | 8ea6439 | 2013-12-06 22:25:48 -0500 | [diff] [blame] | 60 | def __ne__(self, other): |
| 61 | return not self.__eq__(other) |
Behdad Esfahbod | ac4672e | 2013-11-27 16:44:53 -0500 | [diff] [blame] | 62 | def __eq__(self, other): |
| 63 | return str.__eq__(self, self.transcode(other)) |
| 64 | |
| 65 | def __hash__(self): |
| 66 | return str.__hash__(self) |
| 67 | |
| 68 | def tobytes(self): |
| 69 | return self.encode('latin-1') |
Behdad Esfahbod | 5cf4008 | 2013-11-27 19:51:59 -0500 | [diff] [blame] | 70 | |
Behdad Esfahbod | 18316aa | 2013-11-27 21:17:35 -0500 | [diff] [blame] | 71 | def tostr(s, encoding='ascii'): |
Behdad Esfahbod | 5cf4008 | 2013-11-27 19:51:59 -0500 | [diff] [blame] | 72 | if not isinstance(s, str): |
Behdad Esfahbod | 18316aa | 2013-11-27 21:17:35 -0500 | [diff] [blame] | 73 | return s.decode(encoding) |
Behdad Esfahbod | 5cf4008 | 2013-11-27 19:51:59 -0500 | [diff] [blame] | 74 | else: |
| 75 | return s |
Behdad Esfahbod | 18316aa | 2013-11-27 21:17:35 -0500 | [diff] [blame] | 76 | def tobytes(s, encoding='ascii'): |
Behdad Esfahbod | 5cf4008 | 2013-11-27 19:51:59 -0500 | [diff] [blame] | 77 | if not isinstance(s, bytes): |
Behdad Esfahbod | 18316aa | 2013-11-27 21:17:35 -0500 | [diff] [blame] | 78 | return s.encode(encoding) |
Behdad Esfahbod | 5cf4008 | 2013-11-27 19:51:59 -0500 | [diff] [blame] | 79 | else: |
| 80 | return s |
Behdad Esfahbod | 821572c | 2013-11-27 21:09:03 -0500 | [diff] [blame] | 81 | |
| 82 | def bytesjoin(iterable): |
Denis Jacquerye | db08ee2 | 2013-11-29 14:11:19 +0100 | [diff] [blame] | 83 | return b''.join(tobytes(item) for item in iterable) |