将脚本的目录添加到字符串前面

Question

编写一次性脚本时，通常需要从与脚本相同的目录加载配置文件、图像或类似的东西。最好无论脚本从哪个目录执行，它都应该继续正常工作，因此我们可能不想简单地依赖当前工作目录。

如果在您使用它的同一个文件中定义，这样的东西可以正常工作：

from os.path import abspath, dirname, join

def prepend_script_directory(s):
    here = dirname(abspath(__file__))
    return join(here, s)

将相同的函数复制粘贴或重写到每个模块中是不可取的，但是存在一个问题：如果将其移动到单独的库中，然后导入作为一个函数，__file__ 现在正在引用其他一些模块，并且结果不正确。

我们也许可以使用它，但似乎 sys.argv 也可能不可靠。

def prepend_script_directory(s):
    here = dirname(abspath(sys.argv[0]))
    return join(here, s)

如何稳健且正确地编写 prepend_script_directory ？

Answer 1

每当我执行脚本时，我个人都会将 os.chdir 放入脚本的目录中。只是：

import os
os.chdir(os.path.split(__file__)[0])

但是，如果您确实想将这个东西重构到库中，那么您本质上需要一个能够了解其调用者状态的函数。因此，

prepend_script_directory(__file__, blah)

如果你只是想写，

prepend_script_directory(blah)

你必须使用堆栈帧执行特定于 cpython 的技巧：

import inspect

def getCallerModule():
    # gets globals of module called from, and prints out __file__ global
    print(inspect.currentframe().f_back.f_globals['__file__'])

Answer 2

我认为它闻起来不正确的原因是 $PYTHONPATH （或 sys.path）是正确使用的通用机制。

Answer 3

你想要 pkg_resources

import pkg_resources
foo_fname = pkg_resources.resource_filename(__name__, "foo.txt")