Useful context managers

context_managers.py

def tmp_copy(source, destination):
    import shutil
    import os
    
    shutil.copy(source, destination)
    try:
        yield
    finally:
        os.remove(destination)

        
def chdir(directory):
    from pathlib import Path
    import os
    
    cwd = Path.cwd()
    os.chdir(directory)
    try:
        yield
    finally:
        os.chdir(cwd)