Commit 2f5c59ac authored by Luke Macken's avatar Luke Macken

pep8

parent 25ccafbc
...@@ -74,17 +74,18 @@ class PyrasiteIPC(object): ...@@ -74,17 +74,18 @@ class PyrasiteIPC(object):
def listen(self): def listen(self):
"""Listen on a random port""" """Listen on a random port"""
for res in socket.getaddrinfo('localhost', None, socket.AF_UNSPEC, socket.SOCK_STREAM, 0, 0): for res in socket.getaddrinfo('localhost', None, socket.AF_UNSPEC,
socket.SOCK_STREAM, 0, 0):
af, socktype, proto, canonname, sa = res af, socktype, proto, canonname, sa = res
try: try:
self.server_sock = socket.socket(af, socktype, proto) self.server_sock = socket.socket(af, socktype, proto)
except socket.error, msg: except socket.error:
self.server_sock = None self.server_sock = None
continue continue
try: try:
self.server_sock.bind(sa) self.server_sock.bind(sa)
self.server_sock.listen(1) self.server_sock.listen(1)
except socket.error, msg: except socket.error:
self.server_sock.close() self.server_sock.close()
self.server_sock = None self.server_sock = None
continue continue
......
...@@ -50,16 +50,17 @@ class ReverseConnection(threading.Thread, PyrasiteIPC): ...@@ -50,16 +50,17 @@ class ReverseConnection(threading.Thread, PyrasiteIPC):
running = True running = True
while running: while running:
try: try:
for res in socket.getaddrinfo(self.host, self.port, socket.AF_UNSPEC, socket.SOCK_STREAM): for res in socket.getaddrinfo(self.host, self.port,
socket.AF_UNSPEC, socket.SOCK_STREAM):
af, socktype, proto, canonname, sa = res af, socktype, proto, canonname, sa = res
try: try:
self.sock = socket.socket(af, socktype, proto) self.sock = socket.socket(af, socktype, proto)
except socket.error, msg: except socket.error:
self.sock = None self.sock = None
continue continue
try: try:
self.sock.connect(sa) self.sock.connect(sa)
except socket.error, msg: except socket.error:
self.sock.close() self.sock.close()
self.sock = None self.sock = None
continue continue
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment