python - 如何使语句 `import datetime` 绑定(bind) datetime 类型而不是 datetime 模块？

Question

当真正需要的是 from datetime import datetime 时不小心输入了太多次 import datetime 之后，我想知道是否有可能四处乱搞并制作前者做后者。

也就是说，要重新创建此行为(在新打开的解释器 session 中):

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

差点伪造下面的“可调用模块”:

>>> 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，它就不起作用，而且我还找不到任何 hack 来访问内置的 datetime 模块，而我自己文件也称为 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.datetime 不再是类，而是 datetime is datetime.datetime is datetime.datetime.datetime ...。如果有人能解释为什么会这样，我很乐意听听。

(请注意，下面的第一条评论是在我到达最终版本之前发表的)