Skip to content

Commit 79cb34a

Browse files
authored
PYTHON-5314 [v4.12] Fix default imports for modules that worked in v4.8 (#2300) (#2303)
1 parent c837846 commit 79cb34a

File tree

2 files changed

+14
-1
lines changed

2 files changed

+14
-1
lines changed

pymongo/__init__.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,14 @@
106106
from pymongo.write_concern import WriteConcern
107107

108108
# Public module compatibility imports
109-
import pymongo.uri_parser # noqa: F401 # isort: skip
109+
# isort: off
110+
from pymongo import uri_parser # noqa: F401
111+
from pymongo import change_stream # noqa: F401
112+
from pymongo import client_session # noqa: F401
113+
from pymongo import collection # noqa: F401
114+
from pymongo import command_cursor # noqa: F401
115+
from pymongo import database # noqa: F401
116+
# isort: on
110117

111118
version = __version__
112119
"""Current version of PyMongo."""

test/test_default_exports.py

+6
Original file line numberDiff line numberDiff line change
@@ -215,6 +215,12 @@ def test_pymongo_submodule_attributes(self):
215215
self.assertTrue(hasattr(pymongo, "uri_parser"))
216216
self.assertTrue(pymongo.uri_parser)
217217
self.assertTrue(pymongo.uri_parser.parse_uri)
218+
self.assertTrue(pymongo.change_stream)
219+
self.assertTrue(pymongo.client_session)
220+
self.assertTrue(pymongo.collection)
221+
self.assertTrue(pymongo.cursor)
222+
self.assertTrue(pymongo.command_cursor)
223+
self.assertTrue(pymongo.database)
218224

219225
def test_gridfs_imports(self):
220226
import gridfs

0 commit comments

Comments
 (0)