wbrawner/ansible
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add support for multiple ranges in a host

Browse source 
e.g.  db[01:10:3]node-[01:10]

- to do this we split off at the first [...] set, getting the list
  of hosts and then repeat until none left.
- also add an optional third parameter which contains the step. (Default: 1)
  so range can be [01:10:2] -> 01 03 05 07 09
This commit is contained in:
Simon J Mudd 2013-06-06 22:27:30 +02:00
parent 07a1365c55
commit 25822330a2
1 changed files with 20 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
lib/ansible/inventory/expand_hosts.py
View file

@ -70,12 +70,24 @@ def expand_hostname_range(line = None):
# nrange: [1:6]; range() is a built-in. Can't use the name # nrange: [1:6]; range() is a built-in. Can't use the name
# tail: '-node' # tail: '-node'
(head, nrange, tail) = line.replace('[','|').replace(']','|').split('|') # Add support for multiple ranges in a host so:
# db[01:10:3]node-[01:10]
# - to do this we split off at the first [...] set, getting the list
# of hosts and then repeat until none left.
# - also add an optional third parameter which contains the step. (Default: 1)
# so range can be [01:10:2] -> 01 03 05 07 09
# FIXME: make this work for alphabetic sequences too.
(head, nrange, tail) = line.replace('[','|',1).replace(']','|',1).split('|')
bounds = nrange.split(":") bounds = nrange.split(":")
if len(bounds) != 2: if len(bounds) != 2 and len(bounds) != 3:
raise errors.AnsibleError("host range incorrectly specified") raise errors.AnsibleError("host range incorrectly specified")
beg = bounds[0] beg = bounds[0]
end = bounds[1] end = bounds[1]
if len(bounds) == 2:
step = 1
else:
step = bounds[2]
if not beg: if not beg:
beg = "0" beg = "0"
if not end: if not end:
@ -95,10 +107,14 @@ def expand_hostname_range(line = None):
raise errors.AnsibleError("host range format incorrectly specified!") raise errors.AnsibleError("host range format incorrectly specified!")
seq = string.ascii_letters[i_beg:i_end+1] seq = string.ascii_letters[i_beg:i_end+1]
except ValueError: # not a alpha range except ValueError: # not a alpha range
seq = range(int(beg), int(end)+1) seq = range(int(beg), int(end)+1, int(step))
for rseq in seq: for rseq in seq:
hname = ''.join((head, fill(rseq), tail)) hname = ''.join((head, fill(rseq), tail))
if detect_range(hname):
all_hosts.extend( expand_hostname_range( hname ) )
else:
all_hosts.append(hname) all_hosts.append(hname)
return all_hosts return all_hosts