From 37ce6dd5cb8adaf144eadaad22deb42c94ba39a4 Mon Sep 17 00:00:00 2001 From: Alex Bevilacqua Date: Fri, 26 Sep 2025 12:52:02 -0400 Subject: [PATCH] Add support for DriverInfo in MongoClient Added conditional import for DriverInfo from PyMongo. --- djongo/database.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/djongo/database.py b/djongo/database.py index 86ad27de..c1908003 100644 --- a/djongo/database.py +++ b/djongo/database.py @@ -1,6 +1,11 @@ from logging import getLogger from pymongo import MongoClient - +# DriverInfo was added in PyMongo 3.7 +try: + from pymongo.driver_info import DriverInfo +except ImportError: + DriverInfo = None + logger = getLogger(__name__) clients = {} @@ -10,6 +15,10 @@ def connect(db, **kwargs): return clients[db] except KeyError: logger.debug('New MongoClient connection') + kwargs.setdefault("connect", False) + if DriverInfo is not None: + import djongo + kwargs.setdefault("driver", DriverInfo("djongo", djongo.__version__)) clients[db] = MongoClient(**kwargs, connect=False) return clients[db]