39 lines
862 B
Python
Executable file
39 lines
862 B
Python
Executable file
#!/usr/bin/env python3
|
|
#
|
|
# Puppet 'External Node Classifier' to tell puppet what classes to apply to this node.
|
|
#
|
|
# Docs: https://puppet.com/docs/puppet/5.3/nodes_external.html
|
|
#
|
|
|
|
import os
|
|
import re
|
|
import sys
|
|
|
|
import yaml
|
|
|
|
rules_path = os.environ.get("COSMOS_RULES_PATH", "/etc/puppet")
|
|
|
|
node_name = sys.argv[1]
|
|
|
|
rules = dict()
|
|
for p in rules_path.split(":"):
|
|
rules_file = os.path.join(p, "cosmos-rules.yaml")
|
|
if os.path.exists(rules_file):
|
|
with open(rules_file) as fd:
|
|
rules.update(yaml.safe_load(fd))
|
|
|
|
found = False
|
|
classes = dict()
|
|
for reg, cls in rules.items():
|
|
if re.search(reg, node_name):
|
|
if cls:
|
|
classes.update(cls)
|
|
found = True
|
|
|
|
if not found:
|
|
sys.stderr.write(f"{sys.argv[0]}: {node_name} not found in cosmos-rules.yaml\n")
|
|
|
|
print("---\n" + yaml.dump(dict(classes=classes)))
|
|
|
|
sys.exit(0)
|