opentelemetry-python-contrib/reference/ddtrace/contrib/pylibmc/patch.py

import pylibmc

from .client import TracedClient

# Original Client class
_Client = pylibmc.Client


def patch():
    setattr(pylibmc, 'Client', TracedClient)


def unpatch():
    setattr(pylibmc, 'Client', _Client)