black: re-format
black and isort are enabled in the openbmc-build-scripts on Python files
to have a consistent formatting. Re-run the formatter on the whole
repository.
Change-Id: I5e068068ef3faa2f2c77bc1726b134eb6268dcfb
Signed-off-by: Patrick Williams <patrick@stwcx.xyz>
diff --git a/scripts/fru_gen.py b/scripts/fru_gen.py
index 5241f2f..624d052 100755
--- a/scripts/fru_gen.py
+++ b/scripts/fru_gen.py
@@ -1,59 +1,67 @@
#!/usr/bin/env python3
+import argparse
import os
import sys
+
import yaml
-import argparse
from mako.template import Template
def generate_cpp(inventory_yaml, output_dir):
- with open(os.path.join(script_dir, inventory_yaml), 'r') as f:
+ with open(os.path.join(script_dir, inventory_yaml), "r") as f:
ifile = yaml.safe_load(f)
if not isinstance(ifile, dict):
ifile = {}
# Render the mako template
- t = Template(filename=os.path.join(
- script_dir,
- "writefru.mako.cpp"))
+ t = Template(filename=os.path.join(script_dir, "writefru.mako.cpp"))
output_hpp = os.path.join(output_dir, "fru-gen.cpp")
- with open(output_hpp, 'w') as fd:
+ with open(output_hpp, "w") as fd:
fd.write(t.render(fruDict=ifile))
def main():
-
- valid_commands = {
- 'generate-cpp': generate_cpp
- }
+ valid_commands = {"generate-cpp": generate_cpp}
parser = argparse.ArgumentParser(
- description="IPMI FRU parser and code generator")
+ description="IPMI FRU parser and code generator"
+ )
parser.add_argument(
- '-i', '--inventory_yaml', dest='inventory_yaml',
- default='example.yaml', help='input inventory yaml file to parse')
+ "-i",
+ "--inventory_yaml",
+ dest="inventory_yaml",
+ default="example.yaml",
+ help="input inventory yaml file to parse",
+ )
parser.add_argument(
- "-o", "--output-dir", dest="outputdir",
+ "-o",
+ "--output-dir",
+ dest="outputdir",
default=".",
- help="output directory")
+ help="output directory",
+ )
parser.add_argument(
- 'command', metavar='COMMAND', type=str,
+ "command",
+ metavar="COMMAND",
+ type=str,
choices=list(valid_commands.keys()),
- help='Command to run.')
+ help="Command to run.",
+ )
args = parser.parse_args()
- if (not (os.path.isfile(os.path.join(script_dir, args.inventory_yaml)))):
+ if not (os.path.isfile(os.path.join(script_dir, args.inventory_yaml))):
sys.exit("Can not find input yaml file " + args.inventory_yaml)
function = valid_commands[args.command]
function(args.inventory_yaml, args.outputdir)
-if __name__ == '__main__':
+
+if __name__ == "__main__":
script_dir = os.path.dirname(os.path.realpath(__file__))
main()