modify flask version

Author: allen-k1m

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

@@ -254,7 +254,7 @@ def response_hook(span: Span, status: str, response_headers: List):
 
 try:
     flask_version = flask.__version__
-except Exception as e:
+except AttributeError:
     try:
         from importlib import metadata
     except ImportError:
@@ -304,11 +304,11 @@ def get_default_span_name():
 
 
 def _rewrapped_app(
-        wsgi_app,
-        active_requests_counter,
-        duration_histogram,
-        response_hook=None,
-        excluded_urls=None,
+    wsgi_app,
+    active_requests_counter,
+    duration_histogram,
+    response_hook=None,
+    excluded_urls=None,
 ):
     def _wrapped_app(wrapped_app_environ, start_response):
         # We want to measure the time for route matching, etc.
@@ -326,8 +326,8 @@ def _wrapped_app(wrapped_app_environ, start_response):
 
         def _start_response(status, response_headers, *args, **kwargs):
             if flask.request and (
-                    excluded_urls is None
-                    or not excluded_urls.url_disabled(flask.request.url)
+                excluded_urls is None
+                or not excluded_urls.url_disabled(flask.request.url)
             ):
                 span = flask.request.environ.get(_ENVIRON_SPAN_KEY)
 
@@ -348,8 +348,8 @@ def _start_response(status, response_headers, *args, **kwargs):
                             SpanAttributes.HTTP_STATUS_CODE
                         ] = status_code
                     if (
-                            span.is_recording()
-                            and span.kind == trace.SpanKind.SERVER
+                        span.is_recording()
+                        and span.kind == trace.SpanKind.SERVER
                     ):
                         custom_attributes = otel_wsgi.collect_custom_response_headers_attributes(
                             response_headers
@@ -376,11 +376,11 @@ def _start_response(status, response_headers, *args, **kwargs):
 
 
 def _wrapped_before_request(
-        request_hook=None,
-        tracer=None,
-        excluded_urls=None,
-        enable_commenter=True,
-        commenter_options=None,
+    request_hook=None,
+    tracer=None,
+    excluded_urls=None,
+    enable_commenter=True,
+    commenter_options=None,
 ):
     def _before_request():
         if excluded_urls and excluded_urls.url_disabled(flask.request.url):
@@ -435,14 +435,14 @@ def _before_request():
                 if commenter_options.get("framework", True):
                     flask_info["framework"] = f"flask:{flask_version}"
                 if (
-                        commenter_options.get("controller", True)
-                        and flask.request.endpoint
+                    commenter_options.get("controller", True)
+                    and flask.request.endpoint
                 ):
                     flask_info["controller"] = flask.request.endpoint
                 if (
-                        commenter_options.get("route", True)
-                        and flask.request.url_rule
-                        and flask.request.url_rule.rule
+                    commenter_options.get("route", True)
+                    and flask.request.url_rule
+                    and flask.request.url_rule.rule
                 ):
                     flask_info["route"] = flask.request.url_rule.rule
             sqlcommenter_context = context.set_value(
@@ -454,7 +454,7 @@ def _before_request():
 
 
 def _wrapped_teardown_request(
-        excluded_urls=None,
+    excluded_urls=None,
 ):
     def _teardown_request(exc):
         # pylint: disable=E1101
@@ -595,14 +595,14 @@ def _uninstrument(self, **kwargs):
 
     @staticmethod
     def instrument_app(
-            app,
-            request_hook=None,
-            response_hook=None,
-            tracer_provider=None,
-            excluded_urls=None,
-            enable_commenter=True,
-            commenter_options=None,
-            meter_provider=None,
+        app,
+        request_hook=None,
+        response_hook=None,
+        tracer_provider=None,
+        excluded_urls=None,
+        enable_commenter=True,
+        commenter_options=None,
+        meter_provider=None,
     ):
         if not hasattr(app, "_is_instrumented_by_opentelemetry"):
             app._is_instrumented_by_opentelemetry = False