def get_entity_key_value_pairs()

in document_ai_warehouse/common/src/common/utils/document_ai_utils.py [0:0]

• 19 lines of code
• 7 McCabe index (conditional complexity)

    def get_entity_key_value_pairs(docai_document):
        fields = {}
        if hasattr(docai_document, "entities"):
            entities = {}
            for entity in docai_document.entities:
                key = entity.type_
                value = entity.mention_text
                if key not in entities:
                    entities[key] = []
                entities[key].append(value)

            for key in entities:
                values = entities[key]
                N = len(values)

                for i in range(N):
                    if i == 0:
                        fields[key] = values[i]
                    else:
                        fields[key + "_" + str(i + 1)] = values[i]

        return fields