Update __init__.py

2025-07-09 22:23:35 +00:00 · 2021-03-05 11:52:47 -05:00
parent a8c92b7f9a
commit a36282d114
1 changed files with 6 additions and 1 deletions
--- a/src/gam/init.py
+++ b/src/gam/init.py
@ -7791,7 +7791,12 @@ def doCreateOrRotateServiceAccountKeys(iam=None,
                new_data['yubikey_pin'] = input('Enter your YubiKey PIN: ')
                i += 1
            elif myarg == 'yubikeyserialnumber':
-                new_data['yubikey_serial_number'] = sys.argv[i+1]
+                try:
+                    new_data['yubikey_serial_number'] = int(sys.argv[i+1])
+                except ValueError:
+                    controlflow.system_error_exit(
+                            3,
+                            'yubikey_serial_number must be a number')
                i += 2
            elif myarg in ['retainnone', 'retainexisting', 'replacecurrent']:
                mode = myarg