Implemented username and pw check on urllib

Author: Ryo Kather

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

@@ -39,6 +39,7 @@
 import functools
 import types
 from typing import Collection
+import yarl
 from urllib.request import (  # pylint: disable=no-name-in-module,import-error
     OpenerDirector,
     Request,
@@ -144,7 +145,7 @@ def _instrumented_open_call(
 
         labels = {
             SpanAttributes.HTTP_METHOD: method,
-            SpanAttributes.HTTP_URL: url,
+            SpanAttributes.HTTP_URL: str(yarl.URL(url).with_user(None)),
         }
 
         with tracer.start_as_current_span(
@@ -153,7 +154,7 @@ def _instrumented_open_call(
             exception = None
             if span.is_recording():
                 span.set_attribute(SpanAttributes.HTTP_METHOD, method)
-                span.set_attribute(SpanAttributes.HTTP_URL, url)
+                span.set_attribute(SpanAttributes.HTTP_URL, str(yarl.URL(url).with_user(None)))
 
             headers = get_or_create_headers()
             inject(headers)

instrumentation/opentelemetry-instrumentation-urllib/tests/test_urllib_integration.py

@@ -318,6 +318,15 @@ def test_requests_timeout_exception(self, *_, **__):
         span = self.assert_span()
         self.assertEqual(span.status.status_code, StatusCode.ERROR)
 
+    def test_credentials_url(self):
+        url = "http://username:[email protected]/status/200"
+
+        with self.assertRaises(Exception):
+            self.perform_request(url)
+
+        span = self.assert_span()
+        print(span.attributes)
+        self.assertEqual(span.attributes[SpanAttributes.HTTP_URL], self.URL)
 
 class TestRequestsIntegration(RequestsIntegrationTestBase, TestBase):
     @staticmethod

tox.ini

@@ -245,7 +245,7 @@ commands_pre =
 
   flask: pip install {toxinidir}/instrumentation/opentelemetry-instrumentation-flask[test]
 
-  urllib: pip install {toxinidir}/instrumentation/opentelemetry-instrumentation-urllib[test]
+  urllib: pip install yarl {toxinidir}/instrumentation/opentelemetry-instrumentation-urllib[test]
 
   urllib3: pip install {toxinidir}/instrumentation/opentelemetry-instrumentation-urllib3[test]