| """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 |
