def entities_extraction()

in microservices/extraction_service/src/utils/utils_functions.py [0:0]

• 13 lines of code
• 5 McCabe index (conditional complexity)

def entities_extraction(parser_data, required_entities, doc_type):
  """
    This function reads information of default and derived entities
    Parameters
    ----------
    parser_data: specialzed parser result
    required_entities: required extracted entities
    doc_type: Document type
    Returns: Required entities dict
    -------
  """

  # Read the entities from the processor
  parser_entities = parser_data["entities"]
  default_entities = required_entities["default_entities"]
  derived_entities = required_entities.get("derived_entities")
  # Extract default entities
  entity_dict = default_entities_extraction(parser_entities,
                                            default_entities,doc_type)
  Logger.info("Default entities created from Specialized parser response")
  # if any derived entities then extract them
  if derived_entities:
    # this function can be used for all docs, if derived entities
    # are extracted by using regex pattern
    derived_entities_extracted_dict = derived_entities_extraction\
        (parser_data, derived_entities)
    entity_dict.update(derived_entities_extracted_dict)
    Logger.info("Derived entities created from Specialized parser response")
  return entity_dict