Properly handle exception for non-integer input with pass_gen()

Former-commit-id: b2c4bd2cd8cf7db4516f47a48bde46610d2e9f5c [formerly 964561b33b] Former-commit-id: c92d23fa9ee173e5e0540ecbb060b1edf8ec05c9
2026-08-28 04:46:34 -04:00 · 2022-06-20 03:35:19 -04:00
parent 2111a0454e
commit 419ff51e75
1 changed files with 6 additions and 1 deletions
@@ -130,7 +130,12 @@ def pass_gen():
        if __special < __min_special:
            __gen = _pass_gen_function(__complexity, __length)
        return __gen
-    _length = int(input('password length: '))
+    try:
+        _length = int(input('password length: '))
+    except ValueError:
+        print(f"\n\u001b[38;5;9merror: a non-integer value was input for password length\u001b[0m\n")
+        _gen = pass_gen()
+        return _gen
    if _length > 840:
        _length = 840
        print('\n\u001b[38;5;9mpassword length has been limited to the maximum of 840 characters\u001b[0m\n')