refactor: remove old compat text_type, byte_type, basestring

Author: betaboon

mocket/compat.py

@@ -9,21 +9,17 @@
 
 ENCODING: Final[str] = os.getenv("MOCKET_ENCODING", "utf-8")
 
-text_type = str
-byte_type = bytes
-basestring = (str,)
-
 
 def encode_to_bytes(s: str | bytes, encoding: str = ENCODING) -> bytes:
-    if isinstance(s, text_type):
+    if isinstance(s, str):
         s = s.encode(encoding)
-    return byte_type(s)
+    return bytes(s)
 
 
 def decode_from_bytes(s: str | bytes, encoding: str = ENCODING) -> str:
-    if isinstance(s, byte_type):
+    if isinstance(s, bytes):
         s = codecs.decode(s, encoding, "ignore")
-    return text_type(s)
+    return str(s)
 
 
 def shsplit(s: str | bytes) -> list[str]:

mocket/mocket.py

@@ -23,13 +23,7 @@
     urllib3_wrap_socket = None
 
 
-from mocket.compat import (
-    basestring,
-    byte_type,
-    decode_from_bytes,
-    encode_to_bytes,
-    text_type,
-)
+from mocket.compat import decode_from_bytes, encode_to_bytes
 from mocket.utils import (
     MocketMode,
     MocketSocketCore,
@@ -323,7 +317,7 @@ def true_sendall(self, data, *args, **kwargs):
         # make request unique again
         req_signature = _hash_request(hasher, req)
         # port should be always a string
-        port = text_type(self._port)
+        port = str(self._port)
 
         # prepare responses dictionary
         responses = {}
@@ -433,7 +427,7 @@ class Mocket:
     _address = (None, None)
     _entries = collections.defaultdict(list)
     _requests = []
-    _namespace = text_type(id(_entries))
+    _namespace = str(id(_entries))
     _truesocket_recording_dir = None
 
     @classmethod
@@ -524,7 +518,7 @@ def enable(namespace=None, truesocket_recording_dir=None):
         socket.socketpair = socket.__dict__["socketpair"] = socketpair
         ssl.wrap_socket = ssl.__dict__["wrap_socket"] = FakeSSLContext.wrap_socket
         ssl.SSLContext = ssl.__dict__["SSLContext"] = FakeSSLContext
-        socket.inet_pton = socket.__dict__["inet_pton"] = lambda family, ip: byte_type(
+        socket.inet_pton = socket.__dict__["inet_pton"] = lambda family, ip: bytes(
             "\x7f\x00\x00\x01", "utf-8"
         )
         urllib3.util.ssl_.wrap_socket = urllib3.util.ssl_.__dict__["wrap_socket"] = (
@@ -598,13 +592,13 @@ def assert_fail_if_entries_not_served(cls):
 
 
 class MocketEntry:
-    class Response(byte_type):
+    class Response(bytes):
         @property
         def data(self):
             return self
 
     response_index = 0
-    request_cls = byte_type
+    request_cls = bytes
     response_cls = Response
     responses = None
     _served = None
@@ -613,9 +607,7 @@ def __init__(self, location, responses):
         self._served = False
         self.location = location
 
-        if not isinstance(responses, collections_abc.Iterable) or isinstance(
-            responses, basestring
-        ):
+        if not isinstance(responses, collections_abc.Iterable):
             responses = [responses]
 
         if not responses:
@@ -624,7 +616,7 @@ def __init__(self, location, responses):
             self.responses = []
             for r in responses:
                 if not isinstance(r, BaseException) and not getattr(r, "data", False):
-                    if isinstance(r, text_type):
+                    if isinstance(r, str):
                         r = encode_to_bytes(r)
                     r = self.response_cls(r)
                 self.responses.append(r)
@@ -664,7 +656,7 @@ def __init__(
     ):
         self.instance = instance
         self.truesocket_recording_dir = truesocket_recording_dir
-        self.namespace = namespace or text_type(id(self))
+        self.namespace = namespace or str(id(self))
         MocketMode().STRICT = strict_mode
         if strict_mode:
             MocketMode().STRICT_ALLOWED = strict_mode_allowed or []

mocket/mockredis.py

@@ -1,11 +1,9 @@
 from itertools import chain
 
 from mocket.compat import (
-    byte_type,
     decode_from_bytes,
     encode_to_bytes,
     shsplit,
-    text_type,
 )
 from mocket.mocket import Mocket, MocketEntry
 
@@ -20,7 +18,7 @@ def __init__(self, data=None):
         self.data = Redisizer.redisize(data or OK)
 
 
-class Redisizer(byte_type):
+class Redisizer(bytes):
     @staticmethod
     def tokens(iterable):
         iterable = [encode_to_bytes(x) for x in iterable]
@@ -36,15 +34,15 @@ def get_conversion(t):
                     Redisizer.tokens(list(chain(*tuple(x.items()))))
                 ),
                 int: lambda x: f":{x}".encode(),
-                text_type: lambda x: "${}\r\n{}".format(
-                    len(x.encode("utf-8")), x
-                ).encode("utf-8"),
+                str: lambda x: "${}\r\n{}".format(len(x.encode("utf-8")), x).encode(
+                    "utf-8"
+                ),
                 list: lambda x: b"\r\n".join(Redisizer.tokens(x)),
             }[t]
 
         if isinstance(data, Redisizer):
             return data
-        if isinstance(data, byte_type):
+        if isinstance(data, bytes):
             data = decode_from_bytes(data)
         return Redisizer(get_conversion(data.__class__)(data) + b"\r\n")
 

mocket/plugins/httpretty/__init__.py

@@ -1,6 +1,6 @@
 from mocket import Mocket, mocketize
 from mocket.async_mocket import async_mocketize
-from mocket.compat import ENCODING, byte_type, text_type
+from mocket.compat import ENCODING
 from mocket.mockhttp import Entry as MocketHttpEntry
 from mocket.mockhttp import Request as MocketHttpRequest
 from mocket.mockhttp import Response as MocketHttpResponse
@@ -129,6 +129,6 @@ def __getattr__(self, name):
     "HEAD",
     "PATCH",
     "register_uri",
-    "text_type",
-    "byte_type",
+    "str",
+    "bytes",
 )