From 47cdaa95e85782b9dffbd457e27436d51004573a Mon Sep 17 00:00:00 2001
From: tammy-baylis-swi <tammy.baylis@solarwinds.com>
Date: Wed, 9 Oct 2024 15:13:31 -0700
Subject: [PATCH] lint

---
 .../instrumentation/dbapi/__init__.py         | 37 +++++++++++++------
 1 file changed, 25 insertions(+), 12 deletions(-)

diff --git a/instrumentation/opentelemetry-instrumentation-dbapi/src/opentelemetry/instrumentation/dbapi/__init__.py b/instrumentation/opentelemetry-instrumentation-dbapi/src/opentelemetry/instrumentation/dbapi/__init__.py
index 5f053938e8..0c256bc69d 100644
--- a/instrumentation/opentelemetry-instrumentation-dbapi/src/opentelemetry/instrumentation/dbapi/__init__.py
+++ b/instrumentation/opentelemetry-instrumentation-dbapi/src/opentelemetry/instrumentation/dbapi/__init__.py
@@ -53,7 +53,7 @@
 )
 from opentelemetry.semconv.trace import SpanAttributes
 from opentelemetry.trace import SpanKind, TracerProvider, get_tracer
-from opentelemetry.util._importlib_metadata import version
+from opentelemetry.util._importlib_metadata import version as util_version
 
 _DB_DRIVER_ALIASES = {
     "MySQLdb": "mysqlclient",
@@ -432,12 +432,18 @@ def traced_execution(
             if args and self._commenter_enabled:
                 try:
                     args_list = list(args)
-                    db_driver = self._db_api_integration.connect_module.__name__
+                    db_driver = (
+                        self._db_api_integration.connect_module.__name__
+                    )
                     db_version = ""
-                    if db_driver in _DB_DRIVER_ALIASES.keys():
-                        db_version = version(_DB_DRIVER_ALIASES[db_driver])
+                    if db_driver in _DB_DRIVER_ALIASES:
+                        db_version = util_version(
+                            _DB_DRIVER_ALIASES[db_driver]
+                        )
                     else:
-                        db_version = self._db_api_integration.connect_module.__version__
+                        db_version = (
+                            self._db_api_integration.connect_module.__version__
+                        )
 
                     commenter_data = {
                         "db_driver": f"{db_driver}:{db_version.split(' ')[0]}",
@@ -446,9 +452,14 @@ def traced_execution(
                         "driver_paramstyle": self._connect_module.paramstyle,
                     }
 
-                    if self._db_api_integration.database_system == "postgresql":
+                    if (
+                        self._db_api_integration.database_system
+                        == "postgresql"
+                    ):
                         if hasattr(self._connect_module, "__libpq_version__"):
-                            libpq_version = self._connect_module.__libpq_version__
+                            libpq_version = (
+                                self._connect_module.__libpq_version__
+                            )
                         else:
                             libpq_version = (
                                 self._connect_module.pq.__build_version__
@@ -458,12 +469,16 @@ def traced_execution(
                                 "libpq_version": libpq_version,
                             }
                         )
-                    elif self._db_api_integration.database_system == "mysql":                       
+                    elif self._db_api_integration.database_system == "mysql":
                         mysqlc_version = ""
                         if db_driver == "mysql.connector":
-                            mysqlc_version = cursor._cnx._cmysql.get_client_info()
+                            mysqlc_version = (
+                                cursor._cnx._cmysql.get_client_info()
+                            )
                         if db_driver == "MySQLdb":
-                            mysqlc_version = self._db_api_integration.connect_module._mysql.get_client_info()
+                            mysqlc_version = (
+                                self._db_api_integration.connect_module._mysql.get_client_info()
+                            )
 
                         commenter_data.update(
                             {
@@ -471,8 +486,6 @@ def traced_execution(
                             }
                         )
 
-                    _logger.debug("Using commenter_data: %s", commenter_data)
-
                     if self._commenter_options.get(
                         "opentelemetry_values", True
                     ):