Python 将 cwd 更改为脚本目录

在 Python 中，可以使用 os 模块来更改当前工作目录（cwd）。这在操作文件、读取资源等场景中非常有用。但是，要确保在更改 cwd 时不会出现问题，需要采用一些最佳实践。

常规方法

import os

current_dir = os.getcwd()
script_dir = os.path.dirname(os.path.realpath(__file__))
os.chdir(script_dir)

在上面的代码片段中，我们首先获取了当前工作目录 current_dir。 然后，我们使用 os.path.realpath(__file__) 获取当前脚本的绝对路径，再使用 os.path.dirname() 函数得到当前脚本的目录 script_dir。最后，调用 os.chdir() 函数将当前工作目录更改为脚本目录。

最佳实践

虽然上面的代码片段有效，但它并不适用于所有情况。下面是一些最佳实践，以确保更改 cwd 后不会出现问题。

1. 使用 try/finally 语句

更改 cwd 后，try/finally 语句可以确保在代码块执行后将 cwd 更改回原始工作目录。

import os

try:
    current_dir = os.getcwd()
    script_dir = os.path.dirname(os.path.realpath(__file__))
    os.chdir(script_dir)

    # 执行代码块
finally:
    os.chdir(current_dir)

2. 确保脚本目录存在

在更改 cwd 之前，需要确保脚本目录确实存在。可以使用以下函数确保目录存在：

def ensure_dir_exists(dir_path):
    if not os.path.exists(dir_path):
        os.makedirs(dir_path)

然后，可以使用以下代码片段来确保目录存在：

import os

ensure_dir_exists(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'path/to/directory'))

3. 避免更改 cwd

尽管更改 cwd 在某些情况下很有用，但最好不要在编写可重用代码时使用它。相反，使用参数传递文件路径或将文件路径保存在变量中。这样，您的代码就更加灵活和可重用。下面是一个示例：

import os

def my_function(file_path):
    with open(file_path, 'r') as f:
        # 执行代码块

结论

在 Python 中更改 cwd 可以很有用，但需要遵循一些最佳实践。当您需要更改 cwd 时，请确保使用 try/finally 语句确保将 cwd 更改回原始工作目录。此外，确保脚本目录存在，并避免更改 cwd 以提高代码的灵活性和可重用性。