Remove python 2 support (#380)

* Replace mock with unittest.mock
* Replace unittest2 with unitttest
* Remove code that checks for python 2
* Remove p3compat module
* Remove __future__ entries

okonomiyaki/__init__.py

@@ -1,5 +1,3 @@
-from __future__ import absolute_import
-
 #  Copyright (c) 2013 by Enthought, Inc.
 #  All rights reserved.
 try:

okonomiyaki/_cli/tests/test_cli.py

@@ -3,18 +3,14 @@
 import sys
 import tempfile
 import textwrap
+import unittest
 
 import testfixtures
 
 from okonomiyaki.file_formats import EggMetadata, PackageInfo
 from okonomiyaki.utils.test_data import NOSE_1_3_4_RH5_X86_64
 from okonomiyaki._cli import main
 
-if sys.version_info < (2, 7):
-    import unittest2 as unittest
-else:
-    import unittest
-
 
 class TestMain(unittest.TestCase):
     maxDiff = None

okonomiyaki/file_formats/_blacklist/__init__.py

@@ -1,5 +1,3 @@
-from __future__ import absolute_import
-
 from .pkg_info_data import (
     EGG_PKG_INFO_BLACK_LIST, may_be_in_pkg_info_blacklist
 )

okonomiyaki/file_formats/_egg_info.py

@@ -1,7 +1,7 @@
+import io
 import posixpath
 import re
 
-import six
 import zipfile2
 
 from attr import attr, attributes
@@ -14,8 +14,7 @@
 from okonomiyaki.platforms import (
     EPDPlatform, PlatformABI, PythonABI, PythonImplementation)
 from okonomiyaki.utils import (
-    compute_sha256, decode_if_needed, encode_if_needed, parse_assignments)
-from okonomiyaki.utils.py3compat import StringIO, string_types
+    compute_sha256, decode_if_needed, parse_assignments)
 from okonomiyaki.versions import EnpkgVersion, MetadataVersion
 from .legacy import (
     _guess_abi_tag, _guess_platform_abi, _guess_platform_tag, _guess_python_tag)
@@ -119,7 +118,7 @@ def split_egg_name(s):
 
 
 def parse_rawspec(spec_string):
-    spec = parse_assignments(StringIO(spec_string.replace('\r', '')))
+    spec = parse_assignments(io.StringIO(spec_string.replace('\r', '')))
 
     metadata_version_string = spec.get(_TAG_METADATA_VERSION)
     if metadata_version_string is not None:
@@ -144,7 +143,7 @@ def parse_rawspec(spec_string):
     for k, v in res.items():
         # Some values are not string-like, so filter on the type that needs
         # conversion
-        if isinstance(v, six.binary_type):
+        if isinstance(v, bytes):
             res[k] = decode_if_needed(v)
 
     res[_TAG_PACKAGES] = [decode_if_needed(v) for v in res[_TAG_PACKAGES]]
@@ -182,7 +181,7 @@ def text_attr(**kw):
     for k in ("validator", ):
         if k in kw:
             raise ValueError("Cannot pass '{0}' argument".format(k))
-    return attr(validator=instance_of(six.text_type), **kw)
+    return attr(validator=instance_of(str), **kw)
 
 
 def text_or_none_attr(**kw):
@@ -192,10 +191,9 @@ def text_or_none_attr(**kw):
     for k in ("validator", ):
         if k in kw:
             raise ValueError("Cannot pass '{0}' argument".format(k))
-    return attr(validator=optional(instance_of(six.text_type)), **kw)
+    return attr(validator=optional(instance_of(str)), **kw)
 
 
-@six.python_2_unicode_compatible
 @attributes(frozen=True)
 class Requirement(object):
     """
@@ -449,7 +447,7 @@ def _from_data(cls, data, epd_platform):
     @classmethod
     def from_egg(cls, path_or_file):
         sha256 = None
-        if isinstance(path_or_file, string_types):
+        if isinstance(path_or_file, str):
             if (
                 may_be_in_platform_blacklist(path_or_file)
                 or may_be_in_python_tag_blacklist(path_or_file)
@@ -481,7 +479,7 @@ def _create_spec_depend(zp):
                 )
                 return cls._from_data(data, epd_platform)
 
-        if isinstance(path_or_file, string_types):
+        if isinstance(path_or_file, str):
             with zipfile2.ZipFile(path_or_file) as zp:
                 return _create_spec_depend(zp)
         else:
@@ -576,20 +574,12 @@ def to_string(self):
         template = _METADATA_TEMPLATES.get(self.metadata_version, None)
         data = self._to_dict()
 
-        if six.PY2:
-            # Hack to avoid the 'u' prefix to appear in the spec/depend entries
-            for k, v in data.items():
-                data[k] = encode_if_needed(v)
-
         # This is just to ensure the exact same string as the produced by the
         # legacy buildsystem
         if len(self.packages) == 0:
             data[_TAG_PACKAGES] = "[]"
         else:
-            if six.PY2:
-                packages = [decode_if_needed(p) for p in self.packages]
-            else:
-                packages = self.packages
+            packages = self.packages
             data[_TAG_PACKAGES] = (
                 u"[\n{0}\n]".format(
                     "\n".join("  '{0}',".format(p) for p in packages)
@@ -712,7 +702,7 @@ def from_egg(cls, path_or_file, strict=True):
             errors, at the risk of data loss.
         """
         sha256 = None
-        if isinstance(path_or_file, string_types):
+        if isinstance(path_or_file, str):
             if cls._may_be_in_blacklist(path_or_file):
                 sha256 = compute_sha256(path_or_file)
         else:
@@ -772,7 +762,7 @@ def _compute_all_metadata(fp):
 
             return summary, pkg_info_string, spec_depend
 
-        if isinstance(path_or_file, string_types):
+        if isinstance(path_or_file, str):
             with zipfile2.ZipFile(path_or_file) as zp:
                 summary, pkg_info_string, spec_depend = _compute_all_metadata(zp)
         else:
@@ -866,12 +856,12 @@ def __init__(self, raw_name, version, platform, python, abi_tag,
         self.platform = platform
         """ The platform, as a Platform instance."""
 
-        if isinstance(python, string_types):
+        if isinstance(python, str):
             python = PythonImplementation.from_string(python)
         self.python = python
         """ The python implementation."""
 
-        if abi_tag is not None and isinstance(abi_tag, six.string_types):
+        if abi_tag is not None and isinstance(abi_tag, str):
             abi_tag = PythonABI(abi_tag)
 
         self.abi = abi_tag
@@ -880,7 +870,7 @@ def __init__(self, raw_name, version, platform, python, abi_tag,
 
         if (
                 platform_abi is not None
-                and isinstance(platform_abi, six.string_types)):
+                and isinstance(platform_abi, str)):
             platform_abi = PlatformABI(platform_abi)
         self.platform_abi = platform_abi
 
@@ -950,7 +940,7 @@ def name(self):
 
     @property
     def pkg_info(self):
-        if isinstance(self._pkg_info, six.string_types):
+        if isinstance(self._pkg_info, str):
             self._pkg_info = PackageInfo.from_string(self._pkg_info)
         return self._pkg_info
 
@@ -995,7 +985,7 @@ def spec_depend_string(self):
 
     @property
     def upstream_version(self):
-        return six.text_type(self.version.upstream)
+        return str(self.version.upstream)
 
     @property
     def _python(self):
@@ -1020,8 +1010,8 @@ def _spec_depend(self):
             "abi_tag": self.abi_tag,
             "platform_tag": self.platform_tag,
             "platform_abi": self.platform_abi_tag,
-            "packages": [six.text_type(p) for p in self.runtime_dependencies],
-            "metadata_version": six.text_type(self.metadata_version),
+            "packages": [str(p) for p in self.runtime_dependencies],
+            "metadata_version": str(self.metadata_version),
         }
         return LegacySpecDepend._from_data(args, self.platform)
 
@@ -1049,19 +1039,19 @@ def to_json_dict(self):
         if self.platform is None:
             epd_platform = None
         else:
-            epd_platform = six.text_type(self.platform)
+            epd_platform = str(self.platform)
 
         return {
-            _JSON_METADATA_VERSION: six.text_type(self.metadata_version),
+            _JSON_METADATA_VERSION: str(self.metadata_version),
             _JSON__RAW_NAME: self._raw_name,
-            _JSON_VERSION: six.text_type(self.version),
+            _JSON_VERSION: str(self.version),
             _JSON_EPD_PLATFORM: epd_platform,
             _JSON_PYTHON_TAG: self.python_tag,
             _JSON_ABI_TAG: self.abi_tag,
             _JSON_PLATFORM_TAG: self.platform_tag,
             _JSON_PLATFORM_ABI_TAG: self.platform_abi_tag,
             _JSON_RUNTIME_DEPENDENCIES: [
-                six.text_type(p) for p in self.runtime_dependencies],
+                str(p) for p in self.runtime_dependencies],
             _JSON_SUMMARY: self.summary}
 
     # Protocol implementations

okonomiyaki/file_formats/_package_info.py

@@ -4,14 +4,15 @@
 We support 1.0, 1.1, 1.2 and 2.0.
 """
 import contextlib
+import io
+import os
 import os.path
 import warnings
 
 import zipfile2
 
-from okonomiyaki.utils import py3compat, compute_sha256
+from okonomiyaki.utils import compute_sha256
 from okonomiyaki.errors import OkonomiyakiError
-
 from ._blacklist import EGG_PKG_INFO_BLACK_LIST, may_be_in_pkg_info_blacklist
 from ._wheel_info import WheelInfo
 
@@ -97,7 +98,7 @@ def from_path(cls, path, strict=True):
 
     @classmethod
     def from_wheel(cls, path_or_file, strict=True):
-        if isinstance(path_or_file, py3compat.string_types):
+        if isinstance(path_or_file, str):
             wheel_info = WheelInfo.from_path(path_or_file)
 
             with zipfile2.ZipFile(path_or_file) as fp:
@@ -124,7 +125,7 @@ def from_egg(cls, path_or_file, strict=True):
             understood as a zipfile-like object.
         """
         sha256 = None
-        if isinstance(path_or_file, py3compat.string_types):
+        if isinstance(path_or_file, str):
             if may_be_in_pkg_info_blacklist(path_or_file):
                 sha256 = compute_sha256(path_or_file)
         else:
@@ -134,7 +135,7 @@ def from_egg(cls, path_or_file, strict=True):
 
     @classmethod
     def _from_egg(cls, path_or_file, sha256, strict=True):
-        if isinstance(path_or_file, py3compat.string_types):
+        if isinstance(path_or_file, str):
             with zipfile2.ZipFile(path_or_file) as fp:
                 data = _read_pkg_info(fp)
         else:
@@ -149,9 +150,9 @@ def _from_egg(cls, path_or_file, sha256, strict=True):
 
     @classmethod
     def from_string(cls, s):
-        if not isinstance(s, py3compat.text_type):
+        if not isinstance(s, str):
             raise ValueError("Expected text value, got {0!r}".format(type(s)))
-        fp = py3compat.StringIO(s)
+        fp = io.StringIO(s)
         msg = _parse(fp)
 
         kw = {}
@@ -248,7 +249,7 @@ def __init__(self, metadata_version, name, version, platforms=None,
         self.provides_extra = provides_extra or ()
 
     def to_string(self, metadata_version_info=MAX_SUPPORTED_VERSION):
-        s = py3compat.StringIO()
+        s = io.StringIO()
         self._write_field(s, 'Metadata-Version', self.metadata_version)
         self._write_field(s, 'Name', self.name)
         self._write_field(s, 'Version', self.version)

okonomiyaki/file_formats/setuptools_egg.py

@@ -1,15 +1,11 @@
 import os.path
 import re
 import sys
-try:
-    import sysconfig
-except ImportError:  # Python 2.6 support
-    sysconfig = None
+import sysconfig
 import warnings
 
 from okonomiyaki.errors import OkonomiyakiError
 from okonomiyaki.platforms import PythonImplementation
-from okonomiyaki.utils import py3compat
 from ._egg_info import _guess_python_tag
 from ._package_info import PackageInfo
 
@@ -90,14 +86,11 @@ def _guess_abi_from_python(python):
 
 
 def _guess_abi_from_running_python():
-    if sysconfig is None:
+    try:
+        soabi = sysconfig.get_config_var('SOABI')
+    except IOError as e:  # pip issue #1074
+        warnings.warn("{0}".format(e), RuntimeWarning)
         soabi = None
-    else:
-        try:
-            soabi = sysconfig.get_config_var('SOABI')
-        except IOError as e:  # pip issue #1074
-            warnings.warn("{0}".format(e), RuntimeWarning)
-            soabi = None
 
     if soabi and soabi.startswith('cpython-'):
         return 'cp' + soabi.split('-', 2)[1]
@@ -169,7 +162,7 @@ def __init__(self, name, version, platform, python, abi_tag, pkg_info):
         self.version = version
 
         self.platform = platform
-        if isinstance(python, py3compat.string_types):
+        if isinstance(python, str):
             python = PythonImplementation.from_string(python)
         self.python = python
         self.abi_tag = abi_tag

okonomiyaki/file_formats/tests/test__egg_info.py

@@ -1,36 +1,28 @@
 import os
 import os.path as op
 import shutil
-import sys
 import tempfile
 import textwrap
 import zipfile2
-
-import mock
-
-if sys.version_info < (2, 7):  # noqa
-    import unittest2 as unittest
-else:
-    import unittest
-
-from okonomiyaki.errors import (
-    InvalidEggName, InvalidMetadataField, MissingMetadata, UnsupportedMetadata,
-    InvalidRequirementStringHyphen)
-from okonomiyaki.utils import tempdir
-from okonomiyaki.utils.test_data import (
-    MKL_10_3_RH5_X86_64, NOSE_1_3_4_RH6_X86_64, NOSE_1_3_4_RH5_X86_64)
-from okonomiyaki.platforms import EPDPlatform, PlatformABI
-from okonomiyaki.versions import EnpkgVersion, MetadataVersion
+import unittest
+from unittest import mock
 
 from .._egg_info import (
     Requirement, EggMetadata, LegacySpecDepend, parse_rawspec,
     split_egg_name)
-
 from .common import (
     BROKEN_MCCABE_EGG, DATA_DIR, ENSTALLER_EGG, ETS_EGG,
     FAKE_MEDIALOG_BOARDFILE_1_6_1_EGG, FAKE_MEDIALOG_BOARDFILE_1_6_1_PKG_INFO,
     MKL_EGG, NUMEXPR_2_2_2_EGG, PYMULTINEST_EGG, _OSX64APP_EGG,
     PYSIDE_1_0_3_EGG, XZ_5_2_0_EGG)
+from okonomiyaki.errors import (
+    InvalidEggName, InvalidMetadataField, MissingMetadata, UnsupportedMetadata,
+    InvalidRequirementStringHyphen)
+from okonomiyaki.utils import tempdir
+from okonomiyaki.utils.test_data import (
+    MKL_10_3_RH5_X86_64, NOSE_1_3_4_RH6_X86_64, NOSE_1_3_4_RH5_X86_64)
+from okonomiyaki.platforms import EPDPlatform, PlatformABI
+from okonomiyaki.versions import EnpkgVersion, MetadataVersion
 
 
 M = MetadataVersion.from_string