add django, and modernize flask with content

src/lumigo_opentelemetry/instrumentations/django/__init__.py

@@ -2,7 +2,9 @@
 
 from lumigo_opentelemetry.instrumentations import AbstractInstrumentor
 from lumigo_opentelemetry.libs.general_utils import lumigo_safe_execute
-from lumigo_opentelemetry.instrumentations.instrumentation_utils import add_body_attribute
+from lumigo_opentelemetry.instrumentations.instrumentation_utils import (
+    add_body_attribute,
+)
 from lumigo_opentelemetry.libs.json_utils import dump_with_context
 
 
@@ -19,15 +21,23 @@ def install_instrumentation(self) -> None:
 
         def request_hook(span: Span, request: HttpRequest):
             with lumigo_safe_execute("django request_hook"):
-                span.set_attribute("http.request.headers", dump_with_context("requestHeaders", request.headers))
+                span.set_attribute(
+                    "http.request.headers",
+                    dump_with_context("requestHeaders", request.headers),
+                )
                 add_body_attribute(span, request.body, "http.request.body")
 
         def response_hook(span: Span, request: HttpRequest, response: HttpResponse):
             with lumigo_safe_execute("django response_hook"):
-                span.set_attribute("http.response.headers", dump_with_context("responseHeaders", response.headers))
+                span.set_attribute(
+                    "http.response.headers",
+                    dump_with_context("responseHeaders", response.headers),
+                )
                 add_body_attribute(span, response.content, "http.response.body")
 
-        DjangoInstrumentor().instrument(request_hook=request_hook, response_hook=response_hook)
+        DjangoInstrumentor().instrument(
+            request_hook=request_hook, response_hook=response_hook
+        )
 
 
 instrumentor: AbstractInstrumentor = DjangoInstrumentorWrapper()

src/lumigo_opentelemetry/instrumentations/flask/__init__.py

@@ -19,13 +19,21 @@ def install_instrumentation(self) -> None:
 
         def request_hook(span: Span, flask_request_environ: Dict[str, Any]):
             with lumigo_safe_execute("flask_request_hook"):
-                span.set_attribute("http.request.headers", dump_with_context("requestHeaders", flask_request_environ))
+                span.set_attribute(
+                    "http.request.headers",
+                    dump_with_context("requestHeaders", flask_request_environ),
+                )
 
         def response_hook(span: Span, status, response_headers):
             with lumigo_safe_execute("flask_response_hook"):
-                span.set_attribute("http.response.headers", dump_with_context("responseHeaders", response_headers))
-
-        FlaskInstrumentor().instrument(request_hook=request_hook, response_hook=response_hook)
+                span.set_attribute(
+                    "http.response.headers",
+                    dump_with_context("responseHeaders", response_headers),
+                )
+
+        FlaskInstrumentor().instrument(
+            request_hook=request_hook, response_hook=response_hook
+        )
 
 
 instrumentor: AbstractInstrumentor = FlaskInstrumentorWrapper()

src/lumigo_opentelemetry/instrumentations/instrumentations.py

@@ -48,7 +48,7 @@
             "An error occurred while applying the '%s' instrumentation: %s",
             instrumentor.instrumentation_id,
             str(e),
-            exc_info=True
+            exc_info=True,
         )
 
 logger.debug(

src/test/integration/django/app/manage.py

@@ -1,13 +1,13 @@
 #!/usr/bin/env python
 """Django's command-line utility for administrative tasks."""
 import os
-import sys
 
 
 def main():
     """Run administrative tasks."""
     os.environ.setdefault("DJANGO_SETTINGS_MODULE", "app.settings")
-    import lumigo_opentelemetry
+    import lumigo_opentelemetry  # noqa: F401
+
     try:
         from django.core.management import execute_from_command_line
     except ImportError as exc:
@@ -16,7 +16,7 @@ def main():
             "available on your PYTHONPATH environment variable? Did you "
             "forget to activate a virtual environment?"
         ) from exc
-    execute_from_command_line(['manage.py', 'runserver', '5003'])
+    execute_from_command_line(["manage.py", "runserver", "5003"])
 
 
 if __name__ == "__main__":

src/test/integration/django/scripts/start_django

@@ -1,4 +1,4 @@
 #!/bin/sh
 
-pushd app || exit 1
+cd app || exit 1
 python manage.py runserver 5003  &

src/test/integration/django/tests/test_django.py

@@ -1,4 +1,3 @@
-import json
 import unittest
 from test.test_utils.span_exporter import wait_for_exporter
 from test.test_utils.spans_parser import SpansContainer
@@ -25,9 +24,11 @@ def test_200_OK(self):
         self.assertIsNotNone(root)
         self.assertEqual(root["kind"], "SpanKind.SERVER")
         self.assertEqual(root["attributes"]["http.status_code"], 200)
-        self.assertEqual(root["attributes"]["http.request.body"], '{"support": "django"}')
+        self.assertEqual(
+            root["attributes"]["http.request.body"], '{"support": "django"}'
+        )
         self.assertIsNotNone(root["attributes"]["http.request.headers"])
-        self.assertIsNotNone(root["attributes"]['http.response.headers'])
-        self.assertIsNotNone(root["attributes"]['http.response.body'])
+        self.assertIsNotNone(root["attributes"]["http.response.headers"])
+        self.assertIsNotNone(root["attributes"]["http.response.body"])
         self.assertEqual(root["attributes"]["http.method"], "GET")
         self.assertEqual(root["attributes"]["http.host"], "localhost:5003")