python - python 中的 switch case 程序

Question

我正在努力在Python中获得一个switch case程序，当我运行脚本时我想显示三个选项

输入您的选择:

1.插入记录

2.更新记录

3.显示记录

看到这些之后，用户应该能够输入他的选择。由于我是 python 新手，我在 google 上搜索并发现 python 中没有 switch case。

def main():
    print("Enter your choice : ")
    print("1.Insert Records \n2.Update Records \n3.Display Records")
    choice = sys.argv[1]
    if(choice == 1):
        print 1
    if(choice == 2):
        print 2
    if(choice == 3):
        print 3
    else:
        print("You entered a wrong choice")

if __name__ == "__main__":
    main()

这是我尝试过的，但没有用，因为它需要在运行脚本时输入选择(例如python abc.py 1)

Answer 1

您可以使用字典代替 switch/case，如下所示:

from sys import argv

def insertRecord(args):
  ...

def updateRecord(args):
  ...

def displayRecords(args):
  ...

{
  1 : insertRecords,
  2 : updateRecords,
  3 : displayRecords
}[argv[1]](argv[2])

如果你想捕获默认情况，你可以添加如下内容:

from sys import argv

def insertRecord(args):
  ...

def updateRecord(args):
  ...

def displayRecords(args):
  ...

def printHelp():
  ...

try:
  {
    1 : insertRecords,
    2 : updateRecords,
    3 : displayRecords
  }[argv[1]](argv[2])    
except KeyError:
  printHelp()

希望这有帮助。