save.py

from contextlib import contextmanager

@contextmanager
def save(obj, attrs=None):
    """Save attributes of an object, then restore them.

    Example:
    import breakfast
    with save(breakfast, ('eggs', 'bacon')):
        breakfast.Eggs = lambda: "Eggs"
    """
    orig_attrs = {}
    for attr in attrs:
        orig_attrs[attr] = getattr(obj, attr)

    try:
        yield
    finally:
        for attr in attrs:
            try:
                setattr(obj, attr, orig_attrs[attr])
            except:
                pass