def patch()

in aws_xray_sdk/ext/psycopg2/patch.py [0:0]

• 21 lines of code
• 1 McCabe index (conditional complexity)

def patch():
    wrapt.wrap_function_wrapper(
        'psycopg2',
        'connect',
        _xray_traced_connect
    )
    wrapt.wrap_function_wrapper(
        'psycopg2.extensions',
        'register_type',
        _xray_register_type_fix
    )
    wrapt.wrap_function_wrapper(
        'psycopg2.extensions',
        'quote_ident',
        _xray_register_type_fix
    )

    wrapt.wrap_function_wrapper(
        'psycopg2.extras',
        'register_default_jsonb',
        _xray_register_default_jsonb_fix
    )