singlestoredb 0.4.0__py3-none-any.whl → 1.0.4__py3-none-any.whl

singlestoredb/{clients/pymysqlsv → mysql}/__init__.py

@@ -1,4 +1,3 @@
-# type: ignore
 """
 PyMySQL: A pure-Python MySQL client library.
 
@@ -23,6 +22,8 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 THE SOFTWARE.
 """
 import sys
+from typing import Any
+from typing import FrozenSet
 
 from .constants import FIELD_TYPE
 from .err import DatabaseError
@@ -44,32 +45,28 @@ from .times import Timestamp
 from .times import TimestampFromTicks
 
 
-VERSION = (1, 0, 2, None)
-if VERSION[3] is not None:
-    VERSION_STRING = '%d.%d.%d_%s' % VERSION
-else:
-    VERSION_STRING = '%d.%d.%d' % VERSION[:3]
 threadsafety = 1
 apilevel = '2.0'
 paramstyle = 'pyformat'
 
-from . import connections  # noqa: E402
+from . import connection  # noqa: E402
 
 
-class DBAPISet(frozenset):
-    def __ne__(self, other):
+class DBAPISet(FrozenSet[Any]):
+
+    def __ne__(self, other: Any) -> bool:
         if isinstance(other, set):
             return frozenset.__ne__(self, other)
         else:
             return other not in self
 
-    def __eq__(self, other):
+    def __eq__(self, other: Any) -> bool:
         if isinstance(other, frozenset):
             return frozenset.__eq__(self, other)
         else:
             return other in self
 
-    def __hash__(self):
+    def __hash__(self) -> int:
         return frozenset.__hash__(self)
 
 
@@ -101,19 +98,17 @@ DATETIME = TIMESTAMP
 ROWID = DBAPISet()
 
 
-def Binary(x):
+def Binary(x: Any) -> bytes:
     """Return x as a binary type."""
     return bytes(x)
 
 
-Connect = connect = Connection = connections.Connection
+Connect = connect = Connection = connection.Connection  # type: ignore
 
 
-def get_client_info():  # for MySQLdb compatibility
-    version = VERSION
-    if VERSION[3] is None:
-        version = VERSION[:3]
-    return '.'.join(map(str, version))
+def get_client_info() -> str:  # for MySQLdb compatibility
+    from .. import __version__
+    return __version__
 
 
 # we include a doctored version_info here for MySQLdb compatibility
@@ -124,11 +119,11 @@ NULL = 'NULL'
 __version__ = get_client_info()
 
 
-def thread_safe():
+def thread_safe() -> bool:
     return True  # match MySQLdb.thread_safe()
 
 
-def install_as_MySQLdb():
+def install_as_MySQLdb() -> None:
     """
     After this function is called, any application that imports MySQLdb or
     _mysql will unwittingly actually use pymysql.
@@ -169,7 +164,7 @@ __all__ = [
     'Warning',
     'apilevel',
     'connect',
-    'connections',
+    'connection',
     'constants',
     'converters',
     'cursors',

singlestoredb/{clients/pymysqlsv → mysql}/_auth.py

@@ -4,7 +4,6 @@ Implements auth methods
 """
 from .err import OperationalError
 
-
 try:
     from cryptography.hazmat.backends import default_backend
     from cryptography.hazmat.primitives import serialization, hashes
@@ -17,8 +16,10 @@ except ImportError:
 from functools import partial
 import hashlib
 
+from ..config import get_option
+
 
-DEBUG = False
+DEBUG = get_option('debug.connection')
 SCRAMBLE_LENGTH = 20
 sha1_new = partial(hashlib.new, 'sha1')
 
@@ -28,7 +29,7 @@ sha1_new = partial(hashlib.new, 'sha1')
 
 
 def scramble_native_password(password, message):
-    """Scramble used for mysql_native_password"""
+    """Scramble used for mysql_native_password."""
     if not password:
         return b''
 
@@ -76,9 +77,11 @@ def _scalar_clamp(s32):
 
 
 def ed25519_password(password, scramble):
-    """Sign a random scramble with elliptic curve Ed25519.
+    """
+    Sign a random scramble with elliptic curve Ed25519.
 
     Secret and public key are derived from password.
+
     """
     # variable names based on rfc8032 section-5.1.6
     #
@@ -125,7 +128,8 @@ def _roundtrip(conn, send_data):
 
 def _xor_password(password, salt):
     # Trailing NUL character will be added in Auth Switch Request.
-    # See https://github.com/mysql/mysql-server/blob/7d10c82196c8e45554f27c00681474a9fb86d137/sql/auth/sha2_password.cc#L939-L945
+    # See https://github.com/mysql/mysql-server/blob/
+    #       7d10c82196c8e45554f27c00681474a9fb86d137/sql/auth/sha2_password.cc#L939-L945
     salt = salt[:SCRAMBLE_LENGTH]
     password_bytes = bytearray(password)
     # salt = bytearray(salt)  # for PY2 compat.
@@ -136,13 +140,16 @@ def _xor_password(password, salt):
 
 
 def sha2_rsa_encrypt(password, salt, public_key):
-    """Encrypt password with salt and public_key.
+    """
+    Encrypt password with salt and public_key.
 
     Used for sha256_password and caching_sha2_password.
+
     """
     if not _have_cryptography:
         raise RuntimeError(
-            "'cryptography' package is required for sha256_password or caching_sha2_password auth methods",
+            "'cryptography' package is required for sha256_password or "
+            'caching_sha2_password auth methods',
         )
     message = _xor_password(password + b'\0', salt)
     rsa_key = serialization.load_pem_public_key(public_key, default_backend())
@@ -189,9 +196,11 @@ def sha256_password_auth(conn, pkt):
 
 def scramble_caching_sha2(password, nonce):
     # (bytes, bytes) -> bytes
-    """Scramble algorithm used in cached_sha2_password fast path.
+    """
+    Scramble algorithm used in cached_sha2_password fast path.
 
     XOR(SHA256(password), SHA256(SHA256(SHA256(password)), nonce))
+
     """
     if not password:
         return b''
@@ -265,3 +274,25 @@ def caching_sha2_password_auth(conn, pkt):
 
     data = sha2_rsa_encrypt(conn.password, conn.salt, conn.server_public_key)
     pkt = _roundtrip(conn, data)
+
+
+def gssapi_auth(data):
+    try:
+        import gssapi.raw
+    except ImportError:
+        raise ImportError(
+            'The `gssapi` package must be '
+            'installed for Kerberos authentication',
+        )
+
+    ctx = None
+    try:
+        ctx = gssapi.raw.init_sec_context(
+            gssapi.raw.import_name(data, gssapi.raw.NameType.user),
+            flags=[0], lifetime=0,
+        )
+        return ctx.token
+
+    finally:
+        if ctx is not None:
+            gssapi.raw.delete_sec_context(ctx.context)

singlestoredb/{clients/pymysqlsv → mysql}/charset.py

@@ -1,14 +1,16 @@
-# type: ignore
+from typing import Dict
+from typing import Optional
 
 MBLENGTH = {8: 1, 33: 3, 88: 2, 91: 2}
 
 
 class Charset:
-    def __init__(self, id, name, collation, is_default):
+
+    def __init__(self, id: int, name: str, collation: str, is_default: str):
         self.id, self.name, self.collation = id, name, collation
         self.is_default = is_default == 'Yes'
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return 'Charset(id=%s, name=%r, collation=%r)' % (
             self.id,
             self.name,
@@ -16,7 +18,7 @@ class Charset:
         )
 
     @property
-    def encoding(self):
+    def encoding(self) -> str:
         name = self.name
         if name in ('utf8mb4', 'utf8mb3'):
             return 'utf8'
@@ -29,40 +31,41 @@ class Charset:
         return name
 
     @property
-    def is_binary(self):
+    def is_binary(self) -> bool:
         return self.id == 63
 
 
 class Charsets:
-    def __init__(self):
-        self._by_id = {}
-        self._by_name = {}
 
-    def add(self, c):
+    def __init__(self) -> None:
+        self._by_id: Dict[int, Charset] = {}
+        self._by_name: Dict[str, Charset] = {}
+
+    def add(self, c: Charset) -> None:
         self._by_id[c.id] = c
         if c.is_default:
             self._by_name[c.name] = c
 
-    def by_id(self, id):
+    def by_id(self, id: int) -> Charset:
         return self._by_id[id]
 
-    def by_name(self, name):
+    def by_name(self, name: str) -> Optional[Charset]:
         return self._by_name.get(name.lower())
 
 
 _charsets = Charsets()
-"""
-Generated with:
-
-mysql -N -s -e "select id, character_set_name, collation_name, is_default
-from information_schema.collations order by id;" | python -c "import sys
-for l in sys.stdin.readlines():
-        id, name, collation, is_default  = l.split(chr(9))
-        print '_charsets.add(Charset(%s, \'%s\', \'%s\', \'%s\'))' \
-                % (id, name, collation, is_default.strip())
-"
-
-"""
+
+# Generated with:
+#
+# mysql -N -s -e "select id, character_set_name, collation_name, is_default
+# from information_schema.collations order by id;" | python -c "import sys
+# for l in sys.stdin.readlines():
+#         id, name, collation, is_default  = l.split(chr(9))
+#         print '_charsets.add(Charset(%s, \'%s\', \'%s\', \'%s\'))' \
+#                 % (id, name, collation, is_default.strip())
+# "
+#
+#
 _charsets.add(Charset(1, 'big5', 'big5_chinese_ci', 'Yes'))
 _charsets.add(Charset(2, 'latin2', 'latin2_czech_cs', ''))
 _charsets.add(Charset(3, 'dec8', 'dec8_swedish_ci', 'Yes'))