【UMI代码Trick学习】终于知道怎么使用test文件夹了,将当前文件的执行目录设置为根目录,方便调用其他自己写的包
通过在Python文件的开头加入这几行代码来更改文件执行的根目录
# %%
import sys
import os
ROOT_DIR = os.path.dirname(os.path.dirname(__file__))
sys.path.append(ROOT_DIR)
os.chdir(ROOT_DIR)
代码解释
这段代码中,首先使用 os.path.dirname 函数获取当前文件的父目录,然后再次调用 os.path.dirname 函数获取父目录的父目录,这是因为 os.path.dirname(file) 返回的是当前文件的目录路径,再调用一次则获取到的是父目录的路径。获取到父目录的父目录后,将其添加到 sys.path 中,以便后续的模块引入时可以搜索到对应的模块所在路径。最后,使用 os.chdir 函数将当前工作目录切换至父目录的父目录,确保后续代码执行的相对路径正确。
主要功能
主要功能是设置当前工作目录为当前文件的父目录的父目录,并将父目录的父目录添加到系统路径中,以便引入模块时可以正确搜索到目标模块文件。