Skip to content

Commit 0ce55c7

Browse files
authored
Add DriverInfo to MongoClient creation (#165)
* add DriverInfo to MongoClient creation * fix tox warning * fix tox warning * fix tox warning * fix tox warning
1 parent a6c9ed5 commit 0ce55c7

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

flask_pymongo/__init__.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -34,11 +34,16 @@
3434
from pymongo import uri_parser
3535
from werkzeug.wsgi import wrap_file
3636
import pymongo
37+
# DriverInfo was added in PyMongo 3.7
38+
try:
39+
from pymongo.driver_info import DriverInfo
40+
except ImportError:
41+
DriverInfo = None
3742

43+
from flask_pymongo._version import __version__
3844
from flask_pymongo.helpers import BSONObjectIdConverter, JSONEncoder
3945
from flask_pymongo.wrappers import MongoClient
4046

41-
4247
DESCENDING = pymongo.DESCENDING
4348
"""Descending sort order."""
4449

@@ -109,6 +114,8 @@ def init_app(self, app, uri=None, *args, **kwargs):
109114
# Try to delay connecting, in case the app is loaded before forking, per
110115
# http://api.mongodb.com/python/current/faq.html#is-pymongo-fork-safe
111116
kwargs.setdefault("connect", False)
117+
if DriverInfo is not None:
118+
kwargs.setdefault("driver", DriverInfo("Flask-PyMongo", __version__))
112119

113120
self.cx = MongoClient(*args, **kwargs)
114121
if database_name:

0 commit comments

Comments
 (0)