Python识别重复的字典键

Question

在 python 中，我正在寻找一种在从包含重复键的文件中加载静态声明的字典时接收警告或错误的方法，对于我的用例，该文件来自用户输入，所以我想确保字典我接收没有重复的密钥。我知道在加载字典 1 与字典 2 相同并且 python 字典保留最右边的键/值对之后。我正在寻找的是一种在加载之前或加载期间获取警告或错误的方法，这表明 dictionary1 有多个重复的“a”键。

dictionary1 = {"a":1, "a":2, "a":3}
dictionary2 = {"a":3}

我能想到的最好办法是使用字典列表，然后将每个字典添加到最终字典中，如下例所示。这可行，但字典列表不像标准字典那样用户友好。

listofDicts = [{"a":1},{"a":2},{"a":3}]
masterDict = {}
for entry in listofDict:
    for subDict in entry:
        if subDict in masterDict.keys():
            print ("ERROR key \"%s\" already exists with value %d" % (subDict, masterDict[subDict]))
        else:
            masterDict.update({subDict:entry[subDict]})

Answer 1

您可以使用 ast 模块解析包含字典的文件中的 Python 源代码，并查找具有重复键的字典文字：

import ast
import logging

class DuplicateKeyVisitor(ast.NodeVisitor):
    def visit_Dict(self, node):
        seen_keys = set()

        for key_node in node.keys:
            try:
                key = ast.literal_eval(key_node)
            except ValueError:
                continue

            if key in seen_keys:
                logging.warning('Dictionary literal at (%d, %d) has duplicate keys', node.lineno, node.col_offset)

            seen_keys.add(key)

DuplicateKeyVisitor().visit(ast.parse('''
foo = {'a': 1, 'a': 2}
bar = {'a': 1, 'b': 2}
bar = {'a': 1, 'b': 2, 'a': 3}
'''))