将变量设置为已导入它的模块的名称

Question

假设我有两个模块：

first.package.module

# Assign module to a variable
which_module = ???

print("I originally live in {}".format(__name__))   # Prints first.package.module
print("I was run from {}".format(which_module))

second.package.module

from first.package.module import *

third.package.module

from first.package.module import *

如何在 second.package.module 中导入后打印第二行“I was run from”以打印 second.package.module？或者 third.package.module 从那里运行时。

我想要这种行为的原因是由于在 Django 中构建了一个在同一个项目中多次使用的应用程序，即同一个应用程序有多个实例。这些实例中的每一个都有自己的模型，这些模型继承自抽象模型。为了构建可重用的视图和 url，我想动态加载模型，为此，我需要知道哪个模块导入了应用程序并运行它。

Answer 1

您可以打印触发模块加载的模块名称：

import sys

print sys._getframe(1).f_globals['__name__']

在带有新的importlib 堆栈的 Python 3 中，您需要将帧数增加到 6（至少对于 3.5、3.6 和 3.7）：

print(sys._getframe(6).f_globals['__name__'])

跨 CPython 的解决方案是过滤掉堆栈中的任何 importlib._bootstrap* 元素：

import sys

def imported_from(depth=0):
    # skip the frames of this function, and the caller
    f = sys._getframe(2 + depth)
    while f and f.f_code.co_filename.startswith("<frozen importlib._bootstrap"):
        f = f.f_back
    return f and f.f_globals['__name__']

print(imported_from())

上面的imported_from()函数可以在实用模块中定义并导入； sys._getframe(2) 调用确保起始上下文是触发imported_from() 调用的任何内容。提供一个正整数作为depth 参数以增加跳过的帧数。该函数适用于任何提供 sys._getframe() 函数的 Python 版本，无论它是否使用 importlib 堆栈。

请注意：

• 依赖于 CPython 实现细节（暴露帧堆栈在其他 Python 实现中不可用）。见sys._getframe() function documentation：

  CPython 实现细节：此函数应仅用于内部和专用目的。不保证在 Python 的所有实现中都存在。

• 在 Python 3 中，它取决于 importlib 堆栈的确切实现细节；正在进行的开发可以在该堆栈中添加或删除调用。例如，在首次引入importlib 的 Python 3.3 中，要跳过的正确堆栈计数是 9，在 3.4 中它下降到 7，而 3.5 将它下降到 6（这个数字后来一直很稳定）。 imported_from() 解决了这个问题，但引入了一个新的依赖于实现细节的假设：可以通过在文件名中查找 <frozen importlib._bootstrap 前缀来检测与导入有关的堆栈帧。理论上，可以使用未冻结的importlib._bootstrap 编译CPython（不作为marshal 数据数组包含在解释器二进制文件中）。

• 仅在第一次导入模块时有效，此时 Python 会执行模块代码以生成 sys.modules 对象。

最后一点我怎么强调都不为过。 Python 只加载一个模块一次。导入是一个两步过程：加载和绑定。加载步骤仅在模块对象不存在时执行，模块的所有其他导入仅绑定名称。

绑定名称实际上是无法挂钩的；一旦模块已经在内存中，import modulename 只不过是一个 modulename = sys.modules['modulename'] 赋值。