如何使语句 `import datetime` 绑定 datetime 类型而不是 datetime 模块？

Question

在真正需要的是 from datetime import datetime 时不小心输入了太多次 import datetime 之后，我想知道是否有可能破解并让前者执行后者。

也就是说，要重新创建此行为（在新打开的解释器会话中）：

$ python -ic ''
>>> import datetime
>>> datetime(2016, 5, 27)
datetime.datetime(2016, 5, 27, 0, 0)

差点伪造下面的 "callable module"：

>>> import dt
>>> dt(2016, 5, 27)
datetime.datetime(2016, 5, 27, 0, 0)

是这样实现的：

# dt.py
import sys
import datetime

class CallableModule(object):
    def __init__(self, thing):
        self.thing = thing
    def __call__(self, *args, **kwargs):
        return self.thing.__call__(*args, **kwargs)

sys.modules['dt'] = CallableModule(datetime.datetime)

但是，如果我尝试使用模块的文件名 datetime.py，它就不起作用，而且当我自己的文件也是称为 datetime.py。

我们如何暂时取消隐藏 隐藏模块本身中的内置或站点包模块？在这种情况下，有没有什么间接的方法可以到达核心datetime（也许类似于我们如何仍然可以访问sys.__stdout__，即使sys.stdout已经被重定向）？

免责声明：绝不暗示这是一个明智的想法 - 只是对可能感兴趣。

Answer 1

我们开始：

datetime.py:

import sys
import imp
import os

path = [p for p in sys.path if p != os.path.dirname(__file__)]
f, pathname, desc = imp.find_module('datetime', path)
std_datetime = imp.load_module('datetime', f, pathname, desc)
# if this^ is renamed to datetime, everything breaks!
f.close()

class CallableModule(object):

    def __init__(self, module, fn):
        self.module = module
        self.fn = fn

    def __call__(self, *args, **kwargs):
        return self.fn(*args, **kwargs)

    def __getattr__(self, item):
        try:
            return getattr(self.fn, item)
        except AttributeError:
            return getattr(self.module, item)

sys.modules['datetime'] = CallableModule(std_datetime, std_datetime.datetime)

test.py（住在datetime.py隔壁）：

import datetime

print(datetime(1, 2, 3))
print(datetime.timedelta(days=1))
print(datetime.now())

运行test.py，输出：

0001-02-03 00:00:00
1 day, 0:00:00
2016-05-27 23:16:30.954270

它也适用于 from datetime import datetime, timedelta 等

这特别棘手和脆弱，将取决于您的发行版。例如，显然它不适用于 IPython。您必须在标准库模块之前导入 datetime.py。

要了解这有多么奇怪，如果变量 std_datetime（日期时间模块对象）重命名为 datetime，则 datetime.datetime 不再是 class，而是 datetime is datetime.datetime is datetime.datetime.datetime ...。如果有人能解释为什么会这样，我很想听听。

（请注意，下面的第一条评论是在我到达最终版本之前）

Answer 2

对最初的想法稍作修改，这实际上适用于 python3:

# datetime.py
import sys, _datetime
sys.modules['datetime'] = _datetime.datetime

如何使语句 `import datetime` 绑定 datetime 类型而不是 datetime 模块？

How to make statement `import datetime` bind the datetime type instead of the datetime module?

python

python-module

python-import