Skip to content

Commit 47cdaa9

Browse files
lint
1 parent cec3a86 commit 47cdaa9

File tree

1 file changed

+25
-12
lines changed
  • instrumentation/opentelemetry-instrumentation-dbapi/src/opentelemetry/instrumentation/dbapi

1 file changed

+25
-12
lines changed

instrumentation/opentelemetry-instrumentation-dbapi/src/opentelemetry/instrumentation/dbapi/__init__.py

Lines changed: 25 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353
)
5454
from opentelemetry.semconv.trace import SpanAttributes
5555
from opentelemetry.trace import SpanKind, TracerProvider, get_tracer
56-
from opentelemetry.util._importlib_metadata import version
56+
from opentelemetry.util._importlib_metadata import version as util_version
5757

5858
_DB_DRIVER_ALIASES = {
5959
"MySQLdb": "mysqlclient",
@@ -432,12 +432,18 @@ def traced_execution(
432432
if args and self._commenter_enabled:
433433
try:
434434
args_list = list(args)
435-
db_driver = self._db_api_integration.connect_module.__name__
435+
db_driver = (
436+
self._db_api_integration.connect_module.__name__
437+
)
436438
db_version = ""
437-
if db_driver in _DB_DRIVER_ALIASES.keys():
438-
db_version = version(_DB_DRIVER_ALIASES[db_driver])
439+
if db_driver in _DB_DRIVER_ALIASES:
440+
db_version = util_version(
441+
_DB_DRIVER_ALIASES[db_driver]
442+
)
439443
else:
440-
db_version = self._db_api_integration.connect_module.__version__
444+
db_version = (
445+
self._db_api_integration.connect_module.__version__
446+
)
441447

442448
commenter_data = {
443449
"db_driver": f"{db_driver}:{db_version.split(' ')[0]}",
@@ -446,9 +452,14 @@ def traced_execution(
446452
"driver_paramstyle": self._connect_module.paramstyle,
447453
}
448454

449-
if self._db_api_integration.database_system == "postgresql":
455+
if (
456+
self._db_api_integration.database_system
457+
== "postgresql"
458+
):
450459
if hasattr(self._connect_module, "__libpq_version__"):
451-
libpq_version = self._connect_module.__libpq_version__
460+
libpq_version = (
461+
self._connect_module.__libpq_version__
462+
)
452463
else:
453464
libpq_version = (
454465
self._connect_module.pq.__build_version__
@@ -458,21 +469,23 @@ def traced_execution(
458469
"libpq_version": libpq_version,
459470
}
460471
)
461-
elif self._db_api_integration.database_system == "mysql":
472+
elif self._db_api_integration.database_system == "mysql":
462473
mysqlc_version = ""
463474
if db_driver == "mysql.connector":
464-
mysqlc_version = cursor._cnx._cmysql.get_client_info()
475+
mysqlc_version = (
476+
cursor._cnx._cmysql.get_client_info()
477+
)
465478
if db_driver == "MySQLdb":
466-
mysqlc_version = self._db_api_integration.connect_module._mysql.get_client_info()
479+
mysqlc_version = (
480+
self._db_api_integration.connect_module._mysql.get_client_info()
481+
)
467482

468483
commenter_data.update(
469484
{
470485
"mysql_client_version": mysqlc_version,
471486
}
472487
)
473488

474-
_logger.debug("Using commenter_data: %s", commenter_data)
475-
476489
if self._commenter_options.get(
477490
"opentelemetry_values", True
478491
):

0 commit comments

Comments
 (0)