我想知道在Python中确定当前脚本目录的最佳方法是什么。
我发现,由于调用Python代码的方法很多,很难找到一个好的解决方案。
以下是一些问题:
如果脚本使用exec, execfile执行,则__file__未定义
__module__只在模块中定义
用例:
。/ myfile.py
python myfile.py
/ somedir / myfile.py
python somedir / myfile.py
Execfile ('myfile.py')(来自另一个脚本,可以位于另一个目录,并且可以有另一个当前目录。
我知道没有完美的解决方案,但我正在寻找解决大多数情况的最佳方法。
最常用的方法是os.path.dirname(os.path.abspath(__file__)),但如果你用exec()从另一个脚本执行脚本,这就行不通了。
警告
任何使用当前目录的解决方案都会失败,这可以根据脚本调用的方式有所不同,也可以在运行的脚本中更改。
要获得包含当前脚本的目录的绝对路径,您可以使用:
from pathlib import Path
absDir = Path(__file__).parent.resolve()
请注意,.resolve()调用是必需的,因为它使路径成为绝对路径。如果没有resolve(),您将获得类似于'.'的东西。
这个解决方案使用pathlib,它自v3.4(2014)以来一直是Python的stdlib的一部分。与使用os的其他解决方案相比,这是更可取的。
官方的pathlib文档有一个有用的表,将旧的操作系统函数映射到新函数:https://docs.python.org/3/library/pathlib.html#correspondence-to-tools-in-the-os-module
注意:这个答案现在是一个包(也具有安全的相对导入功能)
https://github.com/heetbeet/locate
$ pip install locate
$ python
>>> from locate import this_dir
>>> print(this_dir())
C:/Users/simon
对于.py脚本和交互使用:
我经常使用脚本的目录(用于访问与它们一起存储的文件),但我也经常在交互式shell中运行这些脚本以进行调试。我将this_dir定义为:
运行或导入.py文件时,该文件的基目录。这总是正确的路径。
运行.ipyn笔记本时,当前工作目录。这始终是正确的路径,因为Jupyter将工作目录设置为.ipynb基本目录。
在REPL中运行时,当前工作目录。嗯,当代码从文件中分离出来时,实际的“正确路径”是什么?相反,在调用REPL之前,将更改到“正确路径”作为您的责任。
Python 3.4(及以上版本):
from pathlib import Path
this_dir = Path(globals().get("__file__", "./_")).absolute().parent
Python 2(及以上版本):
import os
this_dir = os.path.dirname(os.path.abspath(globals().get("__file__", "./_")))
解释:
Globals()将所有全局变量作为字典返回。
.get("__file__", "./_")如果globals()中存在键"__file__",则返回键"__file__"的值,否则返回提供的默认值"./_"。
其余的代码只是将__file__(或"./_")展开为一个绝对文件路径,然后返回该文件路径的基目录。
选择:
如果你确定__file__对你的周边代码可用,你可以简化为:
>= Python 3.4: this_dir = Path(__file__).absolute().parent
>= Python 2: this_dir = os.path.dirname(os.path.abspath(__file__))