python - 在Python中使用循环和数组更改条件

Question

我有一个旧的Python脚本，在

if __name__ == '__main__': <br>

我们称之为函数的参数。如果它的参数等于某个字符串，我们应该调用函数。旧代码。

if sys.argv[1] == 'add':
    sys.exit(add(db, usr))
if sys.argv[1] == 'rem':
    sys.exit(rem(db, usr))
if sys.argv[1] == 'rmusr':
    sys.exit(rmusr(db, usr))

一开始我想到了“switch case”，但是我怎么发现，python没有它。我尝试循环和数组。我可以用它来做这个吗？抱歉，我2小时前开始学习Python。但这非常有趣，我希望你能帮助我。

 actions = ['add','rem','rmusr']
    for a in actions:
    if sys.argv[1] == 'a':
        sys.exit(a(db, usr))

正确吗？我发现，我们不应该用“$”来调用变量。请帮忙。

Answer 1

使用字典:

actions = {'add': add, 'rem': rem, 'rmusr': rmusr}

try:
    sys.exit(actions[sys.argv[1]](db, usr))
except KeyError:
   pass
   # or whatever you need to do for an invalid action