Question

So, I’ve got this going on:

def pickClass():
    print('What class are you? FIGHTER, MAGE, or THIEF?')
    classChoice = raw_input()
    if classChoice == 'FIGHTER':
        print('You are a mighty warrior. Are you sure? YES or NO.')
        _confirm = raw_input()
        if _confirm == 'YES':
            print('So be it.')
        elif _confirm == 'NO':
            pickClass()
        else:
            print('YES or NO only!')
            #go back to '_confirm = raw_input()'

The part where I’m stuck is at the very end — how do I go part to that specific part of the code without going through the entire function again?

(I know it’s a little redundant with that print, but whatever, maaaaan)

Answer 1

You will need to restructure your function. Try something like this:

def pickClass():
    valid_classes = ["FIGHTER", "MAGE", "THIEF"]
    while True:
        print('What class are you? FIGHTER, MAGE, or THIEF?')
        classChoice = raw_input()
        if classChoice not in valid_classes:
            print("Invalid class")
        else:
            print("Are you sure you want to be a %s?" % classChoice)
            while True:
                _confirm = raw_input()
                if _confirm == 'YES':
                    print('So be it.')
                    return classChoice
                elif _confirm == 'NO':
                    break
                else:
                    print('YES or NO only!')