如何在python字典中避免重复的if语句

Question

我正在尝试创建一个由用户提交的html表单数据的新字典。我最后编写了重复的if语句，检查xyz键是否在字典中的data形式。我知道这是一种非常不理想的方法，尽管我不太确定如何使用python来实现它。

这是表单数据字典：

form_data = {
'urls': ['www.google.com', 'www.bing.com'],
'useremail': ['my@email.com'],
'emailfield': ['1'],
'addressfield': ['1'],
'addressfield_info':['Company'],
'addressfield_instruction': ['Please only if the company is a LLC'],
'phonefield': ['1'],
'phonefield_instruction': ['please include area code']
}

我想创建一个字典，看起来像这样：

new_dic = {
'urls': ['www.google.com', 'www.bing.com'],
'useremail': ['my@email.com'],
'infofield': [
              {'field': 'email'},
              {'field': 'address', 'info':'Company', 'instruction': 'Please only if the company is a LLC'},
              {'field':'phone', 'instruction': 'please include area code'}
             ]
}

重要提示：'xyzfield_instruction‘是必填的，’The‘和’the‘都是可选的。此外:用户可以添加更多字段，并创建例如“agefield”、“agefield_info”和“agefield_instruction”。

我的问题是如何有效地检查xyzfield (电子邮件、电话等)是否在字典中。如果它在其中，还要检查是否有任何可选字段也在其中。这看起来像这样：

if 'emailfield' in form_data:
    infofield = {'field': 'email'}
    if 'emailfield_info' in form_data:
         infofield['info'] = form_data['emailfield_info']
    if 'emailfield_instruction' in form_data:
         infofield['instruction'] = form_data['emailfield_instruction']

cleaned_data['infofields'].append(infofield)

...

我对每个领域都这样做，因此我有4-5个这样的。另外，我将无法处理用户自己创建的任何字段，因为我不知道前面的名称。

长话短说:我如何才能让它变得更有效率和动态性？

Answer 1

关于如何避免重复代码的标准答案适用于这里-将重复代码提取到一个函数中：

def extract_field(form_data, clean, fieldname, optional=('info', 'instruction')):
   if fieldname+'field' in form_data:
       infofield = { 'field': fieldname }
       for opt in optional:
           optname = '{}field_{}'.format(fieldname, opt)
           if optname in form_data:
              infofield[opt] = form_data[optname]
       clean.append(infofield)

extract_field(form_data, cleaned_data['infofields'], 'email')
extract_field(form_data, cleaned_data['infofields'], 'address')
extract_field(form_data, cleaned_data['infofields'], 'phone')

Answer 2

这假设您只想清理实际提交的任何内容。如果您正在寻找要存在的特定内容，我建议您列出要查找的内容列表，然后遍历该列表并检查是否存在这些内容。

form_data = {
    'urls': ['www.google.com', 'www.bing.com'],
    'useremail': ['my@email.com'],
    'emailfield': ['1'],
    'addressfield': ['1'],
    'addressfield_info':['Company'],
    'addressfield_instruction': ['Please only if the company is a LLC'],
    'phonefield': ['1'],
    'phonefield_instruction': ['please include area code']
}

def make_field_dict(form_data, base):
        field_dict = {}

        name_field = base + "field"
        name_info = base + "field_info"
        name_inst = base + "field_instruction"
        if name_field not in form_data:
            raise KeyError, "%s not found in form_data" % name_field
        if form_data[name_field] != ['1']:
            raise ValueError, "%s not valid in form_data" % name_field
        field_dict["field"] = base
        if name_info in form_data:
            lst = form_data[name_info]
            if len(lst) != 1:
                raise ValueError, "%s not valid in form_data" % name_info
            field_dict["info"] = lst[0]
        if name_inst in form_data:
            lst = form_data[name_inst]
            if len(lst) != 1:
                raise ValueError, "%s not valid in form_data" % name_inst
            field_dict["instruction"] = lst[0]
        return field_dict

def parse_form_data(form_data):
    cleaned_data = {}
    cleaned_data["infofield"] = []
    seen = set()
    for key, value in form_data.items():
        if "field" not in key:
            cleaned_data[key] = value
        else:
            base, _, tail = key.partition("field")
            if base in seen:
                continue
            cleaned_data["infofield"].append(make_field_dict(form_data, base))
            seen.add(base)
    return cleaned_data


new_dic = {
'urls': ['www.google.com', 'www.bing.com'],
'useremail': ['my@email.com'],
'infofield': [
              {'field': 'email'},
              {'field': 'address', 'info':'Company', 'instruction': 'Please only if the company is a LLC'},
              {'field':'phone', 'instruction': 'please include area code'}
             ]
}

clean_data = parse_form_data(form_data)

new_dic['infofield'].sort()
clean_data['infofield'].sort()
assert(new_dic == clean_data)