add exception handling for invalid password length to default length and validate security level

Author: dslsyeoh

exercises/password_generator.py

@@ -2,19 +2,25 @@
 # Copyright (c) 2019. All rights reserved.
 
 import time
-from exercises.utils.password_generator_util import generate_password
+from exercises.utils.password_generator_util import generate_password, security_level_is_valid
 
 
 print("\n======================= PASSWORD GENERATOR =======================")
-password_length = int(input("Enter your password length: "))
-print("Password security level: [E]asy, [M]edium, [H]ard")
-security_level = input("Enter your password security level: ").upper()
-if security_level is None:
- print("Password security level incorrect!, Please try again.")
-else:
- print("Generating password...")
- result = generate_password(password_length, security_level)
- time.sleep(2)
- print("Generated password: {}".format(result))
+password_length = 0
+try:
+ password_length = int(input("Enter your password length: "))
+except ValueError:
+ password_length = 15
+finally:
+ print("Password security level: [E]asy, [M]edium, [H]ard")
+ security_level = input("Enter your password security level: ").upper()
+ if security_level_is_valid(security_level):
+ print("Generating password...")
+ # stimulate generate
+ time.sleep(1)
+ result = generate_password(password_length, security_level)
+ print("Generated password: {}".format(result))
+ else:
+ print("Password security level incorrect, Please try again.")
 print("======================= PASSWORD GENERATOR =======================")
 

exercises/utils/password_generator_util.py

@@ -9,6 +9,10 @@
 REPETITIVE_LIMIT = 3
 
 
+def security_level_is_valid(security_level):
+ return True if security_level_dict.get(security_level) else False
+
+
 def generate_password(password_length, security_level):
  generated_words = []
  old_group_index = None