"""license: Apache License 2.0, see LICENSE for more details."""
import random

try:
    from urlparse import urlsplit
except ImportError:
    # try python3 then
    from urllib.parse import urlsplit

def collect_hosts(hosts, randomize=True):
    """Collect a set of hosts and an optional chroot from a string."""
    host_ports, chroot = hosts.partition("/")[::2]
    chroot = "/" + chroot if chroot else None

    result = []
    for host_port in host_ports.split(","):
        # put all complexity of dealing with
        # IPv4 & IPv6 address:port on the urlsplit
        res = urlsplit("xxx://" + host_port)
        host = res.hostname
        port = int(res.port) if res.port else 2181
        result.append((host.strip(), port))

    if randomize:
        random.shuffle(result)

    return result, chroot