Feature toggle

feature_toggle.py

class FeatureToggle(object):
    def __init__(self, old, new, toggle):
        self.old = old
        self.new = new
        self.toggle = toggle

    def __getattr__(self, name):
        def wrapper(*args, **kwargs):
            instance_to_use = self.new if self.toggle else self.old
            return getattr(instance_to_use, name)(*args, **kwargs)
        return wrapper