Skip to content

Commit b6676d8

Browse files
committed
Address comments
1 parent b9fa54e commit b6676d8

File tree

6 files changed

+17
-22
lines changed

6 files changed

+17
-22
lines changed

pymongo/asynchronous/mongo_client.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -830,10 +830,7 @@ def __init__(
830830
# Username and password passed as kwargs override user info in URI.
831831
username = opts.get("username", username)
832832
password = opts.get("password", password)
833-
if _IS_SYNC:
834-
self._options = options = ClientOptions(username, password, dbase, opts, False)
835-
else:
836-
self._options = options = ClientOptions(username, password, dbase, opts, True)
833+
self._options = options = ClientOptions(username, password, dbase, opts, not _IS_SYNC)
837834

838835
self._default_database_name = dbase
839836
self._lock = _ALock(_create_lock())

pymongo/client_options.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def _parse_pool_options(
148148
password: str,
149149
database: Optional[str],
150150
options: Mapping[str, Any],
151-
async_client: bool,
151+
is_async: bool,
152152
) -> PoolOptions:
153153
"""Parse connection pool options."""
154154
credentials = _parse_credentials(username, password, database, options)
@@ -187,7 +187,7 @@ def _parse_pool_options(
187187
server_api=server_api,
188188
load_balanced=load_balanced,
189189
credentials=credentials,
190-
async_client=async_client,
190+
is_async=is_async,
191191
)
192192

193193

@@ -205,7 +205,7 @@ def __init__(
205205
password: str,
206206
database: Optional[str],
207207
options: Mapping[str, Any],
208-
async_client: bool = False,
208+
is_async: bool = False,
209209
):
210210
self.__options = options
211211
self.__codec_options = _parse_codec_options(options)
@@ -216,9 +216,7 @@ def __init__(
216216
self.__server_selection_timeout = options.get(
217217
"serverselectiontimeoutms", common.SERVER_SELECTION_TIMEOUT
218218
)
219-
self.__pool_options = _parse_pool_options(
220-
username, password, database, options, async_client
221-
)
219+
self.__pool_options = _parse_pool_options(username, password, database, options, is_async)
222220
self.__read_preference = _parse_read_preference(options)
223221
self.__replica_set_name = options.get("replicaset")
224222
self.__write_concern = _parse_write_concern(options)

pymongo/pool_options.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -331,7 +331,7 @@ def __init__(
331331
server_api: Optional[ServerApi] = None,
332332
load_balanced: Optional[bool] = None,
333333
credentials: Optional[MongoCredential] = None,
334-
async_client: Optional[bool] = False,
334+
is_async: Optional[bool] = False,
335335
):
336336
self.__max_pool_size = max_pool_size
337337
self.__min_pool_size = min_pool_size
@@ -351,8 +351,6 @@ def __init__(
351351
self.__load_balanced = load_balanced
352352
self.__credentials = credentials
353353
self.__metadata = copy.deepcopy(_METADATA)
354-
if async_client:
355-
self.__metadata["driver"]["name"] = "AsyncPyMongo"
356354

357355
if appname:
358356
self.__metadata["application"] = {"name": appname}
@@ -365,6 +363,11 @@ def __init__(
365363
# },
366364
# 'platform': 'CPython 3.8.0|MyPlatform'
367365
# }
366+
if is_async:
367+
self.__metadata["driver"]["name"] = "{}|{}".format(
368+
self.__metadata["driver"]["name"],
369+
"async",
370+
)
368371
if driver:
369372
if driver.name:
370373
self.__metadata["driver"]["name"] = "{}|{}".format(

pymongo/synchronous/mongo_client.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -829,10 +829,7 @@ def __init__(
829829
# Username and password passed as kwargs override user info in URI.
830830
username = opts.get("username", username)
831831
password = opts.get("password", password)
832-
if _IS_SYNC:
833-
self._options = options = ClientOptions(username, password, dbase, opts, False)
834-
else:
835-
self._options = options = ClientOptions(username, password, dbase, opts, True)
832+
self._options = options = ClientOptions(username, password, dbase, opts, not _IS_SYNC)
836833

837834
self._default_database_name = dbase
838835
self._lock = _create_lock()

test/asynchronous/test_client.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -341,7 +341,7 @@ async def test_read_preference(self):
341341

342342
def test_metadata(self):
343343
metadata = copy.deepcopy(_METADATA)
344-
metadata["driver"]["name"] = "AsyncPyMongo"
344+
metadata["driver"]["name"] = "PyMongo|async"
345345
metadata["application"] = {"name": "foobar"}
346346
client = AsyncMongoClient("mongodb://foo:27017/?appname=foobar&connect=false")
347347
options = client.options
@@ -360,7 +360,7 @@ def test_metadata(self):
360360
self.assertRaises(TypeError, AsyncMongoClient, driver="abc")
361361
self.assertRaises(TypeError, AsyncMongoClient, driver=("Foo", "1", "a"))
362362
# Test appending to driver info.
363-
metadata["driver"]["name"] = "AsyncPyMongo|FooDriver"
363+
metadata["driver"]["name"] = "PyMongo|async|FooDriver"
364364
metadata["driver"]["version"] = "{}|1.2.3".format(_METADATA["driver"]["version"])
365365
client = AsyncMongoClient(
366366
"foo",
@@ -404,7 +404,7 @@ def test_metadata(self):
404404
@mock.patch.dict("os.environ", {ENV_VAR_K8S: "1"})
405405
def test_container_metadata(self):
406406
metadata = copy.deepcopy(_METADATA)
407-
metadata["driver"]["name"] = "AsyncPyMongo"
407+
metadata["driver"]["name"] = "PyMongo|async"
408408
metadata["env"] = {}
409409
metadata["env"]["container"] = {"orchestrator": "kubernetes"}
410410
client = AsyncMongoClient("mongodb://foo:27017/?appname=foobar&connect=false")
@@ -1940,7 +1940,7 @@ def test_sigstop_sigcont(self):
19401940
async def _test_handshake(self, env_vars, expected_env):
19411941
with patch.dict("os.environ", env_vars):
19421942
metadata = copy.deepcopy(_METADATA)
1943-
metadata["driver"]["name"] = "AsyncPyMongo"
1943+
metadata["driver"]["name"] = "PyMongo|async"
19441944
if expected_env is not None:
19451945
metadata["env"] = expected_env
19461946

tools/synchro.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@
9595
"async-transactions-ref": "transactions-ref",
9696
"async-snapshot-reads-ref": "snapshot-reads-ref",
9797
"default_async": "default",
98-
"AsyncPyMongo": "PyMongo",
98+
"PyMongo|async": "PyMongo",
9999
}
100100

101101
docstring_replacements: dict[tuple[str, str], str] = {

0 commit comments

Comments
 (0)