Mesa (main): pvr: csbgen: Isolate "main" function with __name__ check

[GitLab Mirror]

Module: Mesa
Branch: main
Commit: 2285ddc7207c9ea52a7649cea7ea58b7f5cda6dc
URL:    http://cgit.freedesktop.org/mesa/mesa/commit/?id=2285ddc7207c9ea52a7649cea7ea58b7f5cda6dc

Author: Matt Coster <matt.coster at imgtec.com>
Date:   Thu May 12 10:27:31 2022 +0100

pvr: csbgen: Isolate "main" function with __name__ check

Signed-off-by: Matt Coster <matt.coster at imgtec.com>
Reviewed-by: Frank Binns <frank.binns at imgtec.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/16884>

---

 src/imagination/csbgen/gen_pack_header.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/src/imagination/csbgen/gen_pack_header.py b/src/imagination/csbgen/gen_pack_header.py
index e7616948fe6..7b1472515a1 100644
--- a/src/imagination/csbgen/gen_pack_header.py
+++ b/src/imagination/csbgen/gen_pack_header.py
@@ -31,7 +31,6 @@ from __future__ import annotations
 
 import copy
 import os
-import sys
 import textwrap
 import typing as t
 import xml.parsers.expat as expat
@@ -966,11 +965,14 @@ class Parser:
         file.close()
 
 
-if len(sys.argv) < 2:
-    print("No input xml file specified")
-    sys.exit(1)
+if __name__ == "__main__":
+    import sys
 
-input_file = sys.argv[1]
+    if len(sys.argv) < 2:
+        print("No input xml file specified")
+        sys.exit(1)
 
-p = Parser()
-p.parse(input_file)
+    input_file = sys.argv[1]
+
+    p = Parser()
+    p.parse(input_file)