feat: support python 3.9+ (#36)

Author: jy-tan

.github/workflows/ci.yml

@@ -20,7 +20,7 @@ jobs:
           version: "latest"
 
       - name: Setup Python
-        run: uv python install 3.12
+        run: uv python install 3.9
 
       - name: Install dependencies
         run: uv sync --all-extras
@@ -44,7 +44,7 @@ jobs:
           version: "latest"
 
       - name: Setup Python
-        run: uv python install 3.12
+        run: uv python install 3.9
 
       - name: Install dependencies
         run: uv sync --all-extras
@@ -53,8 +53,11 @@ jobs:
         run: uv run ty check drift/ tests/
 
   test:
-    name: Unit Tests
+    name: Unit Tests (Python ${{ matrix.python-version }})
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.9", "3.14"]
     steps:
       - name: Checkout
         uses: actions/checkout@v4
@@ -65,7 +68,7 @@ jobs:
           version: "latest"
 
       - name: Setup Python
-        run: uv python install 3.12
+        run: uv python install ${{ matrix.python-version }}
 
       - name: Install dependencies
         run: uv sync --all-extras
@@ -86,7 +89,7 @@ jobs:
           version: "latest"
 
       - name: Setup Python
-        run: uv python install 3.12
+        run: uv python install 3.9
 
       - name: Install dependencies
         run: uv sync --all-extras

.github/workflows/e2e.yml

@@ -54,7 +54,7 @@ jobs:
           version: "latest"
 
       - name: Setup Python
-        run: uv python install 3.12
+        run: uv python install 3.9
 
       - name: Setup Docker Buildx
         uses: docker/setup-buildx-action@v3

README.md

@@ -39,7 +39,7 @@ For comprehensive guides and API reference, visit our [full documentation](https
 
 ## Requirements
 
-- Python 3.12+
+- Python 3.9+
 
 Tusk Drift currently supports the following packages and versions:
 

drift/core/batch_processor.py

@@ -197,7 +197,7 @@ def _export_batch(self) -> None:
                     finally:
                         loop.close()
                 else:
-                    adapter.export_spans(batch)  # type: ignore
+                    adapter.export_spans(batch)
 
                 latency_ms = (time.monotonic() - start_time) * 1000
                 self._metrics.record_spans_exported(len(batch))

drift/core/drift_sdk.py

@@ -397,7 +397,7 @@ def _init_auto_instrumentations(self) -> None:
             pass
 
         try:
-            import urllib3  # type: ignore[unresolved-import]
+            import urllib3
 
             from ..instrumentation.urllib3 import Urllib3Instrumentation
 
@@ -452,7 +452,7 @@ def _init_auto_instrumentations(self) -> None:
             pass
 
         try:
-            import django  # type: ignore[unresolved-import]
+            import django
 
             from ..instrumentation.django import DjangoInstrumentation
 

drift/core/resilience.py

@@ -13,6 +13,9 @@
 from collections.abc import Awaitable, Callable
 from dataclasses import dataclass
 from enum import Enum
+from typing import TypeVar
+
+T = TypeVar("T")
 
 logger = logging.getLogger(__name__)
 
@@ -51,7 +54,7 @@ def calculate_backoff_delay(
     return delay
 
 
-async def retry_async[T](
+async def retry_async(
     operation: Callable[[], Awaitable[T]],
     config: RetryConfig | None = None,
     retryable_exceptions: tuple[type[Exception], ...] = (Exception,),

drift/core/span_serialization.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from datetime import UTC, datetime, timedelta
+from datetime import datetime, timedelta, timezone
 from typing import Any
 
 from betterproto.lib.google.protobuf import Struct as ProtoStruct
@@ -114,7 +114,7 @@ def clean_span_to_proto(span: CleanSpanData) -> ProtoSpan:
         is_root_span=span.is_root_span,
         timestamp=datetime.fromtimestamp(
             span.timestamp.seconds + span.timestamp.nanos / 1_000_000_000,
-            tz=UTC,
+            tz=timezone.utc,
         ),
         duration=timedelta(
             seconds=span.duration.seconds,

drift/core/tracing/adapters/api.py

@@ -14,8 +14,10 @@
 import gzip
 import logging
 from dataclasses import dataclass
-from datetime import UTC, datetime, timedelta
-from typing import TYPE_CHECKING, Any, override
+from datetime import datetime, timedelta, timezone
+from typing import TYPE_CHECKING, Any
+
+from typing_extensions import override
 
 from ...resilience import (
     CircuitBreaker,
@@ -270,7 +272,7 @@ def _transform_span_to_protobuf(self, clean_span: CleanSpanData) -> Any:
 
         timestamp = datetime.fromtimestamp(
             clean_span.timestamp.seconds + clean_span.timestamp.nanos / 1_000_000_000,
-            tz=UTC,
+            tz=timezone.utc,
         )
 
         duration = timedelta(
@@ -339,8 +341,8 @@ def convert_json_schema(sdk_schema: Any) -> Any:
                 type=type_value,
                 properties=proto_properties,
                 items=proto_items,
-                encoding=encoding_value,  # type: ignore[arg-type]
-                decoded_type=decoded_type_value,  # type: ignore[arg-type]
+                encoding=encoding_value,
+                decoded_type=decoded_type_value,
                 match_importance=sdk_schema.match_importance,
             )
 
@@ -370,7 +372,7 @@ def convert_json_schema(sdk_schema: Any) -> Any:
             timestamp=timestamp,
             duration=duration,
             is_root_span=clean_span.is_root_span,
-            metadata=metadata_struct,  # type: ignore[arg-type]
+            metadata=metadata_struct,
         )
 
 

drift/core/tracing/adapters/filesystem.py

@@ -6,9 +6,11 @@
 import logging
 from collections import OrderedDict
 from dataclasses import asdict
-from datetime import UTC, datetime
+from datetime import datetime, timezone
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, override
+from typing import TYPE_CHECKING, Any
+
+from typing_extensions import override
 
 from .base import ExportResult, SpanExportAdapter
 
@@ -63,7 +65,7 @@ def _get_or_create_file_path(self, trace_id: str) -> Path:
             return self._trace_file_map[trace_id]
 
         # Create new file with timestamp prefix
-        iso_timestamp = datetime.now(UTC).isoformat().replace(":", "-").replace(".", "-")
+        iso_timestamp = datetime.now(timezone.utc).isoformat().replace(":", "-").replace(".", "-")
         file_path = self._base_directory / f"{iso_timestamp}_trace_{trace_id}.jsonl"
         self._trace_file_map[trace_id] = file_path
 

drift/core/tracing/adapters/memory.py

@@ -2,7 +2,9 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, override
+from typing import TYPE_CHECKING
+
+from typing_extensions import override
 
 from .base import ExportResult, SpanExportAdapter