破解python的import语句

Question

我正在为一个相当特定的目的构建一个 Python 模块。我想做的是在importing 后面获得更多功能。

我想要一个设置，通过它说from my_module import foo 将运行一个函数并传递字符串"foo"。此函数将返回应导入的对象。

例如，也许我想做一个基于云的导入系统。我想将社区脚本存储在云中，然后在用户尝试导入它们时下载它们。

也许我使用代码from cloud import test_module。这将检查缓存以确定test_module 是否已下载。如果是这样，它将返回该模块。如果没有，它会在导入之前下载模块。

我怎样才能在 Python 中完成这样的事情，通过它可以无缝地从云端imported 动态范围的子模块？

Answer 1

对您所要求的功能的全功能支持可能需要使用importlib 并连接到导入机制的各个部分的复杂代码。然而，一个更有限的解决方案可以只使用一个伪装成模块的自定义类来实现。

当你导入一个模块时，Python 首先检查sys.modules 字典，看看这个模块是否是一个键。如果是，则返回与键关联的值。无论值是什么，它都会这样做，因此您可以将任何类型的对象放在sys.modules 中，Python 会将其视为模块。一个模块的代码甚至可以替换它自己在sys.modules中的条目，并且即使是第一次imported也会使用替换！

因此，要实现按需下载其他模块的精美模块，请将模块本身替换为自定义类的实例，并为该类编写一个 __getattr__ 或 __getattribute__ 方法来完成您想要的工作。

这是一个简单的示例模块，它为您在其中查找的任何属性返回一个字符串。该字符串将始终与请求的属性名称相同。在您的代码中，您需要执行您喜欢的网络缓存查找和下载，然后返回获取的模块对象，而不是仅仅返回一个字符串。

class FakeModule(object):
    def __getattribute__(self, name):
        return name

import sys
sys.modules[__name__] = FakeModule()

在我的系统上，我已将其保存为 fakemodule.py。现在，如果我执行from fakemodule import foo，我会在我的本地命名空间中得到foo，其值为'foo'。

请注意，这仅适用于一级深度导入。如果你这样做 from fakemodule.subpackage import name 它将不起作用，因为在 sys.modules 中没有 fakemodule.subpackage 条目。