From: Simon Glass <sjg@chromium.org>
Date: Sun, 31 Jul 2022 02:57:08 +0000 (-0600)
Subject: dtoc: Move main program into its own function
X-Git-Tag: v2025.01-rc5-pxa1908~1315^2~3
X-Git-Url: http://git.dujemihanovic.xyz/html/index.html?a=commitdiff_plain;h=a8ad9aacd34cc1fd1dc7c0c8703c9467d76676b9;p=u-boot.git

dtoc: Move main program into its own function

Use a function for the main program so everything there doesn't look like
a global variable to pylint.

Signed-off-by: Simon Glass <sjg@chromium.org>
---

diff --git a/tools/dtoc/test_fdt.py b/tools/dtoc/test_fdt.py
index afa0bb5885..e10fb528e8 100755
--- a/tools/dtoc/test_fdt.py
+++ b/tools/dtoc/test_fdt.py
@@ -799,23 +799,27 @@ def run_tests(args, processes):
     return (0 if result.wasSuccessful() else 1)
 
 
-if __name__ != '__main__':
-    sys.exit(1)
-
-parser = OptionParser()
-parser.add_option('-B', '--build-dir', type='string', default='b',
-        help='Directory containing the build output')
-parser.add_option('-P', '--processes', type=int,
-                  help='set number of processes to use for running tests')
-parser.add_option('-t', '--test', action='store_true', dest='test',
-                  default=False, help='run tests')
-parser.add_option('-T', '--test-coverage', action='store_true',
-                default=False, help='run tests and check for 100% coverage')
-(options, args) = parser.parse_args()
-
-# Run our meagre tests
-if options.test:
-    ret_code = run_tests(args, options.processes)
-    sys.exit(ret_code)
-elif options.test_coverage:
-    run_test_coverage(options.build_dir)
+def main():
+    """Main program for this tool"""
+    parser = OptionParser()
+    parser.add_option('-B', '--build-dir', type='string', default='b',
+            help='Directory containing the build output')
+    parser.add_option('-P', '--processes', type=int,
+                      help='set number of processes to use for running tests')
+    parser.add_option('-t', '--test', action='store_true', dest='test',
+                      default=False, help='run tests')
+    parser.add_option('-T', '--test-coverage', action='store_true',
+                    default=False, help='run tests and check for 100% coverage')
+    (options, args) = parser.parse_args()
+
+    # Run our meagre tests
+    if options.test:
+        ret_code = run_tests(args, options.processes)
+        return ret_code
+    if options.test_coverage:
+        run_test_coverage(options.build_dir)
+    return 0
+
+if __name__ == '__main__':
+    sys.exit(main())
+sys.exit(1)