Add type version check

Author: kesmit13

singlestoredb/connection.py

@@ -5,6 +5,7 @@
 import io
 import queue
 import re
+import sys
 import warnings
 import weakref
 from collections.abc import Mapping
@@ -35,6 +36,11 @@ def itertuples(self, *args: Any, **kwargs: Any) -> None:
 from .utils.results import Description
 from .utils.results import Result
 
+if sys.version_info < (3, 10):
+    InfileQueue = queue.Queue
+else:
+    InfileQueue = queue.Queue[Union[bytes, str]]
+
 
 # DB-API settings
 apilevel = '2.0'
@@ -498,12 +504,12 @@ def close(self) -> None:
     def execute(
         self, query: str,
         args: Optional[Union[Sequence[Any], Dict[str, Any], Any]] = None,
-        infile_stream: Optional[
+        infile_stream: Optional[  # type: ignore
             Union[
                 io.RawIOBase,
                 io.TextIOBase,
                 Iterator[Union[bytes, str]],
-                queue.Queue[Union[bytes, str]],
+                InfileQueue,
             ]
         ] = None,
     ) -> int:

singlestoredb/http/connection.py

@@ -7,7 +7,6 @@
 import json
 import math
 import os
-import queue
 import re
 import time
 from base64 import b64decode
@@ -422,12 +421,12 @@ def close(self) -> None:
     def execute(
         self, query: str,
         args: Optional[Union[Sequence[Any], Dict[str, Any]]] = None,
-        infile_stream: Optional[
+        infile_stream: Optional[  # type: ignore
             Union[
                 io.RawIOBase,
                 io.TextIOBase,
                 Iterable[Union[bytes, str]],
-                queue.Queue[Union[bytes, str]],
+                connection.InfileQueue,
             ]
         ] = None,
     ) -> int:
@@ -506,12 +505,12 @@ def _execute(
         self, oper: str,
         params: Optional[Union[Sequence[Any], Dict[str, Any]]] = None,
         is_callproc: bool = False,
-        infile_stream: Optional[
+        infile_stream: Optional[  # type: ignore
             Union[
                 io.RawIOBase,
                 io.TextIOBase,
                 Iterable[Union[bytes, str]],
-                queue.Queue[Union[bytes, str]],
+                connection.InfileQueue,
             ]
         ] = None,
     ) -> int: