#!/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))