summaryrefslogtreecommitdiff
path: root/sleekxmpp/xmlstream
diff options
context:
space:
mode:
authorTom Nichols <tmnichols@gmail.com>2010-07-02 16:45:55 -0400
committerTom Nichols <tmnichols@gmail.com>2010-07-02 16:45:55 -0400
commit33602f232c0f9e8895cf0f7589d1046f616b6206 (patch)
treea626760e611667663847cc881d16d5cfe63d45a2 /sleekxmpp/xmlstream
parent7968ca289271caefe09d6501a258fc8490e65abe (diff)
downloadslixmpp-33602f232c0f9e8895cf0f7589d1046f616b6206.tar.gz
slixmpp-33602f232c0f9e8895cf0f7589d1046f616b6206.tar.bz2
slixmpp-33602f232c0f9e8895cf0f7589d1046f616b6206.tar.xz
slixmpp-33602f232c0f9e8895cf0f7589d1046f616b6206.zip
allow 'ensure' to block if a transition is occurring
Diffstat (limited to 'sleekxmpp/xmlstream')
-rw-r--r--sleekxmpp/xmlstream/statemachine.py10
1 files changed, 7 insertions, 3 deletions
diff --git a/sleekxmpp/xmlstream/statemachine.py b/sleekxmpp/xmlstream/statemachine.py
index fab38dc3..8939397b 100644
--- a/sleekxmpp/xmlstream/statemachine.py
+++ b/sleekxmpp/xmlstream/statemachine.py
@@ -147,11 +147,11 @@ class StateMachine(object):
return _StateCtx(self, from_state, to_state, wait)
- def ensure(self, state, wait=0.0):
+ def ensure(self, state, wait=0.0, block_on_transition=False ):
'''
Ensure the state machine is currently in `state`, or wait until it enters `state`.
'''
- return self.ensure_any( (state,), wait=wait )
+ return self.ensure_any( (state,), wait=wait, block_on_transition=block_on_transition )
def ensure_any(self, states, wait=0.0, block_on_transition=False):
@@ -180,7 +180,11 @@ class StateMachine(object):
# threads to indicate they want to remain in a particular state.
# will return immediately if no transition is in process.
- if block_on_transition: self.notifier.wait()
+ if block_on_transition:
+ # we're not in the middle of a transition; don't hold the lock
+ if self.lock.acquire(False): self.lock.release()
+ # wait for the transition to complete
+ else: self.notifier.wait()
start = time.time()
while not self.__current_state in states: