在python中使用dict作为switch语句

Question

我正在尝试使用字典作为 switch 语句从用户输入中调用各种方法。然而，似乎正在发生的事情是，无论用户选择什么方法，然后按照它们在我的字典中的列出顺序循环遍历所有方法，而不是在特定方法调用后退出。

我是 Python 新手，使用过 JAVA，它有一个 switch 语句，并且能够在 switch 中使用 break 关键字，而 python 没有。

我一直在研究 * 和 google，但对我的特定问题没有任何运气，因此我们将不胜感激。

完整代码如下：

import pymysql

# Open DB
db = pymysql.connect()

# Prepare a cursor object using  cursor() method
cursor = db.cursor()

# Create a Student Table
cursor.execute("DROP TABLE IF EXISTS Student")
cursor.execute("CREATE TABLE Student(Id INT PRIMARY KEY AUTO_INCREMENT, Name VARCHAR(25))")

# Method for inserting new student mySQL
def insert() :
    name = input('Enter the Students name: ')
    instatmt = "INSERT INTO Student VALUES(NULL, '%s')" % (name)
    cursor.execute(instatmt)
    db.commit()

    return(print("Successfully inserted student"))

# Method for updating a student record mySQL
def update() :
    name = input('Enter the Students name: ')
    update = input('Enter the Updated name: ')

    upstatmt = "UPDATE Student SET Name='%s' WHERE Name='%s'" % (update, name)
    cursor.execute(upstatmt)
    db.commit()

    return(print("Successfully updated object"))

# Method for deleting a student record mySQL
def delete() :
    name = input('Enter the Students name: ')

    delstatmt = "DELETE FROM Student WHERE Name ='%s'" % (name)
    cursor.execute(delstatmt)
    db.commit()

    return(print("Successfully deleted object"))

# Method for retrieving a student record mySQL
def retrieve() :
    name = input('Enter the Students name: ')

    retstatmt = "SELECT * FROM Student WHERE Name ='%s'" % (name)
    display = cursor.execute(retstatmt)
    print(display)

    return(print("Object Retrieved..."))

# Call method requested by user    
def performAction(argument):

    switcher = {
        'I': insert(),
        'U': update(),
        'D': delete(),
        'R': retrieve(),
        'E': exit
    }
    func = switcher.get(argument, lambda: "Invalid Entry")
    print (func)

# while True :
action = input('Which Operation would you like to perform ( I : Insert, U : Update, D: Delete, R: Retrieve, E: Exit): ')
performAction(action)

# disconnect from server
db.close()

Answer 1

您正在调用这些函数，而不是将它们插入到您的字典中。试试这个：

def performAction(argument):
    switcher = {
        'I': insert,
        'U': update,
        'D': delete,
        'R': retrieve,
        'E': exit
    }
    func = switcher.get(argument, lambda: "Invalid Entry")
    result = func()
    print (func, result)

Answer 2

您正在调用字典中的函数，因此它们都被连续执行，然后再也不会执行。要解决此问题，请删除括号：

def performAction(argument):

    switcher = {
        'I': insert,
        'U': update,
        'D': delete,
        'R': retrieve,
        'E': exit
    }
    func = switcher.get(argument, lambda: print("Invalid Entry"))
    func()

虽然可行，但这是一种非常规的做事方式 尝试做，并且可以写成更易读的方式：

def performAction(argument):
    if argument == 'I':
        insert()
    elif argument == 'U':
        update()
    elif argument == 'D':
        delete()
    elif argument == 'R':
        retrieve()
    elif argument == 'E':
        exit()
    else:
        print("Invalid Entry")

Python 不是 Java，所以没有必要假装它是。

Answer 3

带参数运行程序

updatedb.py --action insert more

action = input('您要执行哪个操作（I：插入，U：更新，D：删除，R：检索，E：退出）：'）

替换为

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-action", dest="action", nargs=1, default="exit", choices=['insert', 'exit', 'delete'], type=str)
args = parser.parse_args()
>>> args
Namespace(action='insert')
>>> args.action
'insert'

执行动作（动作）

替换为

你可以使用eval()来调用函数

result = eval(args.action)