diff --git a/eoptasks/eoptasks.py b/eoptasks/eoptasks.py index b204514..1daa700 100755 --- a/eoptasks/eoptasks.py +++ b/eoptasks/eoptasks.py @@ -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: