fix typing -- i was being so silly earlier

Author: sleepyStick

pymongo/asynchronous/bulk.py

@@ -243,7 +243,9 @@ def add_delete(
     def gen_ordered(
         self,
         requests: Iterable[Any],
-        process: Callable[[Union[_DocumentType, RawBSONDocument, _WriteOp]], bool],
+        process: Union[
+            Callable[[_WriteOp], bool], Callable[[Union[_DocumentType, RawBSONDocument]], bool]
+        ],
     ) -> Iterator[_Run]:
         """Generate batches of operations, batched by type of
         operation, in the order **provided**.
@@ -266,7 +268,9 @@ def gen_ordered(
     def gen_unordered(
         self,
         requests: Iterable[Any],
-        process: Callable[[Union[_DocumentType, RawBSONDocument, _WriteOp]], bool],
+        process: Union[
+            Callable[[_WriteOp], bool], Callable[[Union[_DocumentType, RawBSONDocument]], bool]
+        ],
     ) -> Iterator[_Run]:
         """Generate batches of operations, batched by type of
         operation, in arbitrary order.
@@ -787,7 +791,9 @@ async def validate_batch(self, conn: AsyncConnection, write_concern: WriteConcer
     async def execute(
         self,
         generator: Iterable[Any],
-        process: Callable[[Union[_DocumentType, RawBSONDocument, _WriteOp]], bool],
+        process: Union[
+            Callable[[_WriteOp], bool], Callable[[Union[_DocumentType, RawBSONDocument]], bool]
+        ],
         write_concern: WriteConcern,
         session: Optional[AsyncClientSession],
         operation: str,

pymongo/asynchronous/collection.py

@@ -785,7 +785,7 @@ async def bulk_write(
 
         write_concern = self._write_concern_for(session)
 
-        def process_for_bulk(request: Union[_DocumentType, RawBSONDocument, _WriteOp]) -> bool:
+        def process_for_bulk(request: _WriteOp) -> bool:
             try:
                 return request._add_to_bulk(blk)
             except AttributeError:
@@ -964,7 +964,7 @@ async def insert_many(
             raise TypeError("documents must be a non-empty list")
         inserted_ids: list[ObjectId] = []
 
-        def process_for_bulk(document: Union[_DocumentType, RawBSONDocument, _WriteOp]) -> bool:
+        def process_for_bulk(document: Union[_DocumentType, RawBSONDocument]) -> bool:
             """A generator that validates documents and handles _ids."""
             common.validate_is_document_type("document", document)
             if not isinstance(document, RawBSONDocument):

pymongo/synchronous/bulk.py

@@ -243,7 +243,9 @@ def add_delete(
     def gen_ordered(
         self,
         requests: Iterable[Any],
-        process: Callable[[Union[_DocumentType, RawBSONDocument, _WriteOp]], bool],
+        process: Union[
+            Callable[[_WriteOp], bool], Callable[[Union[_DocumentType, RawBSONDocument]], bool]
+        ],
     ) -> Iterator[_Run]:
         """Generate batches of operations, batched by type of
         operation, in the order **provided**.
@@ -266,7 +268,9 @@ def gen_ordered(
     def gen_unordered(
         self,
         requests: Iterable[Any],
-        process: Callable[[Union[_DocumentType, RawBSONDocument, _WriteOp]], bool],
+        process: Union[
+            Callable[[_WriteOp], bool], Callable[[Union[_DocumentType, RawBSONDocument]], bool]
+        ],
     ) -> Iterator[_Run]:
         """Generate batches of operations, batched by type of
         operation, in arbitrary order.
@@ -787,7 +791,9 @@ def validate_batch(self, conn: Connection, write_concern: WriteConcern) -> None:
     def execute(
         self,
         generator: Iterable[Any],
-        process: Callable[[Union[_DocumentType, RawBSONDocument, _WriteOp]], bool],
+        process: Union[
+            Callable[[_WriteOp], bool], Callable[[Union[_DocumentType, RawBSONDocument]], bool]
+        ],
         write_concern: WriteConcern,
         session: Optional[ClientSession],
         operation: str,

pymongo/synchronous/collection.py

@@ -784,7 +784,7 @@ def bulk_write(
 
         write_concern = self._write_concern_for(session)
 
-        def process_for_bulk(request: Union[_DocumentType, RawBSONDocument, _WriteOp]) -> bool:
+        def process_for_bulk(request: _WriteOp) -> bool:
             try:
                 return request._add_to_bulk(blk)
             except AttributeError:
@@ -963,7 +963,7 @@ def insert_many(
             raise TypeError("documents must be a non-empty list")
         inserted_ids: list[ObjectId] = []
 
-        def process_for_bulk(document: Union[_DocumentType, RawBSONDocument, _WriteOp]) -> bool:
+        def process_for_bulk(document: Union[_DocumentType, RawBSONDocument]) -> bool:
             """A generator that validates documents and handles _ids."""
             common.validate_is_document_type("document", document)
             if not isinstance(document, RawBSONDocument):

test/test_typing.py

@@ -428,13 +428,9 @@ def test_typeddict_document_type_insertion(self) -> None:
     def test_bulk_write_document_type_insertion(self):
         client: MongoClient[MovieWithId] = MongoClient()
         coll: Collection[MovieWithId] = client.test.test
-        coll.bulk_write(
-            [InsertOne(Movie({"name": "THX-1138", "year": 1971}))]  # type:ignore[arg-type]
-        )
+        coll.bulk_write([InsertOne(Movie({"name": "THX-1138", "year": 1971}))])
         mov_dict = {"_id": ObjectId(), "name": "THX-1138", "year": 1971}
-        coll.bulk_write(
-            [InsertOne(mov_dict)]  # type:ignore[arg-type]
-        )
+        coll.bulk_write([InsertOne(mov_dict)])
         coll.bulk_write(
             [
                 InsertOne({"_id": ObjectId(), "name": "THX-1138", "year": 1971})  # pyright: ignore
@@ -445,13 +441,9 @@ def test_bulk_write_document_type_insertion(self):
     def test_bulk_write_document_type_replacement(self):
         client: MongoClient[MovieWithId] = MongoClient()
         coll: Collection[MovieWithId] = client.test.test
-        coll.bulk_write(
-            [ReplaceOne({}, Movie({"name": "THX-1138", "year": 1971}))]  # type:ignore[arg-type]
-        )
+        coll.bulk_write([ReplaceOne({}, Movie({"name": "THX-1138", "year": 1971}))])
         mov_dict = {"_id": ObjectId(), "name": "THX-1138", "year": 1971}
-        coll.bulk_write(
-            [ReplaceOne({}, mov_dict)]  # type:ignore[arg-type]
-        )
+        coll.bulk_write([ReplaceOne({}, mov_dict)])
         coll.bulk_write(
             [
                 ReplaceOne({}, {"_id": ObjectId(), "name": "THX-1138", "year": 1971})  # pyright: ignore