Merge "Only run warn.py's main function if it is the main program."

2016-09-29 03:58:07 +00:00
parent 7ec2a39c3f ac9d5df349
commit 312c310f7c
1 changed files with 10 additions and 6 deletions
--- a/tools/warn.py
+++ b/tools/warn.py
@@ -2342,10 +2342,14 @@ def dump_csv():
  print '{},,{}'.format(total, 'All warnings')


-##### Main function starts here. #########################
+def main():
+  parse_input_file()
+  if args.gencsv:
+    dump_csv()
+  else:
+    dump_html()

-parse_input_file()
-if args.gencsv:
-  dump_csv()
-else:
-  dump_html()
+
+# Run main function if warn.py is the main program.
+if __name__ == '__main__':
+  main()