如何正确确定当前脚本目录？

Question

我想看看确定 Python 中当前脚本目录的最佳方法是什么。

我发现，由于调用Python代码的方式多种多样，很难找到一个好的解决方案。

这里有一些问题：

• 如果使用exec执行脚本，则未定义__file__，
• 仅定义了execfile__module__在模块

中用例：

• ./myfile.py
• python myfile.py
• ./somedir/myfile.py
• python somedir/myfile. py
• execfile('myfile.py') （来自另一个脚本，可以位于另一个目录中，并且可以有另一个当前目录。

我知道没有完美的解决方案，但是我正在寻找解决大多数情况的最佳方法，

最常用的方法是 os.path.dirname(os.path.abspath(__file__)) ，但这确实不起作用。您使用 exec() 从另一个脚本执行脚本

警告

任何使用当前目录的解决方案都将失败，这可能会根据脚本的调用方式而有所不同，也可能会在运行的脚本中进行更改。 。

Answer 1

首先..如果我们谈论注入匿名代码的方法，这里缺少一些用例..

code.compile_command()
code.interact()
imp.load_compiled()
imp.load_dynamic()
imp.load_module()
__builtin__.compile()
loading C compiled shared objects? example: _socket?)

但是，真正的问题是，您的目标是什么 - 您是否试图强制执行某种安全性？或者您只是对正在加载的内容感兴趣。

如果您对安全性感兴趣，那么通过 exec/execfile 导入的文件名是无关紧要的 -你应该使用 rexec，它提供以下功能：

该模块包含 RExec 类，
它支持 r_eval()、r_execfile()、
r_exec() 和 r_import() 方法，其中
是标准的受限版本
Python 函数 eval()、execfile() 和
exec 和 import 语句。代码
在这个受限环境中执行
只能访问模块和
被认为安全的功能；你可以
子类 RExec 添加或删除功能为
想要的。

然而，如果这更多的是一种学术追求......这里有一些愚蠢的方法，你可以
也许能够更深入地挖掘..

示例脚本：

./deep.py

print ' >> level 1'
execfile('deeper.py')
print ' << level 1'

./deeper.py

print '\t >> level 2'
exec("import sys; sys.path.append('/tmp'); import deepest")
print '\t << level 2'

/tmp/deepest.py >

print '\t\t >> level 3'
print '\t\t\t I can see the earths core.'
print '\t\t << level 3'

./codespy.py

import sys, os

def overseer(frame, event, arg):
    print "loaded(%s)" % os.path.abspath(frame.f_code.co_filename)

sys.settrace(overseer)
execfile("deep.py")
sys.exit(0)

输出

loaded(/Users/synthesizerpatel/deep.py)
>> level 1
loaded(/Users/synthesizerpatel/deeper.py)
    >> level 2
loaded(/Users/synthesizerpatel/<string>)
loaded(/tmp/deepest.py)
        >> level 3
            I can see the earths core.
        << level 3
    << level 2
<< level 1

当然，这是一种资源密集型的方法，你需要跟踪
你所有的代码..效率不高。但是，我认为这是一种新颖的方法
因为即使您深入巢穴，它仍然会继续工作。
您不能覆盖“eval”。尽管您可以重写exe​​cfile()。

请注意，此方法仅涵盖 exec/execfile，而不是“导入”。
对于更高级别的“模块”负载挂钩，您可以使用 use
sys.path_hooks（写入由 PyMOTW 提供）。

这就是我的全部想法。

Answer 2

这是一个部分解决方案，仍然比迄今为止所有已发布的解决方案更好。

import sys, os, os.path, inspect

#os.chdir("..")

if '__file__' not in locals():
    __file__ = inspect.getframeinfo(inspect.currentframe())[0]

print os.path.dirname(os.path.abspath(__file__))

现在这适用于所有调用，但如果有人使用 chdir() 更改当前目录，这也会失败。

注意：

• sys.argv[0] 不起作用，如果使用 python -c "execfile('path- tester.py')"
• 我在 https://gist.github.com/1385555 发布了完整的测试 欢迎您改进它。

Answer 3

这在大多数情况下应该有效：

import os,sys
dirname=os.path.dirname(os.path.realpath(sys.argv[0]))

Answer 4

希望这有帮助：-
如果您从任何地方运行脚本/模块，您将能够访问 __file__ 变量，它是表示脚本位置的模块变量。

另一方面，如果您使用解释器，则无法访问该变量，您将在其中获得名称 NameError 和 os.getcwd()如果您从其他地方运行该文件，将为您提供不正确的目录。

这个解决方案应该可以为您提供在所有情况下所需的内容：

from inspect import getsourcefile
from os.path import abspath
abspath(getsourcefile(lambda:0))

我还没有彻底测试它，但它解决了我的问题问题。

Answer 5

如果 __file__ 可用：

# -- script1.py --
import os
file_path = os.path.abspath(__file__)
print(os.path.dirname(file_path))

对于那些我们希望能够从解释器运行命令或获取运行脚本的位置的路径的人：

# -- script2.py --
import os
print(os.path.abspath(''))

这可以从解释器运行。
但是当在脚本中运行（或导入）时，它给出了所在位置的路径
您运行脚本的位置，而不是包含的目录的路径
带有打印的脚本。

示例：

如果您的目录结构

test_dir (in the home dir)
├── main.py
└── test_subdir
    ├── script1.py
    └── script2.py

为

# -- main.py --
import script1.py
import script2.py

输出为：

~/test_dir/test_subdir
~/test_dir

Answer 6

由于之前的答案要求您导入一些模块，我想我会写一个不需要的答案。如果您不想导入任何内容，请使用下面的代码。

this_dir = '/'.join(__file__.split('/')[:-1])
print(this_dir)

如果脚本位于 /path/to/script.py 上，那么这将打印 /path/to。请注意，这将在终端上抛出错误，因为没有执行任何文件。这基本上从 __file__ 解析目录，删除它的最后一部分。在这种情况下，删除 /script.py 以生成输出 /path/to。

Answer 7

print(__import__("pathlib").Path(__file__).parent)

Answer 8

os.path.dirname(os.path.abspath(__file__))

这确实是你能得到的最好的。

使用 exec/execfile 执行脚本是不常见的；通常您应该使用模块基础结构来加载脚本。如果您必须使用这些方法，我建议在传递给脚本的 globals 中设置 __file__ ，以便它可以读取该文件名。

没有其他方法可以获取执行代码中的文件名：正如您所注意到的，CWD 可能位于完全不同的位置。

Answer 9

如果您确实想涵盖通过 execfile(...) 调用脚本的情况，您可以使用 inspect 模块来推断文件名（包括路径） 。据我所知，这适用于您列出的所有情况：

filename = inspect.getframeinfo(inspect.currentframe()).filename
path = os.path.dirname(os.path.abspath(filename))

Answer 10

在 Python 3.4+ 中，您可以使用更简单的 pathlib模块：

from inspect import currentframe, getframeinfo
from pathlib import Path

filename = getframeinfo(currentframe()).filename
parent = Path(filename).resolve().parent

您还可以使用 __file__（如果可用）来完全避免使用 inspect 模块：

from pathlib import Path
parent = Path(__file__).resolve().parent

Answer 11

#!/usr/bin/env python
import inspect
import os
import sys

def get_script_dir(follow_symlinks=True):
    if getattr(sys, 'frozen', False): # py2exe, PyInstaller, cx_Freeze
        path = os.path.abspath(sys.executable)
    else:
        path = inspect.getabsfile(get_script_dir)
    if follow_symlinks:
        path = os.path.realpath(path)
    return os.path.dirname(path)

print(get_script_dir())

它适用于 CPython、Jython、Pypy。如果使用 execfile() 执行脚本，它就可以工作（基于 sys.argv[0] 和 __file__ 的解决方案将在此处失败）。如果脚本位于可执行 zip 文件 (/an Egg) 中，则它可以工作。如果脚本是从 zip 文件“导入”（PYTHONPATH=/path/to/library.zip python -mscript_to_run）的，则它可以工作；在这种情况下它返回存档路径。如果脚本被编译成独立的可执行文件 (sys.frozen)，它就可以工作。它适用于符号链接（realpath 消除了符号链接）。它在交互式解释器中工作；在这种情况下它返回当前工作目录。

Answer 12

os.path... 方法是 Python 2 中的“完成的事情”。

在 Python 3 中，您可以按如下所示找到脚本目录：

from pathlib import Path
script_path = Path(__file__).parent

Answer 13

注意：这个答案现在是一个包（也具有安全的相对导入功能）

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__", "./_") 返回键 "__file__" 中的值（如果它存在于 globals() 中） ，否则返回提供的默认值"./_"。
• 其余代码只是将 __file__ （或 "./_"）扩展为绝对文件路径，然后返回文件路径的基目录。

替代方案：

如果您确定 __file__ 可用于您周围的代码，您可以简化为：

• >= Python 3.4： this_dir = Path(__file__ ).absolute().parent
• >= Python 2: this_dir = os.path.dirname(os.path.abspath(__file__))

Answer 14

会

import os
cwd = os.getcwd()

做你想做的事吗？我不确定“当前脚本目录”到底是什么意思。您给出的用例的预期输出是什么？

Answer 15

只需使用 os.path.dirname(os.path.abspath(__file__)) 并仔细检查是否确实需要使用 exec 的情况。如果您无法将脚本用作模块，这可能是设计有问题的迹象。

请记住 Python #8 的禅宗，如果您认为对于必须适用于 exec 的用例有一个很好的论据，那么请让我们知道一些有关问题背景的更多详细信息。

Answer 16

要获取包含当前脚本的目录的绝对路径，您可以使用：

from pathlib import Path
absDir = Path(__file__).parent.resolve()

请注意，需要 .resolve() 调用，因为这是使路径成为绝对路径的调用。如果没有 resolve()，您将获得类似于 '.' 的内容。

此解决方案使用 pathlib，它是 Python 的一部分stdlib 自 v3.4 (2014) 起。与使用 os 的其他解决方案相比，这是更可取的。

官方 pathlib 文档有一个有用的表，将旧的 os 函数映射到新函数：https://docs.python.org/3/library/pathlib.html#correspondence-to-tools-in -操作系统模块