commit be28ff5647481761f06fc032681393e0b0996626
author Patrick Williams <patrick@stwcx.xyz> Mon Dec 05 10:14:46 2022 -0600
committer Patrick Williams <patrick@stwcx.xyz> Thu Dec 08 11:32:40 2022 -0600
tree 53f645423a4fc3b1cc34a24714a2504e0c2a3b04
parent 064f039fa8eac0da357f8ea8447c74b09c9e0132

python: fix flake8 warnings and format with black

Signed-off-by: Patrick Williams <patrick@stwcx.xyz>
Change-Id: I1b29b18431abc3985c8a2ebe8c0632e39776f5ae

scripts/fru_gen.py

diff --git a/scripts/fru_gen.py b/scripts/fru_gen.py
index 4375086..c79cb64 100755
--- a/scripts/fru_gen.py
+++ b/scripts/fru_gen.py
@@ -1,59 +1,65 @@
 #!/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(inventory_yaml, 'r') as f:
+    with open(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,
-                     "readfru.mako.cpp"))
+        t = Template(filename=os.path.join(script_dir, "readfru.mako.cpp"))
 
         output_hpp = os.path.join(output_dir, "fru-read-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
-    }
-    parser = argparse.ArgumentParser(
-        description="IPMI FRU map code generator")
+    valid_commands = {"generate-cpp": generate_cpp}
+    parser = argparse.ArgumentParser(description="IPMI FRU map 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=valid_commands.keys(),
-        help='Command to run.')
+        help="Command to run.",
+    )
 
     args = parser.parse_args()
 
-    if (not (os.path.isfile(args.inventory_yaml))):
+    if not (os.path.isfile(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()