allow specifying multiple suffixes to be stripped
This commit is contained in:
parent
7357bef79c
commit
b7b2641838
|
@ -85,12 +85,13 @@ def get_servers():
|
|||
sys.exit(1)
|
||||
ignorelist = [x.strip() for x in config.get('config', 'ignore', fallback='').split(',')]
|
||||
|
||||
stripsuffix = config.get('config', 'stripsuffix', fallback='')
|
||||
stripsuffixes = [x.strip() for x in config.get('config', 'stripsuffix', fallback='').split(',')]
|
||||
servergroups = yaml.safe_load(open(servergroup))['servergroups']
|
||||
|
||||
def get_display_name(x):
|
||||
if stripsuffix and x.endswith(stripsuffix):
|
||||
return x[:-len(stripsuffix)]
|
||||
for stripsuffix in stripsuffixes:
|
||||
if stripsuffix and x.endswith(stripsuffix):
|
||||
return x[:-len(stripsuffix)]
|
||||
return x
|
||||
|
||||
for group in servergroups:
|
||||
|
|
Loading…
Reference in New Issue