singlestoredb 0.8.8__cp36-abi3-win32.whl → 0.9.0__cp36-abi3-win32.whl

singlestoredb/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
@@ -51,20 +52,21 @@ paramstyle = 'pyformat'
 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)
 
 
@@ -96,15 +98,15 @@ DATETIME = TIMESTAMP
 ROWID = DBAPISet()
 
 
-def Binary(x):
+def Binary(x: Any) -> bytes:
     """Return x as a binary type."""
     return bytes(x)
 
 
-Connect = connect = Connection = connection.Connection
+Connect = connect = Connection = connection.Connection  # type: ignore
 
 
-def get_client_info():  # for MySQLdb compatibility
+def get_client_info() -> str:  # for MySQLdb compatibility
     from .. import __version__
     return __version__
 
@@ -117,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.

singlestoredb/mysql/_auth.py

@@ -16,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')
 

singlestoredb/mysql/charset.py

@@ -1,15 +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,
@@ -17,7 +18,7 @@ class Charset:
         )
 
     @property
-    def encoding(self):
+    def encoding(self) -> str:
         name = self.name
         if name in ('utf8mb4', 'utf8mb3'):
             return 'utf8'
@@ -30,25 +31,25 @@ 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 __init__(self) -> None:
+        self._by_id: Dict[int, Charset] = {}
+        self._by_name: Dict[str, Charset] = {}
 
-    def add(self, c):
+    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())
 
 

singlestoredb/mysql/connection.py

@@ -46,6 +46,7 @@ from .protocol import (
     LoadLocalPacketWrapper,
 )
 from . import err
+from ..config import get_option
 from ..connection import Connection as BaseConnection
 
 try:
@@ -65,7 +66,7 @@ except (ImportError, KeyError):
     # KeyError occurs when there's no entry in OS database for a current user.
     DEFAULT_USER = None
 
-DEBUG = False
+DEBUG = get_option('debug.connection')
 
 TEXT_TYPES = {
     FIELD_TYPE.BIT,
@@ -1051,7 +1052,7 @@ class Connection(BaseConnection):
         return data
 
     def _write_bytes(self, data):
-        if self._read_timeout is not None:
+        if self._write_timeout is not None:
             self._sock.settimeout(self._write_timeout)
         try:
             self._sock.sendall(data)
@@ -1090,7 +1091,7 @@ class Connection(BaseConnection):
 
         """
         if not self._sock:
-            raise err.InterfaceError(0, '')
+            raise err.InterfaceError(0, 'The connection has been closed')
 
         # If the last query was unbuffered, make sure it finishes before
         # sending new commands

singlestoredb/mysql/constants/CLIENT.py

@@ -1,4 +1,3 @@
-# type: ignore
 # https://dev.mysql.com/doc/internals/en/capability-flags.html#packet-Protocol::CapabilityFlags
 LONG_PASSWORD = 1
 FOUND_ROWS = 1 << 1

singlestoredb/mysql/constants/COMMAND.py

@@ -1,4 +1,3 @@
-# type: ignore
 COM_SLEEP = 0x00
 COM_QUIT = 0x01
 COM_INIT_DB = 0x02

singlestoredb/mysql/constants/CR.py

@@ -1,5 +1,3 @@
-# flake8: noqa
-# type: ignore
 # errmsg.h
 CR_ERROR_FIRST = 2000
 CR_UNKNOWN_ERROR = 2000

singlestoredb/mysql/constants/ER.py

@@ -1,4 +1,3 @@
-# type: ignore
 ERROR_FIRST = 1000
 HASHCHK = 1000
 NISAMCHK = 1001

singlestoredb/mysql/constants/FIELD_TYPE.py

@@ -1,4 +1,3 @@
-# type: ignore
 DECIMAL = 0
 TINY = 1
 SHORT = 2

singlestoredb/mysql/constants/FLAG.py

@@ -1,4 +1,3 @@
-# type: ignore
 NOT_NULL = 1
 PRI_KEY = 2
 UNIQUE_KEY = 4

singlestoredb/mysql/constants/SERVER_STATUS.py

@@ -1,4 +1,3 @@
-# type: ignore
 SERVER_STATUS_IN_TRANS = 1
 SERVER_STATUS_AUTOCOMMIT = 2
 SERVER_MORE_RESULTS_EXISTS = 8

singlestoredb/mysql/converters.py

@@ -1,7 +1,12 @@
-# type: ignore
 import datetime
 import time
 from decimal import Decimal
+from typing import Any
+from typing import Callable
+from typing import Dict
+from typing import Optional
+from typing import Tuple
+from typing import Union
 
 from ..converters import converters as decoders
 from .err import ProgrammingError
@@ -26,7 +31,10 @@ except ImportError:
     has_pygeos = False
 
 
-def escape_item(val, charset, mapping=None):
+Encoders = Dict[type, Callable[..., Union[str, Dict[str, str]]]]
+
+
+def escape_item(val: Any, charset: str, mapping: Optional[Encoders] = None) -> str:
     if mapping is None:
         mapping = encoders
     encoder = mapping.get(type(val), None)
@@ -45,7 +53,11 @@ def escape_item(val, charset, mapping=None):
     return val
 
 
-def escape_dict(val, charset, mapping=None):
+def escape_dict(
+    val: Dict[str, Any],
+    charset: str,
+    mapping: Optional[Encoders] = None,
+) -> Dict[str, str]:
     n = {}
     for k, v in val.items():
         quoted = escape_item(v, charset, mapping)
@@ -53,7 +65,11 @@ def escape_dict(val, charset, mapping=None):
     return n
 
 
-def escape_sequence(val, charset, mapping=None):
+def escape_sequence(
+    val: Any,
+    charset: str,
+    mapping: Optional[Encoders] = None,
+) -> str:
     n = []
     for item in val:
         quoted = escape_item(item, charset, mapping)
@@ -61,28 +77,33 @@ def escape_sequence(val, charset, mapping=None):
     return '(' + ','.join(n) + ')'
 
 
-def escape_set(val, charset, mapping=None):
+def escape_set(val: Any, charset: str, mapping: Optional[Encoders] = None) -> str:
     return ','.join([escape_item(x, charset, mapping) for x in val])
 
 
-def escape_bool(value, mapping=None):
+def escape_bool(value: Any, mapping: Optional[Encoders] = None) -> str:
     return str(int(value))
 
 
-def escape_int(value, mapping=None):
+def escape_int(value: Any, mapping: Optional[Encoders] = None) -> str:
     return str(value)
 
 
-def escape_float(value, mapping=None, nan_as_null=False, inf_as_null=False):
+def escape_float(
+    value: Any,
+    mapping: Optional[Encoders] = None,
+    nan_as_null: bool = False,
+    inf_as_null: bool = False,
+) -> str:
     s = repr(value)
     if s == 'nan':
         if nan_as_null:
             return 'NULL'
-        raise ProgrammingError('%s can not be used with SingleStoreDB' % s)
+        raise ProgrammingError(0, '%s can not be used with SingleStoreDB' % s)
     if s == 'inf':
         if inf_as_null:
             return 'NULL'
-        raise ProgrammingError('%s can not be used with SingleStoreDB' % s)
+        raise ProgrammingError(0, '%s can not be used with SingleStoreDB' % s)
     if 'e' not in s:
         s += 'e0'
     return s
@@ -98,7 +119,7 @@ _escape_table[ord('"')] = '\\"'
 _escape_table[ord("'")] = "\\'"
 
 
-def escape_string(value, mapping=None):
+def escape_string(value: str, mapping: Optional[Encoders] = None) -> str:
     """
     Escapes *value* without adding quote.
 
@@ -108,23 +129,23 @@ def escape_string(value, mapping=None):
     return value.translate(_escape_table)
 
 
-def escape_bytes_prefixed(value, mapping=None):
+def escape_bytes_prefixed(value: bytes, mapping: Optional[Encoders] = None) -> str:
     return "_binary X'{}'".format(value.hex())
 
 
-def escape_bytes(value, mapping=None):
+def escape_bytes(value: bytes, mapping: Optional[Encoders] = None) -> str:
     return "X'{}'".format(value.hex())
 
 
-def escape_str(value, mapping=None):
+def escape_str(value: str, mapping: Optional[Encoders] = None) -> str:
     return "'{}'".format(escape_string(str(value), mapping))
 
 
-def escape_None(value, mapping=None):
+def escape_None(value: str, mapping: Optional[Encoders] = None) -> str:
     return 'NULL'
 
 
-def escape_timedelta(obj, mapping=None):
+def escape_timedelta(obj: datetime.timedelta, mapping: Optional[Encoders] = None) -> str:
     seconds = int(obj.seconds) % 60
     minutes = int(obj.seconds // 60) % 60
     hours = int(obj.seconds // 3600) % 24 + int(obj.days) * 24
@@ -135,7 +156,7 @@ def escape_timedelta(obj, mapping=None):
     return fmt.format(hours, minutes, seconds, obj.microseconds)
 
 
-def escape_time(obj, mapping=None):
+def escape_time(obj: datetime.time, mapping: Optional[Encoders] = None) -> str:
     if obj.microsecond:
         fmt = "'{0.hour:02}:{0.minute:02}:{0.second:02}.{0.microsecond:06}'"
     else:
@@ -143,7 +164,7 @@ def escape_time(obj, mapping=None):
     return fmt.format(obj)
 
 
-def escape_datetime(obj, mapping=None):
+def escape_datetime(obj: datetime.datetime, mapping: Optional[Encoders] = None) -> str:
     if obj.microsecond:
         fmt = "'{0.year:04}-{0.month:02}-{0.day:02} " \
               "{0.hour:02}:{0.minute:02}:{0.second:02}.{0.microsecond:06}'"
@@ -153,20 +174,20 @@ def escape_datetime(obj, mapping=None):
     return fmt.format(obj)
 
 
-def escape_date(obj, mapping=None):
+def escape_date(obj: datetime.date, mapping: Optional[Encoders] = None) -> str:
     fmt = "'{0.year:04}-{0.month:02}-{0.day:02}'"
     return fmt.format(obj)
 
 
-def escape_struct_time(obj, mapping=None):
+def escape_struct_time(obj: Tuple[Any, ...], mapping: Optional[Encoders] = None) -> str:
     return escape_datetime(datetime.datetime(*obj[:6]))
 
 
-def Decimal2Literal(o, d):
+def Decimal2Literal(o: Any, d: Any) -> str:
     return format(o, 'f')
 
 
-def through(x):
+def through(x: Any) -> Any:
     return x
 
 
@@ -179,7 +200,7 @@ def through(x):
 convert_bit = through
 
 
-encoders = {
+encoders: Encoders = {
     bool: escape_bool,
     int: escape_int,
     float: escape_float,
@@ -201,7 +222,7 @@ encoders = {
 
 if has_numpy:
 
-    def escape_numpy(value, mapping=None):
+    def escape_numpy(value: Any, mapping: Optional[Encoders] = None) -> str:
         """Convert numpy arrays to vectors of bytes."""
         return escape_bytes(value.tobytes(), mapping=mapping)
 
@@ -225,7 +246,7 @@ if has_numpy:
 
 if has_shapely:
 
-    def escape_shapely(value, mapping=None):
+    def escape_shapely(value: Any, mapping: Optional[Encoders] = None) -> str:
         """Convert shapely geo objects."""
         return escape_str(shapely.wkt.dumps(value), mapping=mapping)
 
@@ -235,7 +256,7 @@ if has_shapely:
 
 if has_pygeos:
 
-    def escape_pygeos(value, mapping=None):
+    def escape_pygeos(value: Any, mapping: Optional[Encoders] = None) -> str:
         """Convert pygeos objects."""
         return escape_str(pygeos.io.to_wkt(value), mapping=mapping)
 
@@ -243,8 +264,8 @@ if has_pygeos:
 
 
 # for MySQLdb compatibility
-conversions = encoders.copy()
-conversions.update(decoders)
+conversions = encoders.copy()  # type: ignore
+conversions.update(decoders)   # type: ignore
 Thing2Literal = escape_str
 
 # Run doctests with `pytest --doctest-modules pymysql/converters.py`

singlestoredb/mysql/err.py

@@ -1,5 +1,5 @@
-# type: ignore
 import struct
+from typing import Any
 
 from ..exceptions import DatabaseError  # noqa: F401
 from ..exceptions import DataError  # noqa: F401
@@ -18,7 +18,7 @@ from .constants import ER
 error_map = {}
 
 
-def _map_error(exc, *errors):
+def _map_error(exc: Any, *errors: int) -> None:
     for error in errors:
         error_map[error] = exc
 
@@ -83,7 +83,7 @@ _map_error(
 del _map_error, ER
 
 
-def raise_mysql_exception(data):
+def raise_mysql_exception(data: bytes) -> Exception:
     errno = struct.unpack('<h', data[1:3])[0]
     errval = data[9:].decode('utf-8', 'replace')
     errorclass = error_map.get(errno)

singlestoredb/mysql/optionfile.py

@@ -1,20 +1,20 @@
-# type: ignore
 import configparser
+from typing import Any
 
 
 class Parser(configparser.RawConfigParser):
 
-    def __init__(self, **kwargs):
+    def __init__(self, **kwargs: Any) -> None:
         kwargs['allow_no_value'] = True
         configparser.RawConfigParser.__init__(self, **kwargs)
 
-    def __remove_quotes(self, value):
+    def __remove_quotes(self, value: str) -> str:
         quotes = ["'", '"']
         for quote in quotes:
             if len(value) >= 2 and value[0] == value[-1] == quote:
                 return value[1:-1]
         return value
 
-    def get(self, section, option):
+    def get(self, section: str, option: str) -> str:  # type: ignore
         value = configparser.RawConfigParser.get(self, section, option)
         return self.__remove_quotes(value)

singlestoredb/mysql/protocol.py

@@ -5,13 +5,14 @@ import struct
 import sys
 
 from . import err
+from ..config import get_option
 from ..utils.results import Description
 from .charset import MBLENGTH
 from .constants import FIELD_TYPE
 from .constants import SERVER_STATUS
 
 
-DEBUG = False
+DEBUG = get_option('debug.connection')
 
 NULL_COLUMN = 251
 UNSIGNED_CHAR_COLUMN = 251

singlestoredb/mysql/times.py

@@ -1,4 +1,3 @@
-# type: ignore
 from datetime import date
 from datetime import datetime
 from datetime import time
@@ -12,13 +11,13 @@ TimeDelta = timedelta
 Timestamp = datetime
 
 
-def DateFromTicks(ticks):
+def DateFromTicks(ticks: int) -> date:
     return date(*localtime(ticks)[:3])
 
 
-def TimeFromTicks(ticks):
+def TimeFromTicks(ticks: int) -> time:
     return time(*localtime(ticks)[3:6])
 
 
-def TimestampFromTicks(ticks):
+def TimestampFromTicks(ticks: int) -> datetime:
     return datetime(*localtime(ticks)[:6])

singlestoredb/tests/test2.sql

@@ -0,0 +1 @@
+-- Nearly empty file for Stages test