Prevent auto_reconnect interference when disconnecting.

This commit is contained in:
Lance Stout 2012-08-04 21:10:45 -07:00
parent 2324c90232
commit 80a90a6221

View file

@ -692,11 +692,13 @@ class XMLStream(object):
if send_close: if send_close:
self.send_raw(self.stream_footer, now=True) self.send_raw(self.stream_footer, now=True)
if not reconnect:
self.auto_reconnect = False
# Wait for confirmation that the stream was # Wait for confirmation that the stream was
# closed in the other direction. If we didn't # closed in the other direction. If we didn't
# send a stream footer we don't need to wait # send a stream footer we don't need to wait
# since the server won't know to respond. # since the server won't know to respond.
self.auto_reconnect = reconnect
if send_close: if send_close:
log.info('Waiting for %s from server', self.stream_footer) log.info('Waiting for %s from server', self.stream_footer)
self.stream_end_event.wait(4) self.stream_end_event.wait(4)