jjb/cperf/parse-node-yaml.py

   1 ##############################################################################
   2 # Copyright (c) 2018 Tim Rozet (trozet@redhat.com) and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10 import argparse
  11 import sys
  12 import yaml
  13
  14
  15 def get_node_data_by_number(node_type, node_number):
  16     node_idx = 1
  17     for node_name, node_data in data['servers'].items():
  18         if node_type == node_data['type']:
  19             if node_idx == node_number:
  20                 return node_name, node_data
  21             else:
  22                 node_idx += 1
  23
  24
  25 def get_node_value(node_type, node_number, key):
  26     node_name, node_data = get_node_data_by_number(node_type, node_number)
  27     if not key and node_name is not None:
  28         return node_name
  29     elif node_data and isinstance(node_data, dict) and key in node_data:
  30         return node_data[key]
  31
  32
  33 def get_number_of_nodes(node_type):
  34     nodes = data['servers']
  35     num_nodes = 0
  36     for node_name, node_data in nodes.items():
  37         if node_data['type'] == node_type:
  38             num_nodes += 1
  39     return num_nodes
  40
  41
  42 FUNCTION_MAP = {'num_nodes':
  43                 {'func': get_number_of_nodes,
  44                  'args': ['node_type']},
  45                 'get_value':
  46                     {'func': get_node_value,
  47                      'args': ['node_type', 'node_number', 'key']},
  48                 }
  49
  50 if __name__ == "__main__":
  51     parser = argparse.ArgumentParser()
  52     parser.add_argument('command', choices=FUNCTION_MAP.keys())
  53     parser.add_argument('-f', '--file',
  54                         dest='node_file',
  55                         required=True)
  56     parser.add_argument('--node-type',
  57                         default='controller',
  58                         required=False)
  59     parser.add_argument('--node-number',
  60                         default=1,
  61                         type=int,
  62                         required=False)
  63     parser.add_argument('-k', '--key',
  64                         required=False)
  65     args = parser.parse_args(sys.argv[1:])
  66     with open(args.node_file, 'r') as fh:
  67         data = yaml.safe_load(fh)
  68     assert 'servers' in data
  69     func = FUNCTION_MAP[args.command]['func']
  70     args = [getattr(args, x) for x in FUNCTION_MAP[args.command]['args']]
  71     print(func(*args))