Update logging

pennsignals · Jun 2, 2020 · 0e34741 · 0e34741
1 parent d4d8bfc
commit 0e34741
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/src/dsdk/mongo.py b/src/dsdk/mongo.py
@@ -157,8 +157,13 @@ def store_evidence(self, batch: Batch, *args, **kwargs) -> None:
             df.drop(columns=["batch_id"], inplace=True)
 
 
-OPEN = '{"key": "mongo.open", "database": "%s",  "is_master": "%s" }'
-CLOSE = '{"key": "mongo.close", "database": "%s"}'
+OPEN = "".join(
+    ("{", ", ".join(('"key": "mongo.open"', '"database": "%s"',)), "}",)
+)
+
+CLOSE = "".join(
+    ("{", ", ".join(('"key": "mongo.close"', '"database": "%s"',)), "}",)
+)
 
 
 @contextmanager
@@ -186,8 +191,8 @@ def open_database(
     ) as client:
         database = client.get_database()
         # force lazy connection open
-        is_master = client.admin.command("ismaster")
-        logger.info(OPEN, database.name, is_master)
+        client.admin.command("ismaster")
+        logger.info(OPEN, database.name)
         try:
             yield database
         finally:

diff --git a/src/dsdk/mssql.py b/src/dsdk/mssql.py
@@ -73,9 +73,9 @@ def _inject_mssql_uri(mssql_uri: str) -> str:
             type=_inject_mssql_uri,
         )
 
-    OPEN = "".join(("{", ", ".join(('"key": "mssql.open"')), "}"))
+    OPEN = "".join(("{", ", ".join(('"key": "mssql.open"',)), "}"))
 
-    CLOSE = "".join(("{", ", ".join(('"key": "mssql.close"')), "}"))
+    CLOSE = "".join(("{", ", ".join(('"key": "mssql.close"',)), "}"))
 
     CONNECT = """
 select 1 as n