Python如何全局替换内置datetime模块为自定义扩展实现

补丁失效核心是三个常见的猴子补丁踩坑点：

• 你在extensions/__init__.py里第一行执行import datetime as _datetime时，Python已经把原生datetime模块加载进内存。后续哪怕修改sys.modules的映射，已经完成导入的代码持有的是原生模块的直接引用，不会自动指向自定义版本。
• 自定义extensions.datetime里写的from datetime import *，执行时会直接把原生datetime的所有属性（包括原生date类）拷贝到当前模块命名空间。如果其他模块在补丁生效前就写了from datetime import date，拿到的是原生类的直接引用，后续修改模块映射不会改变这个已绑定的引用。
• importlib.reload()不生效是因为模块的顶层导入逻辑仅在第一次导入时执行，reload不会让已经绑定完成的类引用重新从sys.modules拉取最新值。

要实现全局替换原生datetime，核心原则是在任何模块（包括标准库、第三方库、你自己的业务代码）导入datetime之前完成补丁，具体步骤如下：

1. 修正自定义datetime模块的写法 #

修改extensions/datetime.py，不要在顶层直接用通配符导入原生datetime，避免提前绑定原生类：

import sys
import datetime as _original_datetime

# 先把当前模块注册到sys.modules，替换原生datetime的映射
sys.modules["datetime"] = sys.modules[__name__]

# 拷贝原生datetime的所有属性到当前模块，保证接口100%兼容
for _attr in dir(_original_datetime):
    if _attr not in globals():
        globals()[_attr] = getattr(_original_datetime, _attr)

# 重写自定义date类，继承原生date的所有逻辑
class date(_original_datetime.date):
    @property
    def test(self):
        print("test")

2. 把补丁放到程序入口最开头执行 #

必须保证补丁是程序启动后第一行运行的代码，删掉extensions/__init__.py里原来写的sys.modules替换逻辑，直接修改main.py：

# 第一行就执行datetime补丁，这行之前不能有任何其他导入
import extensions.datetime

# 补丁打完之后，再导入所有其他依赖
import extensions
from extensions.calendarweek import CalendarWeek

def main():
    week_start = CalendarWeek()[0]
    print(week_start.__class__)  # 输出<class 'extensions.datetime.date'>
    week_start.test  # 正常打印test

if __name__ == "__main__":
    main()

• 补丁代码必须放在程序入口的最顶部，补丁执行前不能出现任何import datetime、导入依赖datetime的第三方库的代码，否则这些模块会提前持有原生datetime的引用，补丁直接失效。
• 如果需要自定义datetime、timedelta、time等其他datetime相关类，和重写date的逻辑一致：继承原生对应类实现自定义逻辑后，直接覆盖当前模块的同名属性即可，原生接口的所有行为会完整保留，不会破坏现有代码。
• 不需要单独替换sys.modules里的calendarweek映射，只要保证calendarweek在datetime补丁完成后才被导入，它内部引用的datetime自然是你的自定义版本。

内容的提问来源于stack exchange，提问作者user12281826