Skip to content

Commit 34e7013

Browse files
committed
tried linting
1 parent 2757171 commit 34e7013

File tree

2 files changed

+12
-10
lines changed

2 files changed

+12
-10
lines changed

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

+11-9
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ def get_default_span_name():
8383
span_name = otel_wsgi.get_default_span_name(flask.request.environ)
8484
return span_name
8585

86+
8687
def _rewrapped_app(wsgi_app, response_hook=None):
8788
def _wrapped_app(wrapped_app_environ, start_response):
8889
# We want to measure the time for route matching, etc.
@@ -113,7 +114,7 @@ def _start_response(status, response_headers, *args, **kwargs):
113114
status,
114115
)
115116
if response_hook:
116-
response_hook(span, status, response_headers)
117+
response_hook(span, status, response_headers)
117118
return start_response(status, response_headers, *args, **kwargs)
118119

119120
return wsgi_app(wrapped_app_environ, _start_response)
@@ -122,11 +123,9 @@ def _start_response(status, response_headers, *args, **kwargs):
122123

123124

124125
def _wrapped_before_request(request_hook=None, tracer=None):
125-
126126
def _before_request():
127127
if _excluded_urls.url_disabled(flask.request.url):
128128
return
129-
130129
flask_request_environ = flask.request.environ
131130
span_name = get_default_span_name()
132131
token = context.attach(
@@ -159,8 +158,7 @@ def _before_request():
159158
flask_request_environ[_ENVIRON_ACTIVATION_KEY] = activation
160159
flask_request_environ[_ENVIRON_SPAN_KEY] = span
161160
flask_request_environ[_ENVIRON_TOKEN] = token
162-
163-
161+
164162
return _before_request
165163

166164

@@ -190,20 +188,23 @@ class _InstrumentedFlask(flask.Flask):
190188
_tracer_provider = None
191189
_request_hook = None
192190
_response_hook = None
193-
191+
194192
def __init__(self, *args, **kwargs):
195193
super().__init__(*args, **kwargs)
196194

197195
self._original_wsgi_ = self.wsgi_app
198-
199-
self.wsgi_app = _rewrapped_app(self.wsgi_app, _InstrumentedFlask._response_hook)
196+
197+
self.wsgi_app = _rewrapped_app(
198+
self.wsgi_app, _InstrumentedFlask._response_hook
199+
)
200200

201201
tracer = trace.get_tracer(
202202
__name__, __version__, _InstrumentedFlask._tracer_provider
203203
)
204204

205205
_before_request = _wrapped_before_request(
206-
_InstrumentedFlask._request_hook, tracer,
206+
_InstrumentedFlask._request_hook,
207+
tracer,
207208
)
208209
self._before_request = _before_request
209210
self.before_request(_before_request)
@@ -216,6 +217,7 @@ class FlaskInstrumentor(BaseInstrumentor):
216217
217218
See `BaseInstrumentor`
218219
"""
220+
219221
def _instrument(self, **kwargs):
220222
self._original_flask = flask.Flask
221223
request_hook = kwargs.get("request_hook")

instrumentation/opentelemetry-instrumentation-flask/tests/test_programmatic.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@ def test_404(self):
185185
self.assertEqual(span_list[0].name, "HTTP POST")
186186
self.assertEqual(span_list[0].kind, trace.SpanKind.SERVER)
187187
self.assertEqual(span_list[0].attributes, expected_attrs)
188-
188+
189189
def test_internal_error(self):
190190
expected_attrs = expected_attributes(
191191
{

0 commit comments

Comments
 (0)