Add Python 3.14 support (#629)

Author: MarshalX

.github/workflows/codegen.yml

@@ -18,16 +18,6 @@ jobs:
         shell: bash
 
     steps:
-      - name: Run Cimon (eBPF).
-        if: runner.os == 'Linux'
-        uses: cycodelabs/cimon-action@v0
-        with:
-          prevent: true
-          allowed-hosts: >
-            files.pythonhosted.org
-            install.python-poetry.org
-            pypi.org
-
       - name: Checkout repository.
         uses: actions/checkout@v4
 
@@ -39,7 +29,7 @@ jobs:
       - name: Setup Poetry.
         uses: snok/install-poetry@v1
         with:
-          version: 1.8.3
+          version: 2.2.1
 
       - name: Install dependencies.
         run: poetry install --no-interaction

.github/workflows/publish_release.yml

@@ -13,17 +13,6 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-      - name: Run Cimon (eBPF).
-        uses: cycodelabs/cimon-action@v0
-        with:
-          prevent: true
-          allowed-hosts: >
-            files.pythonhosted.org
-            install.python-poetry.org
-            pypi.org
-            upload.pypi.org
-            *.sigstore.dev
-
       - name: Checkout repository.
         uses: actions/checkout@v4
         with:
@@ -37,7 +26,7 @@ jobs:
       - name: Setup Poetry.
         uses: snok/install-poetry@v1
         with:
-          version: 1.8.3
+          version: 2.2.1
 
       - name: Install Poetry Plugin.
         run: poetry self add "poetry-dynamic-versioning[plugin]"

.github/workflows/ruff.yml

@@ -10,14 +10,6 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-      - name: Run Cimon (eBPF).
-        uses: cycodelabs/cimon-action@v0
-        with:
-          prevent: true
-          allowed-hosts: >
-            files.pythonhosted.org
-            pypi.org
-
       - name: Checkout repository.
         uses: actions/checkout@v4
 
@@ -29,10 +21,10 @@ jobs:
       - name: Run linter check.
         uses: astral-sh/ruff-action@v3
         with:
-          version: 0.7.0
+          version: 0.14.1
 
       - name: Run code style check.
         uses: astral-sh/ruff-action@v3
         with:
-          version: 0.7.0
+          version: 0.14.1
           args: "format --check"

.github/workflows/test.yml

@@ -10,7 +10,7 @@ jobs:
     strategy:
       matrix:
         os: [ macos-latest, ubuntu-latest, windows-latest ]
-        python-version: [ "3.9", "3.10", "3.11", "3.12", "3.13" ]
+        python-version: [ "3.9", "3.10", "3.11", "3.12", "3.13", "3.14" ]
 
     runs-on: ${{matrix.os}}
 
@@ -19,19 +19,6 @@ jobs:
         shell: bash
 
     steps:
-      - name: Run Cimon (eBPF).
-        if: runner.os == 'Linux'
-        uses: cycodelabs/cimon-action@v0
-        with:
-          prevent: true
-          allowed-hosts: >
-            files.pythonhosted.org
-            install.python-poetry.org
-            pypi.org
-            *.atproto.blue
-            *.marshal.dev
-            plc.directory
-
       - name: Checkout repository.
         uses: actions/checkout@v4
 
@@ -43,7 +30,7 @@ jobs:
       - name: Setup Poetry.
         uses: snok/install-poetry@v1
         with:
-          version: 1.8.3
+          version: 2.2.1
 
       - name: Install dependencies.
         run: poetry install --without docs

.github/workflows/update_lexicons.yml

@@ -13,15 +13,6 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-      - name: Run Cimon (eBPF).
-        uses: cycodelabs/cimon-action@v0
-        with:
-          prevent: true
-          allowed-hosts: >
-            files.pythonhosted.org
-            install.python-poetry.org
-            pypi.org
-
       - name: Checkout repository.
         uses: actions/checkout@v4
 

packages/atproto/__init__.py

@@ -31,7 +31,7 @@
     verify_jwt_async,
 )
 
-__all__ = [
+__all__ = [  # noqa: RUF022
     # client
     'AsyncClient',
     'Client',

packages/atproto_client/__init__.py

@@ -6,7 +6,7 @@
 __all__ = [
     'AsyncClient',
     'Client',
-    'models',
-    'SessionEvent',
     'Session',
+    'SessionEvent',
+    'models',
 ]

packages/atproto_client/models/dot_dict.py

@@ -134,13 +134,13 @@ def __reduce__(self):  # noqa: ANN204
     @staticmethod
     def __convert(obj: t.Any) -> t.Any:
         if isinstance(obj, dict):
-            return DotDict(t.cast(t.Dict[str, t.Any], obj))
+            return DotDict(t.cast('t.Dict[str, t.Any]', obj))
         if isinstance(obj, list):
-            return [DotDict.__convert(v) for v in t.cast(t.List[t.Any], obj)]
+            return [DotDict.__convert(v) for v in t.cast('t.List[t.Any]', obj)]
         if isinstance(obj, set):
-            return {DotDict.__convert(v) for v in t.cast(t.Set[t.Any], obj)}
+            return {DotDict.__convert(v) for v in t.cast('t.Set[t.Any]', obj)}
         if isinstance(obj, tuple):
-            return tuple(DotDict.__convert(v) for v in t.cast(t.Tuple[t.Any], obj))
+            return tuple(DotDict.__convert(v) for v in t.cast('t.Tuple[t.Any]', obj))
         return obj
 
 

packages/atproto_client/models/string_formats.py

@@ -3,15 +3,17 @@
 import re
 from datetime import datetime
 from functools import wraps
-from typing import Callable, Mapping, Set, Union, cast
+from typing import TYPE_CHECKING, Callable, Mapping, Set, Union, cast
 from urllib.parse import urlparse
 
 from atproto_core.exceptions import InvalidNsidError
 from atproto_core.nsid import validate_nsid as atproto_core_validate_nsid
 from pydantic import BeforeValidator, Field, ValidationInfo
-from pydantic_core import core_schema
 from typing_extensions import Annotated, Literal
 
+if TYPE_CHECKING:
+    from pydantic_core import core_schema
+
 _OPT_IN_KEY: Literal['strict_string_format'] = 'strict_string_format'
 
 # constants
@@ -82,7 +84,7 @@ def only_validate_if_strict(validate_fn: Callable[..., str]) -> Callable[..., st
     def wrapper(v: str, info: ValidationInfo) -> str:
         """Could likely be generalized to support arbitrary signatures."""
         if info and isinstance(info.context, Mapping) and info.context.get(_OPT_IN_KEY, False):
-            return cast(core_schema.WithInfoValidatorFunction, validate_fn)(v, info)
+            return cast('core_schema.WithInfoValidatorFunction', validate_fn)(v, info)
         return v
 
     return wrapper

packages/atproto_client/models/utils.py

@@ -87,15 +87,15 @@ def _get_or_create(
             return None
 
         # we are sure that this is dict because of check above
-        model_data = t.cast(t.Dict[str, t.Any], model_data)
+        model_data = t.cast('t.Dict[str, t.Any]', model_data)
 
         # resolve a record model by type and try to deserialize
         record_type: t.Any = model_data.get(_TYPE_SERVICE_FIELD)
         if record_type not in RECORD_TYPE_TO_MODEL_CLASS:
             return None
 
         # now we are sure that this is str because it's in RECORD_TYPE_TO_MODEL_CLASS
-        record_type = t.cast(str, record_type)
+        record_type = t.cast('str', record_type)
 
         return get_or_create(
             model_data,
@@ -120,15 +120,15 @@ def _get_or_create(
 def get_response_model(response: 'Response', model: t.Type[M]) -> M:
     if model is bool:
         # Could not be False? Because the exception with errors will be raised from the server
-        return t.cast(M, response.success)
+        return t.cast('M', response.success)
     if model is bytes:
-        return t.cast(M, response.content)
+        return t.cast('M', response.content)
 
     if not isinstance(response.content, dict):
         raise ModelError("Can't properly parse response model because JSON is expected in response")
 
     # casting to M because of enabled strict mode
-    return t.cast(M, get_or_create(response.content, model, strict=True))
+    return t.cast('M', get_or_create(response.content, model, strict=True))
 
 
 def get_model_as_dict(model: t.Union[DotDict, BlobRef, ModelBase]) -> t.Dict[str, t.Any]: