add comment

Author: srprash

aws_xray_sdk/ext/psycopg2/patch.py

@@ -7,7 +7,6 @@
 
 
 def patch():
-
     wrapt.wrap_function_wrapper(
         'psycopg2',
         'connect',
@@ -23,6 +22,10 @@ def patch():
         'quote_ident',
         _xray_register_type_fix
     )
+
+    # We should in general not patch internal methods (pyscopg2._json.register_type in this case).
+    # But there isn't a better workaround than this for the time being.
+    # Please see the GH issue for details: https://github.com/aws/aws-xray-sdk-python/issues/243
     wrapt.wrap_function_wrapper(
         'psycopg2._json',
         'register_type',
@@ -31,9 +34,8 @@ def patch():
 
 
 def _xray_traced_connect(wrapped, instance, args, kwargs):
-
     conn = wrapped(*args, **kwargs)
-    parameterized_dsn = { c[0]: c[-1] for c in map(methodcaller('split', '='), conn.dsn.split(' '))}
+    parameterized_dsn = {c[0]: c[-1] for c in map(methodcaller('split', '='), conn.dsn.split(' '))}
     meta = {
         'database_type': 'PostgreSQL',
         'url': 'postgresql://{}@{}:{}/{}'.format(
@@ -49,6 +51,7 @@ def _xray_traced_connect(wrapped, instance, args, kwargs):
 
     return XRayTracedConn(conn, meta)
 
+
 def _xray_register_type_fix(wrapped, instance, args, kwargs):
     """Send the actual connection or curser to register type."""
     our_args = list(copy.copy(args))