diff --git a/bin/roles.py b/bin/roles.py index 1777f8c..32de0cf 100755 --- a/bin/roles.py +++ b/bin/roles.py @@ -6,8 +6,13 @@ roles = [] excluded = ['common-suse', 'minion'] def get(): - for file in os.listdir('salt/role'): - role = os.path.splitext(file)[0] - if not role in excluded: - roles.append(role) + for root in ['pillar', 'salt']: + for rootdir, subdirs, files in os.walk(os.path.join(root, 'role')): + for file in files: + role = os.path.splitext(file)[0] + if not role in excluded and not role in roles: + roles.append(role) return roles + +if __name__ == '__main__': + print(get())