/usr/lib/python2.7/dist-packages/aws_xray_sdk/ext/mysql/patch.py is in python-aws-xray-sdk 0.95-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | import wrapt
import mysql.connector
from aws_xray_sdk.ext.dbapi2 import XRayTracedConn
MYSQL_ATTR = {
'_host': 'name',
'_user': 'user',
}
def patch():
wrapt.wrap_function_wrapper(
'mysql.connector',
'connect',
_xray_traced_connect
)
# patch alias
if hasattr(mysql.connector, 'Connect'):
mysql.connector.Connect = mysql.connector.connect
def _xray_traced_connect(wrapped, instance, args, kwargs):
conn = wrapped(*args, **kwargs)
meta = {}
for attr, key in MYSQL_ATTR.items():
if hasattr(conn, attr):
meta[key] = getattr(conn, attr)
if hasattr(conn, '_server_version'):
version = sanitize_db_ver(getattr(conn, '_server_version'))
if version:
meta['database_version'] = version
return XRayTracedConn(conn, meta)
def sanitize_db_ver(raw):
if not raw or not isinstance(raw, tuple):
return raw
return '.'.join(str(num) for num in raw)
|