blob: aca86e87024a65f7c8bbe81b8207eab8a5d58881 [file] [log] [blame]
Ratan Gupta0f2e7fe2016-12-22 19:00:41 +05301#!/usr/bin/env python
2
3import os
4import sys
5import yaml
6import argparse
7from mako.template import Template
8
9
10def generate_hpp(inventory_yaml, output_dir):
11 with open(os.path.join(script_dir, inventory_yaml), 'r') as f:
12 ifile = yaml.safe_load(f)
Brad Bishop3d0ce842017-02-07 08:33:56 -050013 if not isinstance(ifile, dict):
14 ifile = {}
Ratan Gupta0f2e7fe2016-12-22 19:00:41 +053015
16 # Render the mako template
17
18 t = Template(filename=os.path.join(
19 script_dir,
20 "writefru.mako.hpp"))
21
22 output_hpp = os.path.join(output_dir, "fru-gen.hpp")
23 with open(output_hpp, 'w') as fd:
24 fd.write(t.render(fruDict=ifile))
25
26
27def main():
28
29 valid_commands = {
30 'generate-hpp': generate_hpp
31 }
32 parser = argparse.ArgumentParser(
33 description="IPMI FRU parser and code generator")
34
35 parser.add_argument(
36 '-i', '--inventory_yaml', dest='inventory_yaml',
37 default='example.yaml', help='input inventory yaml file to parse')
38
39 parser.add_argument(
40 "-o", "--output-dir", dest="outputdir",
41 default=".",
42 help="output directory")
43
44 parser.add_argument(
45 'command', metavar='COMMAND', type=str,
46 choices=valid_commands.keys(),
47 help='Command to run.')
48
49 args = parser.parse_args()
50
51 if (not (os.path.isfile(os.path.join(script_dir, args.inventory_yaml)))):
52 sys.exit("Can not find input yaml file " + args.inventory_yaml)
53
54 function = valid_commands[args.command]
55 function(args.inventory_yaml, args.outputdir)
56
57if __name__ == '__main__':
58 script_dir = os.path.dirname(os.path.realpath(__file__))
59 main()