Вы используете break
, который выведет вас из любого верхнего уровня for:
или while:
цикла, в котором вы находитесь. Однако в этом коде вы вообще не используете циклы.Я предполагаю, что вы хотите оставить свою функцию и вернуться к своему основному сценарию, но вы уже вернулись до этого сегмента.Вот что я думаю вы ищете:
def correct():
user_input = input("What is your desire")
try:
number = int(user_input)
if number > 2:
return None
else:
return '\n' + number + ' is what you shall receive'
except:
print("You need to input a number")