find_available_ports.py

find_available_ports.py

def get_available_ports(port_from=5000, port_to=6000):
    import socket
    for port in range(port_from, port_to):
        try:
            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            s.bind(('localhost', port))
            s.close()
            yield port
        except:
            continue


def get_port():
    global port_generator

    for _ in range(2):
        try:
            return next(port_generator)
        except NameError:
            port_generator = get_available_ports()
    raise Exception("Get port failed!")