Added py3 variable to _util

Tests/check_icns_dos.py

@@ -2,7 +2,7 @@
 # Run from anywhere that PIL is importable.
 
 from PIL import Image
-from helper import py3
+from PIL._util import py3
 from io import BytesIO
 
 if py3:

Tests/check_j2k_dos.py

@@ -2,7 +2,7 @@
 # Run from anywhere that PIL is importable.
 
 from PIL import Image
-from helper import py3
+from PIL._util import py3
 from io import BytesIO
 
 if py3:

Tests/helper.py

@@ -8,6 +8,7 @@ import os
 import unittest
 
 from PIL import Image, ImageMath
+from PIL._util import py3
 
 import logging
 logger = logging.getLogger(__name__)
@@ -265,8 +266,6 @@ class PillowLeakTestCase(PillowTestCase):
 
 # helpers
 
-py3 = sys.version_info.major >= 3
-
 if not py3:
     # Remove DeprecationWarning in Python 3
     PillowTestCase.assertRaisesRegex = PillowTestCase.assertRaisesRegexp

Tests/test_file_eps.py

@@ -1,6 +1,7 @@
-from helper import unittest, PillowTestCase, hopper, py3
+from helper import unittest, PillowTestCase, hopper
 
 from PIL import Image, EpsImagePlugin
+from PIL._util import py3
 import io
 
 # Our two EPS test files (they are identical except for their bounding boxes)

Tests/test_file_libtiff.py

@@ -1,6 +1,7 @@
 from __future__ import print_function
-from helper import unittest, PillowTestCase, hopper, py3
+from helper import unittest, PillowTestCase, hopper
 from PIL import features
+from PIL._util import py3
 
 from ctypes import c_float
 import io

Tests/test_file_png.py

@@ -1,5 +1,6 @@
-from helper import unittest, PillowTestCase, PillowLeakTestCase, hopper, py3
+from helper import unittest, PillowTestCase, PillowLeakTestCase, hopper
 from PIL import Image, ImageFile, PngImagePlugin
+from PIL._util import py3
 
 from io import BytesIO
 import zlib

Tests/test_file_tiff.py

@@ -3,9 +3,10 @@ from io import BytesIO
 import struct
 import sys
 
-from helper import unittest, PillowTestCase, hopper, py3
+from helper import unittest, PillowTestCase, hopper
 
 from PIL import Image, TiffImagePlugin
+from PIL._util import py3
 from PIL.TiffImagePlugin import X_RESOLUTION, Y_RESOLUTION, RESOLUTION_UNIT
 
 logger = logging.getLogger(__name__)

Tests/test_font_pcf.py

@@ -2,7 +2,7 @@ from helper import unittest, PillowTestCase
 
 from PIL import Image, FontFile, PcfFontFile
 from PIL import ImageFont, ImageDraw
-from helper import py3
+from PIL._util import py3
 
 codecs = dir(Image.core)
 

Tests/test_format_hsv.py

@@ -1,6 +1,7 @@
-from helper import unittest, PillowTestCase, hopper, py3
+from helper import unittest, PillowTestCase, hopper
 
 from PIL import Image
+from PIL._util import py3
 
 import colorsys
 import itertools

Tests/test_image.py

@@ -1,6 +1,7 @@
-from helper import unittest, PillowTestCase, hopper, py3
+from helper import unittest, PillowTestCase, hopper
 
 from PIL import Image
+from PIL._util import py3
 import os
 
 

Tests/test_image_getim.py

@@ -1,4 +1,5 @@
-from helper import unittest, PillowTestCase, hopper, py3
+from helper import unittest, PillowTestCase, hopper
+from PIL._util import py3
 
 
 class TestImageGetIm(PillowTestCase):

Tests/test_imagepath.py

@@ -1,6 +1,7 @@
-from helper import unittest, PillowTestCase, py3
+from helper import unittest, PillowTestCase
 
 from PIL import ImagePath, Image
+from PIL._util import py3
 
 import array
 import struct

Tests/test_imagetk.py

@@ -1,5 +1,6 @@
-from helper import unittest, PillowTestCase, hopper, py3
+from helper import unittest, PillowTestCase, hopper
 from PIL import Image
+from PIL._util import py3
 
 
 try:

src/PIL/EpsImagePlugin.py

@@ -26,6 +26,7 @@ import os
 import sys
 from . import Image, ImageFile
 from ._binary import i32le as i32
+from ._util import py3
 
 __version__ = "0.5"
 
@@ -206,12 +207,12 @@ class EpsImageFile(ImageFile.ImageFile):
         # Rewrap the open file pointer in something that will
         # convert line endings and decode to latin-1.
         try:
-            if bytes is str:
-                # Python2, no encoding conversion necessary
-                fp = open(self.fp.name, "Ur")
-            else:
+            if py3:
                 # Python3, can use bare open command.
                 fp = open(self.fp.name, "Ur", encoding='latin-1')
+            else:
+                # Python2, no encoding conversion necessary
+                fp = open(self.fp.name, "Ur")
         except:
             # Expect this for bytesio/stringio
             fp = PSFile(self.fp)

src/PIL/Image.py

@@ -25,6 +25,7 @@
 #
 
 from . import VERSION, PILLOW_VERSION, _plugins
+from ._util import py3
 
 import logging
 import warnings
@@ -1260,10 +1261,10 @@ class Image(object):
 
         self.load()
         try:
-            if bytes is str:
-                return [i8(c) for c in self.im.getpalette()]
-            else:
+            if py3:
                 return list(self.im.getpalette())
+            else:
+                return [i8(c) for c in self.im.getpalette()]
         except ValueError:
             return None  # no palette
 
@@ -1586,10 +1587,10 @@ class Image(object):
             palette = ImagePalette.raw(data.rawmode, data.palette)
         else:
             if not isinstance(data, bytes):
-                if bytes is str:
-                    data = "".join(chr(x) for x in data)
-                else:
+                if py3:
                     data = bytes(data)
+                else:
+                    data = "".join(chr(x) for x in data)
             palette = ImagePalette.raw(rawmode, data)
         self.mode = "P"
         self.palette = palette

src/PIL/ImageFont.py

@@ -26,7 +26,7 @@
 #
 
 from . import Image
-from ._util import isDirectory, isPath
+from ._util import isDirectory, isPath, py3
 import os
 import sys
 
@@ -314,10 +314,10 @@ def load_path(filename):
     for directory in sys.path:
         if isDirectory(directory):
             if not isinstance(filename, str):
-                if bytes is str:
-                    filename = filename.encode("utf-8")
-                else:
+                if py3:
                     filename = filename.decode("utf-8")
+                else:
+                    filename = filename.encode("utf-8")
             try:
                 return load(os.path.join(directory, filename))
             except IOError:

src/PIL/ImageMath.py

@@ -16,6 +16,7 @@
 #
 
 from . import Image, _imagingmath
+from ._util import py3
 
 try:
     import builtins
@@ -100,7 +101,7 @@ class _Operand(object):
         # an image is "true" if it contains at least one non-zero pixel
         return self.im.getbbox() is not None
 
-    if bytes is str:
+    if not py3:
         # Provide __nonzero__ for pre-Py3k
         __nonzero__ = __bool__
         del __bool__
@@ -151,7 +152,7 @@ class _Operand(object):
     def __rpow__(self, other):
         return self.apply("pow", other, self)
 
-    if bytes is str:
+    if not py3:
         # Provide __div__ and __rdiv__ for pre-Py3k
         __div__ = __truediv__
         __rdiv__ = __rtruediv__

src/PIL/ImageQt.py

@@ -17,7 +17,7 @@
 #
 
 from . import Image
-from ._util import isPath
+from ._util import isPath, py3
 from io import BytesIO
 
 qt_is_installed = True
@@ -123,10 +123,10 @@ def _toqclass_helper(im):
     # handle filename, if given instead of image name
     if hasattr(im, "toUtf8"):
         # FIXME - is this really the best way to do this?
-        if str is bytes:
-            im = unicode(im.toUtf8(), "utf-8")
-        else:
+        if py3:
             im = str(im.toUtf8(), "utf-8")
+        else:
+            im = unicode(im.toUtf8(), "utf-8")
     if isPath(im):
         im = Image.open(im)
 

src/PIL/PSDraw.py

@@ -16,6 +16,7 @@
 #
 
 from . import EpsImagePlugin
+from ._util import py3
 import sys
 
 ##
@@ -34,7 +35,7 @@ class PSDraw(object):
         self.fp = fp
 
     def _fp_write(self, to_write):
-        if bytes is str or self.fp == sys.stdout:
+        if not py3 or self.fp == sys.stdout:
             self.fp.write(to_write)
         else:
             self.fp.write(bytes(to_write, 'UTF-8'))

src/PIL/PdfParser.py

@@ -4,6 +4,7 @@ import mmap
 import os
 import re
 import zlib
+from ._util import py3
 
 try:
     from UserDict import UserDict  # Python 2.x
@@ -11,12 +12,12 @@ except ImportError:
     UserDict = collections.UserDict  # Python 3.x
 
 
-if str == bytes:  # Python 2.x
-    def make_bytes(s):  # pragma: no cover
-        return s        # pragma: no cover
-else:  # Python 3.x
+if py3:  # Python 3.x
     def make_bytes(s):
         return s.encode("us-ascii")
+else:  # Python 2.x
+    def make_bytes(s):  # pragma: no cover
+        return s        # pragma: no cover
 
 
 # see 7.9.2.2 Text String Type on page 86 and D.3 PDFDocEncoding Character Set on page 656
@@ -72,10 +73,10 @@ PDFDocEncoding = {
 def decode_text(b):
     if b[:len(codecs.BOM_UTF16_BE)] == codecs.BOM_UTF16_BE:
         return b[len(codecs.BOM_UTF16_BE):].decode("utf_16_be")
-    elif str == bytes:  # Python 2.x
-        return u"".join(PDFDocEncoding.get(ord(byte), byte) for byte in b)
-    else:
+    elif py3:  # Python 3.x
         return "".join(PDFDocEncoding.get(byte, chr(byte)) for byte in b)
+    else:  # Python 2.x
+        return u"".join(PDFDocEncoding.get(ord(byte), byte) for byte in b)
 
 
 class PdfFormatError(RuntimeError):
@@ -214,20 +215,18 @@ class PdfName:
     allowed_chars = set(range(33,127)) - set(ord(c) for c in "#%/()<>[]{}")
 
     def __bytes__(self):
-        if str == bytes:  # Python 2.x
-            result = bytearray(b"/")
-            for b in self.name:
-                if ord(b) in self.allowed_chars:
-                    result.append(b)
-                else:
-                    result.extend(b"#%02X" % ord(b))
-        else:  # Python 3.x
-            result = bytearray(b"/")
-            for b in self.name:
+        result = bytearray(b"/")
+        for b in self.name:
+            if py3:  # Python 3.x
                 if b in self.allowed_chars:
                     result.append(b)
                 else:
                     result.extend(make_bytes("#%02X" % b))
+            else:  # Python 2.x
+                if ord(b) in self.allowed_chars:
+                    result.append(b)
+                else:
+                    result.extend(b"#%02X" % ord(b))
         return bytes(result)
 
     __str__ = __bytes__
@@ -281,7 +280,7 @@ class PdfDict(UserDict):
         out.extend(b"\n>>")
         return bytes(out)
 
-    if str == bytes:
+    if not py3:
         __str__ = __bytes__
 
 
@@ -289,13 +288,13 @@ class PdfBinary:
     def __init__(self, data):
         self.data = data
 
-    if str == bytes:  # Python 2.x
+    if py3:  # Python 3.x
+        def __bytes__(self):
+            return make_bytes("<%s>" % "".join("%02X" % b for b in self.data))
+    else:  # Python 2.x
         def __str__(self):
             return "<%s>" % "".join("%02X" % ord(b) for b in self.data)
 
-    else:  # Python 3.x
-        def __bytes__(self):
-            return make_bytes("<%s>" % "".join("%02X" % b for b in self.data))
 
 
 class PdfStream:
@@ -333,7 +332,7 @@ def pdf_repr(x):
         return bytes(PdfDict(x))
     elif isinstance(x, list):
         return bytes(PdfArray(x))
-    elif (str == bytes and isinstance(x, unicode)) or (str != bytes and isinstance(x, str)):
+    elif (py3 and isinstance(x, str)) or (not py3 and isinstance(x, unicode)):
         return pdf_repr(encode_text(x))
     elif isinstance(x, bytes):
         return b"(" + x.replace(b"\\", b"\\\\").replace(b"(", b"\\(").replace(b")", b"\\)") + b")"  # XXX escape more chars? handle binary garbage

src/PIL/PngImagePlugin.py

@@ -38,6 +38,7 @@ import struct
 
 from . import Image, ImageFile, ImagePalette
 from ._binary import i8, i16be as i16, i32be as i32, o16be as o16, o32be as o32
+from ._util import py3
 
 __version__ = "0.9"
 
@@ -437,7 +438,7 @@ class PngStream(ChunkStream):
             k = s
             v = b""
         if k:
-            if bytes is not str:
+            if py3:
                 k = k.decode('latin-1', 'strict')
                 v = v.decode('latin-1', 'replace')
 
@@ -473,7 +474,7 @@ class PngStream(ChunkStream):
             v = b""
 
         if k:
-            if bytes is not str:
+            if py3:
                 k = k.decode('latin-1', 'strict')
                 v = v.decode('latin-1', 'replace')
 
@@ -510,7 +511,7 @@ class PngStream(ChunkStream):
                     return s
             else:
                 return s
-        if bytes is not str:
+        if py3:
             try:
                 k = k.decode("latin-1", "strict")
                 lang = lang.decode("utf-8", "strict")

src/PIL/SgiImagePlugin.py

@@ -24,6 +24,7 @@
 
 from . import Image, ImageFile
 from ._binary import i8, o8, i16be as i16
+from ._util import py3
 import struct
 import os
 
@@ -165,7 +166,7 @@ def _save(im, fp, filename):
     pinmax = 255
     # Image name (79 characters max, truncated below in write)
     imgName = os.path.splitext(os.path.basename(filename))[0]
-    if str is not bytes:
+    if py3:
         imgName = imgName.encode('ascii', 'ignore')
     # Standard representation of pixel in the file
     colormap = 0

src/PIL/TiffImagePlugin.py

@@ -43,6 +43,7 @@ from __future__ import division, print_function
 
 from . import Image, ImageFile, ImagePalette, TiffTags
 from ._binary import i8, o8
+from ._util import py3
 
 import collections
 from fractions import Fraction
@@ -519,7 +520,7 @@ class ImageFileDirectory_v2(collections.MutableMapping):
     def __contains__(self, tag):
         return tag in self._tags_v2 or tag in self._tagdata
 
-    if bytes is str:
+    if not py3:
         def has_key(self, tag):
             return tag in self
 
@@ -528,7 +529,7 @@ class ImageFileDirectory_v2(collections.MutableMapping):
 
     def _setitem(self, tag, value, legacy_api):
         basetypes = (Number, bytes, str)
-        if bytes is str:
+        if not py3:
             basetypes += unicode,
 
         info = TiffTags.lookup(tag)
@@ -549,14 +550,14 @@ class ImageFileDirectory_v2(collections.MutableMapping):
                 elif all(isinstance(v, float) for v in values):
                     self.tagtype[tag] = 12
                 else:
-                    if bytes is str:
-                        # Never treat data as binary by default on Python 2.
-                        self.tagtype[tag] = 2
-                    else:
+                    if py3:
                         if all(isinstance(v, str) for v in values):
                             self.tagtype[tag] = 2
+                    else:
+                        # Never treat data as binary by default on Python 2.
+                        self.tagtype[tag] = 2
 
-        if self.tagtype[tag] == 7 and bytes is not str:
+        if self.tagtype[tag] == 7 and py3:
             values = [value.encode("ascii", 'replace') if isinstance(
                       value, str) else value]
 
@@ -1503,7 +1504,7 @@ def _save(im, fp, filename):
             if tag not in TiffTags.LIBTIFF_CORE:
                 continue
             if tag not in atts and tag not in blocklist:
-                if isinstance(value, unicode if bytes is str else str):
+                if isinstance(value, str if py3 else unicode):
                     atts[tag] = value.encode('ascii', 'replace') + b"\0"
                 elif isinstance(value, IFDRational):
                     atts[tag] = float(value)

src/PIL/WmfImagePlugin.py

@@ -23,13 +23,14 @@ from __future__ import print_function
 
 from . import Image, ImageFile
 from ._binary import i16le as word, si16le as short, i32le as dword, si32le as _long
+from ._util import py3
 
 
 __version__ = "0.2"
 
 _handler = None
 
-if str != bytes:
+if py3:
     long = int
 
 

src/PIL/_binary.py

@@ -12,19 +12,20 @@
 #
 
 from struct import unpack, pack
+from ._util import py3
 
-if bytes is str:
-    def i8(c):
-        return ord(c)
-
-    def o8(i):
-        return chr(i & 255)
-else:
+if py3:
     def i8(c):
         return c if c.__class__ is int else c[0]
 
     def o8(i):
         return bytes((i & 255,))
+else:
+    def i8(c):
+        return ord(c)
+
+    def o8(i):
+        return chr(i & 255)
 
 
 # Input, le = little endian, be = big endian

src/PIL/_util.py

@@ -1,17 +1,19 @@
-import os
+import os, sys
 
-if bytes is str:
-    def isStringType(t):
-        return isinstance(t, basestring)
+py3 = sys.version_info.major >= 3
 
-    def isPath(f):
-        return isinstance(f, basestring)
-else:
+if py3:
     def isStringType(t):
         return isinstance(t, str)
 
     def isPath(f):
         return isinstance(f, (bytes, str))
+else:
+    def isStringType(t):
+        return isinstance(t, basestring)
+
+    def isPath(f):
+        return isinstance(f, basestring)
 
 
 # Checks if an object is a string, and that it points to a directory.