error handling for --print-platform

rvykydal · Apr 19, 2024 · 508ea08 · 508ea08
1 parent 647da2f
commit 508ea08
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/scripts/generate-permian-query.py b/scripts/generate-permian-query.py
@@ -13,7 +13,7 @@ def parse_args():
                          nargs=1, action="append", help="skip tests with TYPE (tag)")
     _parser.add_argument("--testtype", "-t", type=str, metavar="TYPE",
                          help="only run tests with TYPE (tag)")
-    _parser.add_argument("--platform", "-p", type=str, metavar="PLATFORM",
+    _parser.add_argument("--platform", "-p", type=str, metavar="PLATFORM", default="",
                          help="platform to be used for tests")
     _parser.add_argument("--print-platform", action="store_true",
                          help="print the platform option")
@@ -27,8 +27,11 @@ def parse_args():
     args = parse_args()
 
     if args.print_platform:
-        print(args.platform)
-        exit()
+        if args.platform:
+            print(args.platform)
+            exit()
+        else:
+            exit(1)
 
     conditions = []
     if args.tests: