From ac9d5df349621c423b9fd9e70397bb23e1d2680b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Meike=20Baumg=C3=A4rtner?= <meikeb@google.com>
Date: Wed, 28 Sep 2016 10:48:06 -0700
Subject: [PATCH] Only run warn.py's main function if it is the main program.

Test: run warn.py --byproject build.log
Change-Id: I155cce8c952d6d8a02c8e1809bf61a46c847c26e
---
 tools/warn.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/tools/warn.py b/tools/warn.py
index dc6aacd1a..27a269532 100755
--- a/tools/warn.py
+++ b/tools/warn.py
@@ -2333,10 +2333,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()