python - 如何避免使用过多的 if else ？

Question

这就是我试图通过代码实现的概念。我希望用户在按下“0”时能够离开登录部分，下面的代码可以工作，但是无论如何我可以缩短这些代码，因为我觉得如果使用其他代码的话，就太多了。

def signup():
     print("Press '0' to return mainpage")
     firstname = input("Please enter your first name : ")
     if firstname == "0":
          print("Mainpage returned")
          mainpage()
     else:
          lastname  = input("Please enter your last name  : ")
          if lastname == "0":
               print("Mainpage returned")
               mainpage()
          else:
               username = firstname+lastname
               print("Welcome", username)

def mainpage():
     option = input("Press '1' to signup: ")
     if option == '1':
          print("Sign up page accessed")
          signup()
mainpage()

Answer 1

您可以使用提前返回来删除一些嵌套。

def signup():
     print("Press '0' to return mainpage")
     firstname = input("Please enter your first name : ")
     if firstname == "0":
          print("Mainpage returned")
          mainpage()
          return
 
     lastname  = input("Please enter your last name  : ")
     if lastname == "0":
         print("Mainpage returned")
         mainpage()
         return

     username = firstname+lastname
     print("Welcome", username)