Python 将字符串输入匹配到现有字典的简单方法

Question

我正在编写一个可以在多个表单上运行的测试。我创建了一个批处理文件，它将发送一个包含表单名称的参数（因此从技术上讲，每个批处理文件只运行一次测试）。对于每个要与参数输入匹配的测试，我都有一个字典。所以假设我有：

import sys

form1 = {
  a = 3,
  b = 4
}
form2 = {
  a = 7,
  b = 8
}

args = sys.argv[1] #this would be the form input as a string; example: "form1"

我想要一种方法让它知道“form1”与字典 form1 匹配。显然，我可以这样做：

if args == "form1":
  form = form1
elif args == "form2":
  form = form2
...

但是，由于有 27 个测试，我宁愿不必处理这些问题。有没有更好的方法来做到这一点？大致如下：

form = matchvar(args)
runtest(form)

谢谢。

Answer 1

我敢推荐一本字典吗？

formdict = { "form1": form1, "form2": form2 }

然后您可以使用formdict.get(args,[]) 访问正确的字典。

>>> formdict["form1"]["a"]
3
>>> formdict["form2"]["a"]
7

Answer 2

使用包含的表单创建字典，然后使用从命令行获得的值查找键：

forms = dict()
forms['form1'] = {a:3, b:4}
forms['form2'] = {a:7, b:8}

form = forms.get(args)
if form:
   runtest(form)
else:
   print('Invalid form: {}'.format(args))