blob: 8c28ddb243f0dc80397a70bcca8f84d14be772b5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#!/usr/bin/env python3
import yaml
import sys
# https://puppet.com/docs/puppet/7/nodes_external.html
with open('/nodes.yaml') as f:
data = yaml.load(f)
def main(args):
if len(args) == 1:
print('usage ./node-classifier <node-name>')
return
nodename = sys.argv[1]
instance = data.get(nodename)
if not instance:
instance = data.get('default')
print(yaml.dump(instance))
if __name__ == '__main__':
sys.exit(main(sys.argv))
|