Cleanup, restore PEP8.

This commit is contained in:
Lance Stout 2010-10-16 21:15:31 -04:00
parent 93fbcad277
commit 505a63da3a
5 changed files with 48 additions and 42 deletions

View file

@ -125,7 +125,6 @@ class ClientXMPP(BaseXMPP):
self.bound = False
self.bindfail = False
def connect(self, address=tuple()):
"""
Connect to the XMPP server.
@ -319,7 +318,9 @@ class ClientXMPP(BaseXMPP):
sasl_ns,
auth))
elif 'sasl:ANONYMOUS' in self.features and not self.boundjid.user:
self.send("<auth xmlns='urn:ietf:params:xml:ns:xmpp-sasl' mechanism='ANONYMOUS' />")
self.send("<auth xmlns='%s' mechanism='%s' />" % (
sasl_ns,
'ANONYMOUS'))
else:
logging.error("No appropriate login method.")
self.disconnect()

View file

@ -108,7 +108,6 @@ class Presence(RootStanza):
self._delAttr('type')
self._delSub('show')
def setPriority(self, value):
"""
Set the entity's priority value. Some server use priority to

View file

@ -167,10 +167,12 @@ class Scheduler(object):
key=lambda task: task.next)
except KeyboardInterrupt:
self.run = False
if self.parentstop is not None: self.parentstop.set()
if self.parentstop is not None:
self.parentstop.set()
except SystemExit:
self.run = False
if self.parentstop is not None: self.parentstop.set()
if self.parentstop is not None:
self.parentstop.set()
logging.debug("Qutting Scheduler thread")
if self.parentqueue is not None:
self.parentqueue.put(('quit', None, None))

View file

@ -239,12 +239,13 @@ class XMLStream(object):
# Repeatedly attempt to connect until a successful connection
# is established.
connected = self.state.transition('disconnected', 'connected', func=self._connect)
connected = self.state.transition('disconnected', 'connected',
func=self._connect)
while reattempt and not connected:
connected = self.state.transition('disconnected', 'connected', func=self._connect)
connected = self.state.transition('disconnected', 'connected',
func=self._connect)
return connected
def _connect(self):
self.stop.clear()
self.socket = self.socket_class(Socket.AF_INET, Socket.SOCK_STREAM)
@ -267,7 +268,8 @@ class XMLStream(object):
return True
except Socket.error as serr:
error_msg = "Could not connect to %s:%s. Socket Error #%s: %s"
logging.error(error_msg % (self.address[0], self.address[1], serr.errno, serr.strerror))
logging.error(error_msg % (self.address[0], self.address[1],
serr.errno, serr.strerror))
time.sleep(1)
return False
@ -283,7 +285,8 @@ class XMLStream(object):
and processing should be restarted.
Defaults to False.
"""
self.state.transition('connected', 'disconnected', wait=0.0, func=self._disconnect, args=(reconnect,))
self.state.transition('connected', 'disconnected', wait=0.0,
func=self._disconnect, args=(reconnect,))
def _disconnect(self, reconnect=False):
# Send the end of stream marker.
@ -305,14 +308,15 @@ class XMLStream(object):
self.event("disconnected", direct=True)
return True
def reconnect(self):
"""
Reset the stream's state and reconnect to the server.
"""
logging.debug("reconnecting...")
self.state.transition('connected', 'disconnected', wait=0.0, func=self._disconnect, args=(True,))
return self.state.transition('disconnected', 'connected', wait=0.0, func=self._connect)
self.state.transition('connected', 'disconnected', wait=0.0,
func=self._disconnect, args=(True,))
return self.state.transition('disconnected', 'connected',
wait=0.0, func=self._connect)
def set_socket(self, socket, ignore=False):
"""
@ -517,8 +521,8 @@ class XMLStream(object):
# processed in the queue.
with self.__event_handlers_lock:
try:
handler_index = self.__event_handlers[name].index(handler)
self.__event_handlers[name].pop(handler_index)
h_index = self.__event_handlers[name].index(handler)
self.__event_handlers[name].pop(h_index)
except:
pass
@ -754,7 +758,7 @@ class XMLStream(object):
if handler.checkDelete():
self.__handlers.pop(self.__handlers.index(handler))
except:
pass #not thread safe
pass # not thread safe
unhandled = False
# Some stanzas require responses, such as Iq queries. A default