Недавно я опубликовал question о том, как отложить выполнение функции в Python (эквивалент Javascript setTimeout
), и получается быть простой задачей, используя threading.Timer
(ну, просто, пока функция не имеет общего состояния с другим кодом, но это создавало бы проблемы в любой среде, управляемой событиями).
Теперь я стараюсь делать лучше и эмулировать setInterval
. Для тех, кто не знаком с Javascript, setInterval
позволяет повторять вызов функции каждые x секунд, не блокируя выполнение другого кода. Я создал этот декоратор:
import time, threading
def setInterval(interval, times = -1):
# This will be the actual decorator,
# with fixed interval and times parameter
def outer_wrap(function):
# This will be the function to be
# called
def wrap(*args, **kwargs):
# This is another function to be executed
# in a different thread to simulate setInterval
def inner_wrap():
i = 0
while i != times:
time.sleep(interval)
function(*args, **kwargs)
i += 1
threading.Timer(0, inner_wrap).start()
return wrap
return outer_wrap
для использования следующим образом
@setInterval(1, 3)
def foo(a):
print(a)
foo('bar')
# Will print 'bar' 3 times with 1 second delays
и мне кажется, что он работает нормально. Моя проблема в том, что
- кажется слишком сложным, и я боюсь, что, возможно, пропустил более простой/лучший механизм.
- декоратор можно вызвать без второго параметра, и в этом случае он будет продолжаться вечно. Когда я говорю "foreover", я имею в виду навсегда - даже вызов
sys.exit()
из основного потока не остановит его и не ударитCtrl+c
. Единственный способ остановить это - убить процесс python извне. Я хотел бы иметь возможность посылать сигнал из основного потока, который остановит обратный вызов. Но я начинаю с потоками - как я могу общаться между ними?
EDIT В случае, если кто-то задается вопросом, это окончательная версия декоратора, благодаря помощи jd
import threading
def setInterval(interval, times = -1):
# This will be the actual decorator,
# with fixed interval and times parameter
def outer_wrap(function):
# This will be the function to be
# called
def wrap(*args, **kwargs):
stop = threading.Event()
# This is another function to be executed
# in a different thread to simulate setInterval
def inner_wrap():
i = 0
while i != times and not stop.isSet():
stop.wait(interval)
function(*args, **kwargs)
i += 1
t = threading.Timer(0, inner_wrap)
t.daemon = True
t.start()
return stop
return wrap
return outer_wrap
Он может использоваться с фиксированным количеством повторений, как указано выше
@setInterval(1, 3)
def foo(a):
print(a)
foo('bar')
# Will print 'bar' 3 times with 1 second delays
или может быть оставлен для запуска до тех пор, пока он не получит сигнал останова
import time
@setInterval(1)
def foo(a):
print(a)
stopper = foo('bar')
time.sleep(5)
stopper.set()
# It will stop here, after printing 'bar' 5 times.