diff options
| author | Tom Nichols <tmnichols@gmail.com> | 2010-07-02 16:45:55 -0400 |
|---|---|---|
| committer | Tom Nichols <tmnichols@gmail.com> | 2010-07-02 16:45:55 -0400 |
| commit | 33602f232c0f9e8895cf0f7589d1046f616b6206 (patch) | |
| tree | a626760e611667663847cc881d16d5cfe63d45a2 /sleekxmpp/xmlstream | |
| parent | 7968ca289271caefe09d6501a258fc8490e65abe (diff) | |
| download | slixmpp-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.py | 10 |
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: |