blob: ca0dd35e580d9e1f20d98da9446dd80b1c19f679 [file] [log] [blame]
"""license: Apache License 2.0, see LICENSE for more details."""
import random
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:
return result, chroot