restore py39 compat

Author: blink1073

gridfs/asynchronous/grid_file.py

@@ -46,6 +46,7 @@
 from pymongo.asynchronous.collection import AsyncCollection
 from pymongo.asynchronous.cursor import AsyncCursor
 from pymongo.asynchronous.database import AsyncDatabase
+from pymongo.asynchronous.helpers import anext
 from pymongo.common import validate_string
 from pymongo.errors import (
     BulkWriteError,

gridfs/synchronous/grid_file.py

@@ -57,6 +57,7 @@
 from pymongo.synchronous.collection import Collection
 from pymongo.synchronous.cursor import Cursor
 from pymongo.synchronous.database import Database
+from pymongo.synchronous.helpers import next
 
 _IS_SYNC = True
 

pymongo/asynchronous/cursor.py

@@ -37,6 +37,7 @@
 from bson.code import Code
 from bson.son import SON
 from pymongo import _csot, helpers_shared
+from pymongo.asynchronous.helpers import anext
 from pymongo.collation import validate_collation_or_none
 from pymongo.common import (
     validate_is_document_type,

pymongo/asynchronous/helpers.py

@@ -16,7 +16,9 @@
 from __future__ import annotations
 
 import asyncio
+import builtins
 import socket
+import sys
 from typing import (
     Any,
     Callable,
@@ -84,3 +86,17 @@ async def _getaddrinfo(
         return await loop.getaddrinfo(host, port, **kwargs)  # type: ignore[return-value]
     else:
         return socket.getaddrinfo(host, port, **kwargs)
+
+
+if sys.version_info >= (3, 10):
+    anext = builtins.anext
+    aiter = builtins.aiter
+else:
+
+    async def anext(cls: Any) -> Any:
+        """Compatibility function until we drop 3.9 support: https://docs.python.org/3/library/functions.html#anext."""
+        return await cls.__anext__()
+
+    def aiter(cls: Any) -> Any:
+        """Compatibility function until we drop 3.9 support: https://docs.python.org/3/library/functions.html#anext."""
+        return cls.__aiter__()

pymongo/synchronous/cursor.py

@@ -55,6 +55,7 @@
     _RawBatchQuery,
 )
 from pymongo.response import PinnedResponse
+from pymongo.synchronous.helpers import next
 from pymongo.typings import _Address, _CollationIn, _DocumentOut, _DocumentType
 from pymongo.write_concern import validate_boolean
 

pymongo/synchronous/helpers.py

@@ -16,7 +16,9 @@
 from __future__ import annotations
 
 import asyncio
+import builtins
 import socket
+import sys
 from typing import (
     Any,
     Callable,
@@ -84,3 +86,17 @@ def _getaddrinfo(
         return loop.getaddrinfo(host, port, **kwargs)  # type: ignore[return-value]
     else:
         return socket.getaddrinfo(host, port, **kwargs)
+
+
+if sys.version_info >= (3, 10):
+    next = builtins.next
+    iter = builtins.iter
+else:
+
+    def next(cls: Any) -> Any:
+        """Compatibility function until we drop 3.9 support: https://docs.python.org/3/library/functions.html#next."""
+        return cls.__next__()
+
+    def iter(cls: Any) -> Any:
+        """Compatibility function until we drop 3.9 support: https://docs.python.org/3/library/functions.html#next."""
+        return cls.__iter__()

test/asynchronous/test_change_stream.py

@@ -48,6 +48,7 @@
 from bson.raw_bson import DEFAULT_RAW_BSON_OPTIONS, RawBSONDocument
 from pymongo import AsyncMongoClient
 from pymongo.asynchronous.command_cursor import AsyncCommandCursor
+from pymongo.asynchronous.helpers import anext
 from pymongo.errors import (
     InvalidOperation,
     OperationFailure,

test/asynchronous/test_client.py

@@ -92,6 +92,7 @@
 from pymongo.asynchronous.command_cursor import AsyncCommandCursor
 from pymongo.asynchronous.cursor import AsyncCursor, CursorType
 from pymongo.asynchronous.database import AsyncDatabase
+from pymongo.asynchronous.helpers import anext
 from pymongo.asynchronous.mongo_client import AsyncMongoClient
 from pymongo.asynchronous.pool import (
     AsyncConnection,

test/asynchronous/test_collation.py

@@ -21,6 +21,7 @@
 from test.utils_shared import EventListener, OvertCommandListener
 from typing import Any
 
+from pymongo.asynchronous.helpers import anext
 from pymongo.collation import (
     Collation,
     CollationAlternate,

test/asynchronous/test_collection.py

@@ -25,6 +25,7 @@
 from typing import Any, Iterable, no_type_check
 
 from pymongo.asynchronous.database import AsyncDatabase
+from pymongo.asynchronous.helpers import anext
 
 sys.path[0:0] = [""]