From: Ralf Jung <post@ralfj.de>
Date: Sat, 26 Oct 2013 20:16:08 +0000 (+0200)
Subject: oops, Waker needs locking
X-Git-Url: https://git.ralfj.de/saartuer.git/commitdiff_plain/226118cbaed9b2f092e2ae764260bf3b63951938

oops, Waker needs locking
---

diff --git a/waker.py b/waker.py
index 1cf2884..9c2cf06 100644
--- a/waker.py
+++ b/waker.py
@@ -1,5 +1,6 @@
 from libtuer import ThreadRepeater
 from collections import namedtuple
+from threading import Lock
 
 SLEEP_TIME = 0.5
 
@@ -10,26 +11,29 @@ class Waker():
 		self._sm = sm
 		self._t = ThreadRepeater(self._wake, SLEEP_TIME, name="Waker")
 		self._tobewokens = []
+		self._tobewokens_lock = Lock()
 	
 	def register(f, time, one_shot = False):
 		'''Register a function which is called approximately every <time> seconds (or just once, if one_shot is True). f should return quickly, or it will delay the waker!'''
 		time = max(time//SLEEP_TIME, 1)
-		self._tobewokens.append(ToBeWoken(f, time, 0, one_shot))
+		with self._tobewokens_lock:
+			self._tobewokens.append(ToBeWoken(f, time, 0, one_shot))
 	
 	def _wake(self):
-		delete = []
-		# run the functions we ought to run
-		for tobewoken, idx in zip(self._tobewokens, range(len(self._tobewokens))):
-			tobewoken.time_since_call += 1
-			if tobewoken.time_since_call >= tobewoken.period:
-				tobewoken.f()
-				tobewoken.time_since_call = 0
-				if tobewoken.one_shot:
-					delete.append(idx)
-		# delete what we have to delete - in reverse order so the indices stay valid!
-		delete.reverse()
-		for idx in delete:
-			del self._tobewokens[idx]
+		with self._tobewokens_lock:
+			delete = []
+			# run the functions we ought to run
+			for tobewoken, idx in zip(self._tobewokens, range(len(self._tobewokens))):
+				tobewoken.time_since_call += 1
+				if tobewoken.time_since_call >= tobewoken.period:
+					tobewoken.f()
+					tobewoken.time_since_call = 0
+					if tobewoken.one_shot:
+						delete.append(idx)
+			# delete what we have to delete - in reverse order so the indices stay valid!
+			delete.reverse()
+			for idx in delete:
+				del self._tobewokens[idx]
 	
 	def stop(self):
 		self._t.stop()