commit 312c310f7c31d849aa05ce4d41229c4665a15d6a
author Treehugger Robot <treehugger-gerrit@google.com> Thu Sep 29 03:58:07 2016 +0000
committer Gerrit Code Review <noreply-gerritcodereview@google.com> Thu Sep 29 03:58:07 2016 +0000
tree 6198b0694259aaa0f3b2d7e325dc2e9585f2b4fb
parent 7ec2a39c3f723b871e0beffbdd5b8485cf0eb0ae
parent ac9d5df349621c423b9fd9e70397bb23e1d2680b

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

tools/warn.py

diff --git a/tools/warn.py b/tools/warn.py
index 2e7927a..4ed4952 100755
--- a/tools/warn.py
+++ b/tools/warn.py
@@ -2342,10 +2342,14 @@
   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()