Mercurial > repos > shellac > guppy_basecaller
diff env/lib/python3.7/site-packages/galaxy/util/sockets.py @ 0:26e78fe6e8c4 draft
"planemo upload commit c699937486c35866861690329de38ec1a5d9f783"
author | shellac |
---|---|
date | Sat, 02 May 2020 07:14:21 -0400 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/env/lib/python3.7/site-packages/galaxy/util/sockets.py Sat May 02 07:14:21 2020 -0400 @@ -0,0 +1,43 @@ +import random +import shlex +import socket +import subprocess + + +def unused_port(range=None): + if range: + return __unused_port_on_range(range) + else: + return __unused_port_rangeless() + + +def __unused_port_rangeless(): + # TODO: Allow ranges (though then need to guess and check)... + s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + s.bind(('localhost', 0)) + addr, port = s.getsockname() + s.close() + return port + + +def __unused_port_on_range(range): + assert range[0] and range[1] + + # Find all ports that are already occupied + cmd_netstat = shlex.split("netstat tuln") + p1 = subprocess.Popen(cmd_netstat, stdout=subprocess.PIPE) + + occupied_ports = set() + for line in p1.stdout.read().split('\n'): + if line.startswith('tcp') or line.startswith('tcp6'): + col = line.split() + local_address = col[3] + local_port = local_address.split(':')[1] + occupied_ports.add(int(local_port)) + + # Generate random free port number. + while True: + port = random.randrange(range[0], range[1]) + if port not in occupied_ports: + break + return port