Add PyMySQL dbapi commenter case

tammy-baylis-swi · Oct 9, 2024 · b0a9efd · b0a9efd
1 parent 881f4af
commit b0a9efd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...n/opentelemetry-instrumentation-dbapi/src/opentelemetry/instrumentation/dbapi/__init__.py b/...n/opentelemetry-instrumentation-dbapi/src/opentelemetry/instrumentation/dbapi/__init__.py
@@ -410,7 +410,7 @@ def get_statement(self, cursor, args):  # pylint: disable=no-self-use
             return statement.decode("utf8", "replace")
         return statement
 
-    def traced_execution(
+    def traced_execution(  # pylint: disable=too-many-branches
         self,
         cursor,
         query_method: typing.Callable[..., typing.Any],
@@ -479,6 +479,10 @@ def traced_execution(
                             mysqlc_version = (
                                 self._db_api_integration.connect_module._mysql.get_client_info()
                             )
+                        if db_driver == "pymysql":
+                            mysqlc_version = (
+                                self._db_api_integration.connect_module.get_client_info()
+                            )
 
                         commenter_data.update(
                             {