From 041d9ac3dc924b4dc55a53735645ae1de450a513 Mon Sep 17 00:00:00 2001
From: Dylan Baker <dylan.c.baker@intel.com>
Date: Tue, 1 Nov 2022 12:47:21 -0700
Subject: [PATCH] util/glsl2spirv: let argparse actually enforce the
 restrictions we've set

Reviewed-by: Luis Felipe Strano Moraes <luis.strano@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/19449>
---
 src/util/glsl2spirv.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/util/glsl2spirv.py b/src/util/glsl2spirv.py
index 78468a2737e..9e6241eacb6 100644
--- a/src/util/glsl2spirv.py
+++ b/src/util/glsl2spirv.py
@@ -40,7 +40,8 @@ def get_args():
 
     parser.add_argument('--glsl-version',
                         dest="glsl_ver",
-                        help="{100 | 110 | 120 | 130 | 140 | 150 | 300es | 310es | 320es | 330 | 400 | 410 | 420 | 430 | 440 | 450 | 460} set GLSL version, overrides #version in shader sources. Default is 460.")
+                        choices=['100', '110', '120', '130', '140', '150', '300es', '310es', '330', '400', '410', '420', '430', '440', '450', '460'],
+                        help="Override GLSL #version declaration in source.")
 
     parser.add_argument("-Olib",
                         action='store_true',
@@ -56,7 +57,8 @@ def get_args():
 
     parser.add_argument("--stage",
                         default="vert",
-                        help="Uses specified stage rather than parsing the file extension choices for <stage> are vert, tesc, tese, geom, frag, or comp.")
+                        choices=['vert', 'tesc', 'tese', 'geom', 'frag', 'comp'],
+                        help="Uses specified stage rather than parsing the file extension")
     args = parser.parse_args()
     return args