Get in-out wrapper for function

getInOut_wrapper.py

import os

def getInOut(orig_fun=None, path_in=None, path_out=None):
    def _decorate(fun):
        @functools.wraps(fun)
        def wrapper(*args, **kwargs):
            os.chdir(path_in)
            fun(*args, **kwargs)
            os.chdir(path_out)

    if orig_fun:
        return _decorate(orig_fun)
    else:
        return _decorate