def filter_documents()

in microservices/upload_service/src/utils/process_task_helpers.py [0:0]

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

def filter_documents(configs: List[Dict]):
  """Filter the supporting documents and application form"""
  print("filter_documents")
  print(configs)

  supporting_docs = []
  application_form = []
  for config in configs:
    case_id = config.get("case_id")
    uid = config.get("uid")
    gcs_url = config.get("gcs_url")
    cl_result = get_classification(case_id, uid, gcs_url)
    if cl_result.status_code == 200:
      document_type = cl_result.json().get("doc_type")
      document_class = cl_result.json().get("doc_class")
      Logger.info(
          f"Classification successful for {uid}:document_type:{document_type},\
      document_class:{document_class}.")

      if document_type == "application_form":
        config["document_class"] = document_class
        application_form.append(config)
      elif document_type == "supporting_documents":
        config["document_class"] = document_class
        supporting_docs.append(config)
    else:
      Logger.error(f"Classification FAILED for {uid}")
  print(
      f"Application form:{application_form} and"\
      f" supporting_docs:{supporting_docs}")
  Logger.info(
      f"Application form:{application_form} and "\
        f"supporting_docs:{supporting_docs}")
  return application_form, supporting_docs