update imports for tracer.Pin privatization

Author: emmettbutler

utils/build/docker/python/django/app/urls.py

@@ -53,9 +53,13 @@ def track_custom_event(self, *args, **kwargs):
     track_user_sdk = TUS()
 
 try:
-    from ddtrace.trace import Pin, tracer
+    from ddtrace._trace.pin import Pin
+    from ddtrace.trace import tracer
 except ImportError:
-    from ddtrace import tracer, Pin
+    try:
+        from ddtrace.trace import Pin, tracer
+    except ImportError:
+        from ddtrace import tracer, Pin
 
 
 ddtrace.patch_all(urllib3=True)
@@ -545,7 +549,8 @@ def view_iast_xss_secure(request):
 
 @csrf_exempt
 def view_iast_stacktraceleak_insecure(request):
-    return HttpResponse("""
+    return HttpResponse(
+        """
   Traceback (most recent call last):
   File "/usr/local/lib/python3.9/site-packages/some_module.py", line 42, in process_data
     result = complex_calculation(data)
@@ -568,7 +573,8 @@ def view_iast_stacktraceleak_insecure(request):
 ValueError: Constraint violation at step 9
 
 Lorem Ipsum Foobar
-""")
+"""
+    )
 
 
 @csrf_exempt

utils/build/docker/python/fastapi/main.py

@@ -38,11 +38,15 @@
 from ddtrace.appsec import trace_utils as appsec_trace_utils
 
 try:
-    from ddtrace.trace import Pin
+    from ddtrace._trace.pin import Pin
     from ddtrace.trace import tracer
 except ImportError:
-    from ddtrace import Pin
-    from ddtrace import tracer
+    try:
+        from ddtrace.trace import Pin
+        from ddtrace.trace import tracer
+    except ImportError:
+        from ddtrace import Pin
+        from ddtrace import tracer
 
 ddtrace.patch_all(urllib3=True)
 

utils/build/docker/python/flask/app.py

@@ -104,11 +104,15 @@
 from exception_replay_controller import exception_replay_blueprint
 
 try:
-    from ddtrace.trace import Pin
+    from ddtrace._trace.pin import Pin
     from ddtrace.trace import tracer
 except ImportError:
-    from ddtrace import Pin
-    from ddtrace import tracer
+    try:
+        from ddtrace.trace import Pin
+        from ddtrace.trace import tracer
+    except ImportError:
+        from ddtrace import Pin
+        from ddtrace import tracer
 
 # Patch kombu and urllib3 since they are not patched automatically
 ddtrace.patch_all(kombu=True, urllib3=True)
@@ -250,23 +254,27 @@ def check_and_create_users_table():
     cur = postgres_db.cursor()
 
     # Check if 'users' exists
-    cur.execute("""
+    cur.execute(
+        """
         SELECT EXISTS (
             SELECT FROM information_schema.tables
             WHERE table_name = 'users'
         );
-    """)
+    """
+    )
     table_exists = cur.fetchone()[0]
 
     if not table_exists:
-        cur.execute("""
+        cur.execute(
+            """
             CREATE TABLE users (
                 id VARCHAR(255) PRIMARY KEY,
                 username VARCHAR(255),
                 email VARCHAR(255),
                 password VARCHAR(255)
             );
-        """)
+        """
+        )
         postgres_db.commit()
 
         users_data = [