限制函数在规定时间内只能执行一次

最新推荐文章于 2021-05-19 06:51:20 发布

转载最新推荐文章于 2021-05-19 06:51:20 发布 · 2k 阅读

本文介绍了一个Python装饰器，用于控制函数的调用频率，确保函数至少在指定的时间间隔后才能再次运行。通过使用datetime和timedelta模块，该装饰器能够有效防止函数在短时间内被重复调用。

https://www.jianshu.com/p/a5412c1815dc

import time
from datetime import datetime
from datetime import timedelta


def timeguard(time_interval, default=None):
    def decorator(function):
        # For first time always run the function
        function.__last_run = datetime.min
        def guard(*args, **kwargs):
            now = datetime.now()
            if now - function.__last_run >= time_interval:
                function.__last_run = now
                return function(*args, **kwargs)
            elif default is not None:
                return default(*args, **kwargs)
        return guard
    return decorator


if __name__ == "__main__":
    
    @timeguard(timedelta(seconds=4),None)
    def add():
        print 12

    add()
    add()
    add()