Merge branch 'master' of github.com:mongodb/mongo-python-driver

Author: blink1073

doc/api/gridfs/asynchronous/grid_file.rst

@@ -0,0 +1,19 @@
+:mod:`grid_file` -- Async tools for representing files stored in GridFS
+=======================================================================
+
+.. automodule:: gridfs.asynchronous.grid_file
+   :synopsis: Async tools for representing files stored in GridFS
+
+   .. autoclass:: AsyncGridIn
+      :members:
+
+      .. autoattribute:: _id
+
+   .. autoclass:: AsyncGridOut
+      :members:
+
+      .. autoattribute:: _id
+      .. automethod:: __aiter__
+
+   .. autoclass:: AsyncGridOutCursor
+      :members:

doc/api/gridfs/asynchronous/index.rst

@@ -0,0 +1,18 @@
+:mod:`gridfs async` -- Async tools for working with GridFS
+==========================================================
+
+.. warning:: This API is currently in beta, meaning the classes, methods,
+   and behaviors described within may change before the full release.
+   If you come across any bugs during your use of this API,
+   please file a Jira ticket in the "Python Driver" project at https://jira.mongodb.org/browse/PYTHON.
+
+.. automodule:: gridfs.asynchronous
+   :synopsis: Async tools for working with GridFS
+   :members: AsyncGridFS, AsyncGridFSBucket
+
+Sub-modules:
+
+.. toctree::
+   :maxdepth: 2
+
+   grid_file

doc/api/gridfs/index.rst

@@ -8,7 +8,8 @@
 Sub-modules:
 
 .. toctree::
-   :maxdepth: 2
+   :maxdepth: 3
 
+   asynchronous/index
    errors
    grid_file

gridfs/asynchronous/__init__.py

@@ -0,0 +1,42 @@
+# Copyright 2009-present MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""GridFS is a specification for storing large objects in Mongo.
+
+The :mod:`gridfs` package is an implementation of GridFS on top of
+:mod:`pymongo`, exposing a file-like interface.
+
+.. seealso:: The MongoDB documentation on `gridfs <https://dochub.mongodb.org/core/gridfs>`_.
+"""
+from __future__ import annotations
+
+from gridfs.asynchronous.grid_file import (
+    AsyncGridFS,
+    AsyncGridFSBucket,
+    AsyncGridIn,
+    AsyncGridOut,
+    AsyncGridOutCursor,
+)
+from gridfs.errors import NoFile
+from gridfs.grid_file_shared import DEFAULT_CHUNK_SIZE
+
+__all__ = [
+    "AsyncGridFS",
+    "AsyncGridFSBucket",
+    "NoFile",
+    "DEFAULT_CHUNK_SIZE",
+    "AsyncGridIn",
+    "AsyncGridOut",
+    "AsyncGridOutCursor",
+]

test/performance/async_perf_test.py

@@ -362,6 +362,17 @@ async def do_task(self):
         await self.corpus.insert_many(self.documents, ordered=True)
 
 
+class TestSmallDocCollectionBulkInsert(SmallDocInsertTest, AsyncPyMongoTestCase):
+    async def asyncSetUp(self):
+        await super().asyncSetUp()
+        self.models = []
+        for doc in self.documents:
+            self.models.append(InsertOne(namespace="perftest.corpus", document=doc))
+
+    async def do_task(self):
+        await self.corpus.bulk_write(self.models, ordered=True)
+
+
 class TestSmallDocClientBulkInsert(SmallDocInsertTest, AsyncPyMongoTestCase):
     @async_client_context.require_version_min(8, 0, 0, -24)
     async def asyncSetUp(self):
@@ -412,6 +423,17 @@ async def do_task(self):
         await self.corpus.insert_many(self.documents, ordered=True)
 
 
+class TestLargeDocCollectionBulkInsert(LargeDocInsertTest, AsyncPyMongoTestCase):
+    async def asyncSetUp(self):
+        await super().asyncSetUp()
+        self.models = []
+        for doc in self.documents:
+            self.models.append(InsertOne(namespace="perftest.corpus", document=doc))
+
+    async def do_task(self):
+        await self.corpus.bulk_write(self.models, ordered=True)
+
+
 class TestLargeDocClientBulkInsert(LargeDocInsertTest, AsyncPyMongoTestCase):
     @async_client_context.require_version_min(8, 0, 0, -24)
     async def asyncSetUp(self):

test/performance/perf_test.py

@@ -443,6 +443,17 @@ def do_task(self):
         self.corpus.insert_many(self.documents, ordered=True)
 
 
+class TestSmallDocCollectionBulkInsert(SmallDocInsertTest, unittest.TestCase):
+    def setUp(self):
+        super().setUp()
+        self.models = []
+        for doc in self.documents:
+            self.models.append(InsertOne(namespace="perftest.corpus", document=doc))
+
+    def do_task(self):
+        self.corpus.bulk_write(self.models, ordered=True)
+
+
 class TestSmallDocClientBulkInsert(SmallDocInsertTest, unittest.TestCase):
     @client_context.require_version_min(8, 0, 0, -24)
     def setUp(self):
@@ -493,6 +504,17 @@ def do_task(self):
         self.corpus.insert_many(self.documents, ordered=True)
 
 
+class TestLargeDocCollectionBulkInsert(LargeDocInsertTest, unittest.TestCase):
+    def setUp(self):
+        super().setUp()
+        self.models = []
+        for doc in self.documents:
+            self.models.append(InsertOne(namespace="perftest.corpus", document=doc))
+
+    def do_task(self):
+        self.corpus.bulk_write(self.models, ordered=True)
+
+
 class TestLargeDocClientBulkInsert(LargeDocInsertTest, unittest.TestCase):
     @client_context.require_version_min(8, 0, 0, -24)
     def setUp(self):