python制限関数実行回数

861 ワード

この間pythonの職場を面接して、1つの問題に出会って、どのように関数を指定の時間内に1回だけ実行することを制限しますか?この問題で得られたpython修飾関数.
解決策は次のとおりです.
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()