Only run warn.py's main function if it is the main program.
Test: run warn.py --byproject build.log
Change-Id: I155cce8c952d6d8a02c8e1809bf61a46c847c26e
diff --git a/tools/warn.py b/tools/warn.py
index dc6aacd..27a2695 100755
--- a/tools/warn.py
+++ b/tools/warn.py
@@ -2333,10 +2333,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()