worker
while ループが終了したときに、内から呼び出されるコールバックで thread を None に設定する必要があります。
編集:バックグラウンドプロセスの即時再起動もサポートするようになりました
import time
import threading
class CoolThing(object):
def __init__(self):
self.thread = None
def run_in_background(self, callback, period=0.5):
wait_count = 0
while True:
if self.thread:
if self.thread[1].running or wait_count>10:
raise RuntimeError()
time.sleep(0.5)
wait_count += 1
else:
break
def worker():
t0 = time.time()
worker.running = True
while worker.running:
if time.time()-t0>2:
callback()
t0 = time.time()
time.sleep(period)
worker.callback()
worker.callback = self.dispose
self.thread = (threading.Thread(target=worker), worker)
self.thread[0].start()
def stop_background(self, join=False):
if not self.thread:
raise RuntimeError
self.thread[1].running = False
if join:
self.thread[0].join()
self.stopping = True
def dispose(self):
self.thread = None
self.stopping
def my_callback():
print "Beep"
if __name__=="__main__":
cool_thing = CoolThing()
cool_thing.run_in_background(my_callback, 0.5)
time.sleep(10)
cool_thing.stop_background()
# Immediatley restart process
cool_thing.run_in_background(my_callback, 0.5)
time.sleep(10)
cool_thing.stop_background()
print cool_thing.thread
time.sleep(3)
print cool_thing.thread
出力を与えます:
Beep
Beep
Beep
(<Thread(Thread-2, started 10760)>, <function worker at 0x02DEDD70>)
None
したがって、stop_background を呼び出した直後は、self.thread はまだ設定されていますが、後でNone
. worker.callback-variable を保存して、その名前で dispose() を呼び出すこともできますが、この方法ではコードがより柔軟になります。
編集 2 : 新しい要件、新しいコード サンプル
ワーカー (SRP) 用に別のクラスを作成し、CoolThing はそのようなワーカーのリストを保持します。がrun_background(...)
開始されると、まだ実行中のワーカーがあるかどうか (停止要求がないかどうか) を確認し、RuntimeError を発生させます。それ以外の場合は、新しいワーカーを開始します。
stop_background()
各ワーカーに停止するように指示し、各ワーカーがコールバックを呼び出します。これにより、このワーカーがすべてのワーカーのリストから削除されます。
import time
import threading
class Worker(threading.Thread):
def __init__(self, callback, period=0.5, finished_callback = None):
threading.Thread.__init__(self)
self.callback = callback
self.period = period
self._stop_requested = False
self._finished_callback = finished_callback
def run(self):
t0 = time.time()
while not self._stop_requested:
if time.time()-t0>2:
self.callback()
t0 = time.time()
time.sleep(self.period)
if self._finished_callback:
self._finished_callback(self)
def request_stop(self):
self._stop_requested = True
@property
def stopping(self):
return self._stop_requested
class CoolThing(object):
def __init__(self):
self.workers = []
self.workers_lock = threading.Lock()
def run_in_background(self, callback, period=0.5):
if len([w for w in self.workers if not w.stopping])>0:
raise RuntimeError()
worker = Worker(callback, period, finished_callback=self.dispose)
with self.workers_lock:
self.workers.append(worker)
worker.start()
def stop_background(self, join=False):
if len(self.workers) == 0:
raise RuntimeError()
for worker in self.workers:
worker.request_stop()
if join:
for worker in self.workers:
worker.join()
def dispose(self, worker):
with self.workers_lock:
self.workers.remove(worker)
def my_callback():
print "Beep"
if __name__=="__main__":
cool_thing = CoolThing()
cool_thing.run_in_background(my_callback, 0.5)
time.sleep(10)
print cool_thing.workers
cool_thing.stop_background()
# Immediatley restart process
cool_thing.run_in_background(my_callback, 0.5)
print cool_thing.workers
time.sleep(5)
print cool_thing.workers
time.sleep(5)
cool_thing.stop_background()
print cool_thing.workers
time.sleep(3)
print cool_thing.workers