wukan1986 / mocktime

模拟时间工具

mocktime

通过对time.time()和datetime.now()的修改，实现模拟时间功能。可用于历史事件回测，或未来事件调试。

项目起源

1. 本项目来自于使用定时任务调度框架实现程序化交易回测的一个设想
2. 策略一般由行情事件驱动，但在开盘前和收盘后无行情，只能靠定时任务来驱动。如：before_trading,after_trading
3. K线行情事件本质上可认为也是固定频率的定时任务。如：on_bar
4. 定时任务框架一般只能用于实盘，因为它的时间机制是基于操作系统，而回测的时间是基于历史数据
5. 框架中大量的使用到datetime.now(), 它无法根据历史进行修改，而本项目就是为了打破这个限制而做的尝试
6. 最先支持了apscheduler，之后支持了日志库logging和loguru
7. 编写了schedule的示例后，想到可能还有其它调度库需要模拟时间
8. 后来还想到模拟时间功能可以快速的检测任务规则是否编写正确，很有必要独立成一个项目

原理

1. time.time()打补丁，使其可以修改成任意时间，这样其它模块就可以取到历史时间
2. datetime.now()中使用了time.time()，替换它成历史时间
3. 每个任务执行完后更新time.time()为下次任务的触发时间

安装

pip install mocktime -U

示例

import time
from datetime import datetime

import mocktime

if __name__ == '__main__':
    mocktime.configure(mock=True, tick=False)
    print('真实时间:', mocktime.now(), mocktime.time())
    print('默认模拟时间，带时区:', datetime.now(), time.time())

    mocktime.configure(mock=False, tick=False)
    print('还原成真实时间:', datetime.now(), time.time())

    mocktime.configure(mock=True, tick=False)
    mocktime.time_update(datetime(2023, 1, 2, 3, 4, 5))
    print('更新模拟时间:', datetime.now())

    mocktime.time_add(5.1)
    print('模拟时间+5.1s:', datetime.now())

    mocktime.configure(mock=True, tick=True)
    time.sleep(5)
    print('开启时间流逝，等5s，时间更新:', datetime.now())
    mocktime.configure(mock=True, tick=False)
    time.sleep(5)
    print('关闭时间流逝，等5s，时间不变:', datetime.now())

    mocktime.configure(mock=True, tick=True)
    print('开启时间流逝后再更新时间:', datetime.now())
    mocktime.time_update(datetime(2023, 1, 2, 3, 4, 5))
    time.sleep(5)
    print('从新设置的时间开始流逝:', datetime.now())

输出如下

真实时间: 2023-11-24 15:37:16.168329 1700811436.1683288
默认模拟时间，带时区: 1970-01-01 08:00:00 0.0
还原成真实时间: 2023-11-24 15:37:16.168329 1700811436.1683288
更新模拟时间: 2023-01-02 03:04:05
模拟时间+5.1s: 2023-01-02 03:04:10.100000
开启时间流逝，等5s，时间更新: 2023-01-02 03:04:15.106348
关闭时间流逝，等5s，时间不变: 2023-01-02 03:04:15.106348
开启时间流逝后再更新时间: 2023-01-02 03:04:15.106348
从新设置的时间开始流逝: 2023-01-02 03:04:10.010793

调度任务示例

apscheduler examples/apscheduler/demo.py

1. 导入mocktime后，设置好过去的开始与结束时间，但任务并没有继续执行。因为模拟时间是固定不变的
2. 需继承BlockingScheduler，在每批次的任务执行完后立即更新时间mocktime.time_add

schedule examples/schedule/demo.py

1. 同样遇到需更新模拟时间问题
2. 在循环中取get_next_run，然后更新时间mocktime.time_update

dagster examples/dagster/usercustomize.py

1. 难点在于mocktime需要跨进程,通过mocktime.multiprocess实现
2. 由于跨进程通讯复杂，所以只实现了tick=True的功能

模拟时间三种运行模式

1. 真实时间模式。当前真实时间开始、正常速度执行。可以不需mocktime参与

   mocktime.configure(mock=False, tick=False)

2. 加速模拟模式。任意时间开始(历史和未来都可以)，加速执行。需mocktime修改时间，还需在任务循环中更新mocktime.time_update 。一般用于历史回测

   mocktime.configure(mock=True, tick=False)

3. 时间流逝模式。任意时间开始(历史和未来都可以)，正常速度执行。仅需mocktime修改时间。一般用于任务触发测试，可免于修改操作系统时间

   mocktime.configure(mock=True, tick=True)

4. 不存在的模式。时间流逝，但又不是模拟模式，不就是真实时间吗？！

   mocktime.configure(mock=False, tick=True) # 当tick=True时，强行mock=True

5. 跨进程模式。只是时间流逝模式configure(mock=True, tick=True)的一种应用场景

   mocktime.multiprocess(r'd:\a.pkl', datetime(2022, 1, 1).timestamp())

参考资料

1.《深度剖析CPython解释器》34. 侵入Python虚拟机，动态修改底层数据结构和运行时 - https://www.cnblogs.com/traditional/p/15489296.html