SAMUEL

from random import *

def get_validate_userdata():
    name = input('please enter your name')
    surname = input('enter your surname')
    YrGroup = input('please enter the year group you are in')
    Username= str(randint(10,99))+name[0:2]+surname[0:2]
    password = input('enter your password')
    print(name,surname,YrGroup,Username,password)

def register_user():
    print('register user was invoked')
    get_validate_userdata()
    write_user_data()

def Login_user():
    print('Login user invoked')

def menu_selector():
#gets the users input and validates it
    invalid_input = True
    while invalid_input:
        try:
            user_choice= int(input('choose an option:'))
            if user_choice == 1:
                #print('Register was selected')
                register_user()
                invalid_input = False
            elif user_choice == 2:
                #print('Login was selected')
                Login_user()
                invalid_input = False
            elif user_choice == 3:
                #print('Quit was selected')
                invalid_input = False
            else:
                print('you must only enter 1, 2 or 3')
        except ValueError:
            print('Only numbers can be entered')

def main_menu():
    #displays menu screen
    title = '''
    Welcome Comrade you are accepted here
    -------------------------------------'''
    menu='''
    1: Register
    2: Login
    3: Quit'''

    print(title)
    print()
    print(menu)
    menu_selector()