MOTOR-1335 AsyncIOMotorClient is not suscriptable (#293)

Author: blink1073

doc/requirements.rst

@@ -67,7 +67,7 @@ Use `the PyMongo compatibility matrix`_ to determine what MongoDB version is
 supported by PyMongo. Use the compatibility matrix above to determine what
 MongoDB version Motor supports.
 
-.. _the PyMongo compatibility matrix: https://mongodb.com/docs/drivers/pymongo#mongodb-compatibility
+.. _the PyMongo compatibility matrix: https://mongodb.com/docs/drivers/pymongo#compatibility
 
 Motor and Tornado
 `````````````````

motor/core.py

@@ -90,6 +90,11 @@ def __repr__(self):
 
 
 class AgnosticBaseProperties(AgnosticBase):
+    # Allow the use of generics at runtime.
+    @classmethod
+    def __class_getitem__(cls, key: str) -> object:
+        return cls
+
     codec_options = ReadOnlyProperty()
     read_preference = ReadOnlyProperty()
     read_concern = ReadOnlyProperty()
@@ -1376,6 +1381,11 @@ def get_io_loop(self):
 class AgnosticBaseCursor(AgnosticBase):
     """Base class for AgnosticCursor and AgnosticCommandCursor"""
 
+    # Allow the use of generics at runtime.
+    @classmethod
+    def __class_getitem__(cls, key: str) -> object:
+        return cls
+
     _async_close = AsyncRead(attr_name="close")
     _refresh = AsyncRead()
     address = ReadOnlyProperty()
@@ -1940,6 +1950,11 @@ class AgnosticChangeStream(AgnosticBase):
 
     resume_token = ReadOnlyProperty()
 
+    # Allow the use of generics at runtime.
+    @classmethod
+    def __class_getitem__(cls, key: str) -> object:
+        return cls
+
     def __init__(
         self,
         target,
@@ -2131,6 +2146,11 @@ class AgnosticClientEncryption(AgnosticBase):
     get_key_by_alt_name = AsyncCommand()
     remove_key_alt_name = AsyncCommand()
 
+    # Allow the use of generics at runtime.
+    @classmethod
+    def __class_getitem__(cls, key: str) -> object:
+        return cls
+
     def __init__(
         self,
         kms_providers,

synchro/synchrotest.py

@@ -341,7 +341,7 @@ def want_method(method, classname):
 
     # Run the tests from the pymongo target dir with our custom plugin.
     os.chdir(sys.argv[1])
-    code = pytest.main(sys.argv[2:], plugins=[SynchroPytestPlugin()])
+    code = pytest.main(sys.argv[2:] + ["-p", "no:warnings"], plugins=[SynchroPytestPlugin()])
 
     if code != 0:
         sys.exit(code)

test/check_runtime_types.py

@@ -0,0 +1,17 @@
+from typing import Any, Dict
+
+from motor.motor_asyncio import (
+    AsyncIOMotorChangeStream,
+    AsyncIOMotorClient,
+    AsyncIOMotorClientEncryption,
+    AsyncIOMotorCollection,
+    AsyncIOMotorCursor,
+    AsyncIOMotorDatabase,
+)
+
+client: AsyncIOMotorClient[Dict[str, Any]]
+db: AsyncIOMotorDatabase[Dict[str, Any]]
+cur: AsyncIOMotorCursor[Dict[str, Any]]
+coll: AsyncIOMotorCollection[Dict[str, Any]]
+cs: AsyncIOMotorChangeStream[Dict[str, Any]]
+enc: AsyncIOMotorClientEncryption[Dict[str, Any]]

tox.ini

@@ -161,3 +161,4 @@ commands =
     mypy --install-types --non-interactive motor
     mypy --install-types --non-interactive test/test_typing.py
     pytest test/test_mypy_fails.py
+    python test/check_runtime_types.py