added pubsub state stanzas and scheduled events

This commit is contained in:
Nathan Fritz
2010-05-27 19:58:57 +08:00
committed by Thom Nichols
parent 2e7024419a
commit 194e6bcb51
8 changed files with 67 additions and 23 deletions

View File

@@ -180,6 +180,7 @@ class XMLStream(object):
self.state.set('reconnect', False)
self.disconnect()
self.run = False
self.scheduler.run = False
self.eventqueue.put(('quit', None, None))
return
except CloseStream:
@@ -228,6 +229,7 @@ class XMLStream(object):
edepth += -1
if edepth == 0 and event == b'end':
self.disconnect(reconnect=self.state['reconnect'])
logging.debug("Ending readXML loop")
return False
elif edepth == 1:
#self.xmlin.put(xmlobj)
@@ -236,11 +238,13 @@ class XMLStream(object):
except RestartStream:
return True
except CloseStream:
logging.debug("Ending readXML loop")
return False
if root:
root.clear()
if event == b'start':
edepth += 1
logging.debug("Ending readXML loop")
def _sendThread(self):
while self.run:
@@ -270,6 +274,7 @@ class XMLStream(object):
logging.debug("Disconnecting...")
self.state.set('disconnecting', True)
self.run = False
self.scheduler.run = False
if self.state['connected']:
self.sendRaw(self.stream_footer)
time.sleep(1)
@@ -341,7 +346,8 @@ class XMLStream(object):
args[0].exception(e)
elif etype == 'schedule':
try:
handler.run(*args)
logging.debug(args)
handler(*args[0])
except:
logging.error(traceback.format_exc())
elif etype == 'quit':