def table_not_found()

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

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

  def table_not_found(self, table_entities):
    """Make all user input entities to None

    Args:
        table_entities (dict): user input from config

    Returns:
        list: with all entities provided by user to None
    """
    out = []
    if table_entities["isheader"]:
      inp_header = table_entities["headers"]

    for user_inp in table_entities["entity_extraction"]:
      try:
        entity_data = {}
        suffix, _ = user_inp["entity_suffix"], user_inp["row_no"]
        col = user_inp["col"]
        if suffix in (None, ""):
          suffix = ""
        entity_name = f"{inp_header[col]} {suffix}"
        entity_data = {
                        "value": None,
                        "extraction_confidence": None,
                        "value_coordinates": None,
                        "manual_extraction": False,
                        "corrected_value": None
                      }
        entity_data["entity"] = entity_name
        entity_data["key_coordinates"] = None
        entity_data["page_height"] = self.master_dict[0]["height"]
        entity_data["page_width"] = self.master_dict[0]["width"]
        entity_data["page_no"] = None

        out.append(deepcopy(entity_data))
      except Exception as e:
        Logger.error(e)
        continue
    return out