Fix confusing return statements, closes #67.

This commit is contained in:
David Wilson 2017-11-12 11:14:51 +05:30
parent 3831ac360f
commit bbcf1a0bd4
2 changed files with 4 additions and 2 deletions

View File

@ -80,7 +80,8 @@ class Stream(mitogen.master.Stream):
time.time() + 10.0):
LOG.debug('%r: received %r', self, buf)
if buf.endswith('EC0\n'):
return self._ec0_received()
self._ec0_received()
return
elif PERMDENIED_PROMPT in buf.lower():
if self.password is not None and password_sent:
raise PasswordError(self.password_incorrect_msg)

View File

@ -71,7 +71,8 @@ class Stream(mitogen.master.Stream):
time.time() + 10.0):
LOG.debug('%r: received %r', self, buf)
if buf.endswith('EC0\n'):
return self._ec0_received()
self._ec0_received()
return
elif PASSWORD_PROMPT in buf.lower():
if self.password is None:
raise PasswordError(self.password_required_msg)