driver-six

Author: vgali7

cassandra/auth.py

@@ -278,7 +278,7 @@ def get_initial_challenge(self):
     def evaluate_challenge(self, challenge):
         if challenge == six.b('PLAIN-START'):
             data = "\x00%s\x00%s" % (self.username, self.password)
-            return data if six.PY2 else data.encode()
+            return data.encode()
         raise Exception('Did not receive a valid challenge response from server')
 
 

cassandra/cluster.py

@@ -107,8 +107,6 @@
 except ImportError:
     from cassandra.util import WeakSet  # NOQA
 
-if six.PY3:
-    long = int
 
 def _is_eventlet_monkey_patched():
     if 'eventlet.patcher' not in sys.modules:
@@ -3345,10 +3343,6 @@ def user_type_registered(self, keyspace, user_type, klass):
                 'User type %s does not exist in keyspace %s' % (user_type, keyspace))
 
         field_names = type_meta.field_names
-        if six.PY2:
-            # go from unicode to string to avoid decode errors from implicit
-            # decode when formatting non-ascii values
-            field_names = [fn.encode('utf-8') for fn in field_names]
 
         def encode(val):
             return '{ %s }' % ' , '.join('%s : %s' % (

cassandra/compat.py

@@ -12,9 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import six
-
-if six.PY2:
-    from collections import Mapping
-elif six.PY3:
-    from collections.abc import Mapping
+from collections.abc import Mapping

cassandra/concurrent.py

@@ -145,7 +145,7 @@ def _execute(self, idx, statement, params):
         except Exception as exc:
             # exc_info with fail_fast to preserve stack trace info when raising on the client thread
             # (matches previous behavior -- not sure why we wouldn't want stack trace in the other case)
-            e = sys.exc_info() if self._fail_fast and six.PY2 else exc
+            e = exc
 
             # If we're not failing fast and all executions are raising, there is a chance of recursing
             # here as subsequent requests are attempted. If we hit this threshold, schedule this result/retry
@@ -165,11 +165,7 @@ def _on_error(self, result, future, idx):
 
     @staticmethod
     def _raise(exc):
-        if six.PY2 and isinstance(exc, tuple):
-            (exc_type, value, traceback) = exc
-            six.reraise(exc_type, value, traceback)
-        else:
-            raise exc
+        raise exc
 
 
 class ConcurrentExecutorGenResults(_ConcurrentExecutor):

cassandra/connection.py

@@ -605,11 +605,8 @@ def wrapper(self, *args, **kwargs):
 
 DEFAULT_CQL_VERSION = '3.0.0'
 
-if six.PY3:
-    def int_from_buf_item(i):
-        return i
-else:
-    int_from_buf_item = ord
+def int_from_buf_item(i):
+    return i
 
 
 class _ConnectionIOBuffer(object):

cassandra/cqlengine/__init__.py

@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import six
-
 
 # Caching constants.
 CACHING_ALL = "ALL"
@@ -31,7 +29,5 @@ class ValidationError(CQLEngineException):
 
 
 class UnicodeMixin(object):
-    if six.PY3:
-        __str__ = lambda x: x.__unicode__()
-    else:
-        __str__ = lambda x: six.text_type(x).encode('utf-8')
+    __str__ = lambda x: x.__unicode__()
+

cassandra/cqltypes.py

@@ -54,10 +54,9 @@
 from cassandra import util
 
 _little_endian_flag = 1  # we always serialize LE
-if six.PY3:
-    import ipaddress
+import ipaddress
 
-_ord = ord if six.PY2 else lambda x: x
+lambda x: x
 
 apache_cassandra_type_prefix = 'org.apache.cassandra.db.marshal.'
 
@@ -66,16 +65,12 @@
 
 log = logging.getLogger(__name__)
 
-if six.PY3:
-    _number_types = frozenset((int, float))
-    long = int
+_number_types = frozenset((int, float))
+long = int
 
-    def _name_from_hex_string(encoded_name):
-        bin_str = unhexlify(encoded_name)
-        return bin_str.decode('ascii')
-else:
-    _number_types = frozenset((int, long, float))
-    _name_from_hex_string = unhexlify
+def _name_from_hex_string(encoded_name):
+    bin_str = unhexlify(encoded_name)
+    return bin_str.decode('ascii')
 
 
 def trim_if_startswith(s, prefix):
@@ -383,8 +378,6 @@ def apply_parameters(cls, subtypes, names=None):
             raise ValueError("%s types require %d subtypes (%d given)"
                              % (cls.typename, cls.num_subtypes, len(subtypes)))
         newname = cls.cass_parameterized_type_with(subtypes)
-        if six.PY2 and isinstance(newname, unicode):
-            newname = newname.encode('utf-8')
         return type(newname, (cls,), {'subtypes': subtypes, 'cassname': cls.cassname, 'fieldnames': names})
 
     @classmethod
@@ -415,16 +408,10 @@ class _UnrecognizedType(_CassandraType):
     num_subtypes = 'UNKNOWN'
 
 
-if six.PY3:
-    def mkUnrecognizedType(casstypename):
-        return CassandraTypeType(casstypename,
-                                 (_UnrecognizedType,),
-                                 {'typename': "'%s'" % casstypename})
-else:
-    def mkUnrecognizedType(casstypename):  # noqa
-        return CassandraTypeType(casstypename.encode('utf8'),
-                                 (_UnrecognizedType,),
-                                 {'typename': "'%s'" % casstypename})
+def mkUnrecognizedType(casstypename):
+    return CassandraTypeType(casstypename,
+                                (_UnrecognizedType,),
+                                {'typename': "'%s'" % casstypename})
 
 
 class BytesType(_CassandraType):
@@ -500,25 +487,21 @@ def serialize(byts, protocol_version):
         return int8_pack(byts)
 
 
-if six.PY2:
-    class AsciiType(_CassandraType):
-        typename = 'ascii'
-        empty_binary_ok = True
-else:
-    class AsciiType(_CassandraType):
-        typename = 'ascii'
-        empty_binary_ok = True
 
-        @staticmethod
-        def deserialize(byts, protocol_version):
-            return byts.decode('ascii')
+class AsciiType(_CassandraType):
+    typename = 'ascii'
+    empty_binary_ok = True
 
-        @staticmethod
-        def serialize(var, protocol_version):
-            try:
-                return var.encode('ascii')
-            except UnicodeDecodeError:
-                return var
+    @staticmethod
+    def deserialize(byts, protocol_version):
+        return byts.decode('ascii')
+
+    @staticmethod
+    def serialize(var, protocol_version):
+        try:
+            return var.encode('ascii')
+        except UnicodeDecodeError:
+            return var
 
 
 class FloatType(_CassandraType):
@@ -603,7 +586,7 @@ def serialize(addr, protocol_version):
                 # since we've already determined the AF
                 return socket.inet_aton(addr)
         except:
-            if six.PY3 and isinstance(addr, (ipaddress.IPv4Address, ipaddress.IPv6Address)):
+            if isinstance(addr, (ipaddress.IPv4Address, ipaddress.IPv6Address)):
                 return addr.packed
             raise ValueError("can't interpret %r as an inet address" % (addr,))
 
@@ -975,8 +958,6 @@ class UserType(TupleType):
     def make_udt_class(cls, keyspace, udt_name, field_names, field_types):
         assert len(field_names) == len(field_types)
 
-        if six.PY2 and isinstance(udt_name, unicode):
-            udt_name = udt_name.encode('utf-8')
 
         instance = cls._cache.get((keyspace, udt_name))
         if not instance or instance.fieldnames != field_names or instance.subtypes != field_types:
@@ -992,8 +973,6 @@ def make_udt_class(cls, keyspace, udt_name, field_names, field_types):
 
     @classmethod
     def evict_udt_class(cls, keyspace, udt_name):
-        if six.PY2 and isinstance(udt_name, unicode):
-            udt_name = udt_name.encode('utf-8')
         try:
             del cls._cache[(keyspace, udt_name)]
         except KeyError:

cassandra/datastax/graph/graphson.py

@@ -164,9 +164,7 @@ class Int64TypeIO(IntegerTypeIO):
 
     @classmethod
     def deserialize(cls, value, reader=None):
-        if six.PY3:
-            return value
-        return long(value)
+        return value
 
 
 class FloatTypeIO(GraphSONTypeIO):
@@ -274,8 +272,7 @@ class BlobTypeIO(GraphSONTypeIO):
     @classmethod
     def serialize(cls, value, writer=None):
         value = base64.b64encode(value)
-        if six.PY3:
-            value = value.decode('utf-8')
+        value = value.decode('utf-8')
         return value
 
     @classmethod
@@ -845,12 +842,8 @@ class GraphSON1Serializer(_BaseGraphSONSerializer):
     GraphSON1Serializer.register(ipaddress.IPv4Address, InetTypeIO)
     GraphSON1Serializer.register(ipaddress.IPv6Address, InetTypeIO)
 
-if six.PY2:
-    GraphSON1Serializer.register(buffer, ByteBufferTypeIO)
-    GraphSON1Serializer.register(unicode, TextTypeIO)
-else:
-    GraphSON1Serializer.register(memoryview, ByteBufferTypeIO)
-    GraphSON1Serializer.register(bytes, ByteBufferTypeIO)
+GraphSON1Serializer.register(memoryview, ByteBufferTypeIO)
+GraphSON1Serializer.register(bytes, ByteBufferTypeIO)
 
 
 class _BaseGraphSONDeserializer(object):
@@ -922,9 +915,7 @@ def deserialize_int(cls, value):
 
     @classmethod
     def deserialize_bigint(cls, value):
-        if six.PY3:
-            return cls.deserialize_int(value)
-        return long(value)
+        return cls.deserialize_int(value)
 
     @classmethod
     def deserialize_double(cls, value):
@@ -1007,8 +998,6 @@ def serialize(self, value, writer=None):
 
 
 GraphSON2Serializer.register(int, IntegerTypeIO)
-if six.PY2:
-    GraphSON2Serializer.register(long, IntegerTypeIO)
 
 
 class GraphSON2Deserializer(_BaseGraphSONDeserializer):

cassandra/datastax/insights/reporter.py

@@ -199,9 +199,9 @@ def _get_startup_data(self):
                 },
                 'platformInfo': {
                     'os': {
-                        'name': uname_info.system if six.PY3 else uname_info[0],
-                        'version': uname_info.release if six.PY3 else uname_info[2],
-                        'arch': uname_info.machine if six.PY3 else uname_info[4]
+                        'name': uname_info.system,
+                        'version': uname_info.release,
+                        'arch': uname_info.machine
                     },
                     'cpus': {
                         'length': multiprocessing.cpu_count(),

cassandra/encoder.py

@@ -32,11 +32,7 @@
 from cassandra.util import (OrderedDict, OrderedMap, OrderedMapSerializedKey,
                             sortedset, Time, Date, Point, LineString, Polygon)
 
-if six.PY3:
-    import ipaddress
-
-if six.PY3:
-    long = int
+import ipaddress
 
 
 def cql_quote(term):
@@ -97,21 +93,14 @@ def __init__(self):
             Polygon: self.cql_encode_str_quoted
         }
 
-        if six.PY2:
-            self.mapping.update({
-                unicode: self.cql_encode_unicode,
-                buffer: self.cql_encode_bytes,
-                long: self.cql_encode_object,
-                types.NoneType: self.cql_encode_none,
-            })
-        else:
-            self.mapping.update({
-                memoryview: self.cql_encode_bytes,
-                bytes: self.cql_encode_bytes,
-                type(None): self.cql_encode_none,
-                ipaddress.IPv4Address: self.cql_encode_ipaddress,
-                ipaddress.IPv6Address: self.cql_encode_ipaddress
-            })
+
+        self.mapping.update({
+            memoryview: self.cql_encode_bytes,
+            bytes: self.cql_encode_bytes,
+            type(None): self.cql_encode_none,
+            ipaddress.IPv4Address: self.cql_encode_ipaddress,
+            ipaddress.IPv6Address: self.cql_encode_ipaddress
+        })
 
     def cql_encode_none(self, val):
         """
@@ -134,16 +123,9 @@ def cql_encode_str(self, val):
     def cql_encode_str_quoted(self, val):
         return "'%s'" % val
 
-    if six.PY3:
-        def cql_encode_bytes(self, val):
-            return (b'0x' + hexlify(val)).decode('utf-8')
-    elif sys.version_info >= (2, 7):
-        def cql_encode_bytes(self, val):  # noqa
-            return b'0x' + hexlify(val)
-    else:
-        # python 2.6 requires string or read-only buffer for hexlify
-        def cql_encode_bytes(self, val):  # noqa
-            return b'0x' + hexlify(buffer(val))
+    def cql_encode_bytes(self, val):
+        return (b'0x' + hexlify(val)).decode('utf-8')
+
 
     def cql_encode_object(self, val):
         """
@@ -240,10 +222,9 @@ def cql_encode_all_types(self, val, as_text_type=False):
             return encoded.decode('utf-8')
         return encoded
 
-    if six.PY3:
-        def cql_encode_ipaddress(self, val):
-            """
-            Converts an ipaddress (IPV4Address, IPV6Address) to a CQL string. This
-            is suitable for ``inet`` type columns.
-            """
-            return "'%s'" % val.compressed
+    def cql_encode_ipaddress(self, val):
+        """
+        Converts an ipaddress (IPV4Address, IPV6Address) to a CQL string. This
+        is suitable for ``inet`` type columns.
+        """
+        return "'%s'" % val.compressed