在python上使用临时workdir

Question

在python上设置workdir是可能的(操作系统chdir（'path'））比如设置局部变量？我的意思是，只有在调用一个函数时才可能使用不同的目录？你知道吗

即我有：

foo_function
    |
    module1---a.json
    |         |
    |         b.csv
    |
    module2---a.json
    |          |
    |         b.csv
    module3

我想要（举例）：

my_path_1 = '/module1/'
my_path_2 = '/module2/'
def foo(a,b,my_path):
    os.chdir('my_path')
    os.mkdir('csv')
    os.mkdir('json')
    shutil.move(a+'.json','json')
    shutil.move(b+'.csv','csv')

例如，现在如果我调用一个函数B来删除module3，只需删除它，而不使用“my\u path”的新目录来搜索module3（foo的最后一个调用）

Answer 1

除非必须更改目录，否则最好使用绝对路径。你知道吗

def foo(a,b,my_path):
    csv_dir = os.path.join(my_path, 'csv')
    json_dir = os.path.join(my_path, 'json')

    os.mkdir(csv_dir)
    os.mkdir(json_dir)

    shutil.move(os.path.join(my_path, a+'.json'), json_dir)
    shutil.move(os.path.join(my_path, b+'.csv'),' csv_dir)

Answer 2

您可以创建上下文管理器来执行以下操作：

@contextlib.contextmanager
def temp_chdir(path):
    prev = os.getcwd()
    os.chdir(path)
    try:
        yield
    finally:
        os.chdir(prev)

 with temp_chdir("~/some_dir"):
    funcion_a()

或者做个装饰师

def chdir_dec(path):

    def decorator(fn):
        @wraps(fn)
        def _fn(*args, **kwargs):
            with temp_chdir(path):
                return fn(*args, **kwargs)
        return _fn

    return decorator

# example ... 

@chdir_dec('../some_path')
def function_a():
    os.system("ls")