Fix formatting.

DataDog · Jan 2, 2025 · 9f12157 · 9f12157
1 parent bf2776b
commit 9f12157
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 6 deletions.
diff --git a/ddtrace/profiling/exporter/http.py b/ddtrace/profiling/exporter/http.py
@@ -66,9 +66,9 @@ def __init__(
         self.version: typing.Optional[str] = version
         self.tags: typing.Dict[str, str] = tags if tags is not None else {}
         self.max_retry_delay: typing.Optional[float] = max_retry_delay
-        self.endpoint_call_counter_span_processor: typing.Optional[
-            EndpointCallCounterProcessor
-        ] = endpoint_call_counter_span_processor
+        self.endpoint_call_counter_span_processor: typing.Optional[EndpointCallCounterProcessor] = (
+            endpoint_call_counter_span_processor
+        )
 
         self.__post_init__()
 
@@ -220,8 +220,18 @@ def export(
             "family": "python",
             "attachments": [item["filename"].decode("utf-8") for item in data],
             "tags_profiler": self._get_tags(service),
-            "start": (datetime.datetime.fromtimestamp(start_time_ns / 1e9, tz=datetime.timezone.utc).replace(microsecond=0).isoformat() + "Z"),
-            "end": (datetime.datetime.fromtimestamp(end_time_ns / 1e9, tz=datetime.timezone.utc).replace(microsecond=0).isoformat() + "Z"),
+            "start": (
+                datetime.datetime.fromtimestamp(start_time_ns / 1e9, tz=datetime.timezone.utc)
+                .replace(microsecond=0)
+                .isoformat()
+                + "Z"
+            ),
+            "end": (
+                datetime.datetime.fromtimestamp(end_time_ns / 1e9, tz=datetime.timezone.utc)
+                .replace(microsecond=0)
+                .isoformat()
+                + "Z"
+            ),
         }  # type: Dict[str, Any]
 
         if self.endpoint_call_counter_span_processor is not None:

diff --git a/tests/appsec/iast_packages/packages/pkg_pyjwt.py b/tests/appsec/iast_packages/packages/pkg_pyjwt.py
@@ -3,6 +3,7 @@
 
 https://pypi.org/project/PyJWT/
 """
+
 import datetime
 
 from flask import Blueprint
@@ -25,7 +26,10 @@ def pkg_pyjwt_view():
         secret_key = "your-256-bit-secret"
         user_payload = request.args.get("package_param", "default-user")
 
-        payload = {"user": user_payload, "exp": datetime.datetime.now(datetime.timezone.utc) + datetime.timedelta(seconds=30)}
+        payload = {
+            "user": user_payload,
+            "exp": datetime.datetime.now(datetime.timezone.utc) + datetime.timedelta(seconds=30),
+        }
 
         try:
             # Encode the payload to create a JWT