update form and register util

Author: dslsyeoh

exercises/txt_based_user_register_and_login.py

@@ -1,13 +1,13 @@
 # Author Steven Yeoh
 # Copyright (c) 2019. All rights reserved.
 
-from exercises.utils.form_util import get_form
+from exercises.utils.form_util import display_form
 
 print("\n=========================== MENU ===========================")
 print(" [R]egister account")
 print(" [L]ogin")
-user_option = input(" Enter your option: ").upper()
+user_option = input(" Enter option: ").upper()
 print("=========================== MENU ===========================")
-get_form(user_option)
+display_form(user_option)
 
 

exercises/utils/form_util.py

@@ -1,24 +1,27 @@
 # Author Steven Yeoh
 # Copyright (c) 2019. All rights reserved.
 
-from exercises.utils import login_util, register_util
+from getpass import getpass
+from exercises.utils import register_util, validate_util
 
 
-def get_form(option):
- if option == "R":
+def display_form(user_option):
+ if user_option == "R":
  register_form()
- elif option == "M":
+ elif user_option == "M":
  login_form()
  else:
- print("Invalid option, please try again.")
+ print("Incorrect option, please try again.")
 
 
 def register_form():
  print("\n=========================== REGISTER ===========================")
  username = input(" Enter username: ")
- password = input(" Enter password: ")
- confirm_password = input(" Enter confirm password: ")
- print("=========================== REGISTER ===========================")
+ password = getpass(" Password: ")
+ confirm_password = getpass(" Enter confirm password: ")
+ print("=========================== REGISTER ===========================\n")
+ errors = validate_util.validate_user_input(username, password, confirm_password)
+ validate_util.print_errors(errors) if bool(errors) else register_util.add_new_user(username, password)
 
 
 def login_form():

exercises/utils/register_util.py

@@ -1,3 +1,8 @@
 # Author Steven Yeoh
 # Copyright (c) 2019. All rights reserved.
 
+
+def add_new_user(username, password):
+ # store into txt file
+ print("Successfully created user, {}.".format(username))
+