fix: use typing.Union instead of |, which is Python 3.10+ only (closes #10)

Author: mpetazzoni

wherobots/db/connection.py

@@ -4,7 +4,7 @@
 import threading
 import uuid
 from dataclasses import dataclass
-from typing import Callable, Any
+from typing import Any, Callable, Union
 
 import cbor2
 import pyarrow
@@ -52,9 +52,9 @@ def __init__(
         self,
         ws: websockets.sync.client.ClientConnection,
         read_timeout: float = DEFAULT_READ_TIMEOUT_SECONDS,
-        results_format: ResultsFormat | None = None,
-        data_compression: DataCompression | None = None,
-        geometry_representation: GeometryRepresentation | None = None,
+        results_format: Union[ResultsFormat, None] = None,
+        data_compression: Union[DataCompression, None] = None,
+        geometry_representation: Union[GeometryRepresentation, None] = None,
     ):
         self.__ws = ws
         self.__read_timeout = read_timeout

wherobots/db/driver.py

@@ -10,6 +10,7 @@
 import requests
 import tenacity
 import threading
+from typing import Union
 import websockets.sync.client
 
 from .constants import (
@@ -45,9 +46,9 @@ def connect(
     region: Region = None,
     wait_timeout: float = DEFAULT_SESSION_WAIT_TIMEOUT_SECONDS,
     read_timeout: float = DEFAULT_READ_TIMEOUT_SECONDS,
-    results_format: ResultsFormat | None = None,
-    data_compression: DataCompression | None = None,
-    geometry_representation: GeometryRepresentation | None = None,
+    results_format: Union[ResultsFormat, None] = None,
+    data_compression: Union[DataCompression, None] = None,
+    geometry_representation: Union[GeometryRepresentation, None] = None,
 ) -> Connection:
     if not token and not api_key:
         raise ValueError("At least one of `token` or `api_key` is required")
@@ -145,9 +146,9 @@ def connect_direct(
     uri: str,
     headers: dict[str, str] = None,
     read_timeout: float = DEFAULT_READ_TIMEOUT_SECONDS,
-    results_format: ResultsFormat | None = None,
-    data_compression: DataCompression | None = None,
-    geometry_representation: GeometryRepresentation | None = None,
+    results_format: Union[ResultsFormat, None] = None,
+    data_compression: Union[DataCompression, None] = None,
+    geometry_representation: Union[GeometryRepresentation, None] = None,
 ) -> Connection:
     q = queue.SimpleQueue()
     uri_with_protocol = append_protocol(uri, PROTOCOL_VERSION)