def main()

in utils/audio_upload.py [0:0]

• 42 lines of code
• 4 McCabe index (conditional complexity)

def main():
  pargs = _ParseArgs()
  project_id = pargs.project_id
  impersonated_service_account = pargs.impersonated_service_account
  insights_endpoint = pargs.insights_endpoint
  api_version = pargs.insights_api_version
  agent_id = pargs.agent_id
  agent_channel = int(pargs.agent_channel)
  xml_bucket = pargs.xml_gcs_bucket
  # analyze_conv = pargs.analyze
  inspect_template = pargs.inspect_template
  deidentify_template = pargs.deidentify_template
  audio_format = pargs.audio_format

  source_audio_gcs_bucket = pargs.source_audio_gcs_bucket

  # pargs.source_local_audio_path or
  if source_audio_gcs_bucket:
    # Inputs are audio files.
    audio_uris, metadata = _GetGcsUris(
        source_audio_gcs_bucket, project_id, impersonated_service_account
    )

    # get processed audio files
    processed_audio_files = _GetProcessedAudios(project_id)

    # remove processed files
    audio_uris, metadata = _RemoveProcessedFiles(
        audio_uris, processed_audio_files, metadata
    )

    if not audio_uris:
      print('No audio file to process')
      # conversation_names = []
    else:
      operation_names = _UploadBulkAudio(
          audio_uris,
          metadata,
          project_id,
          insights_endpoint,
          api_version,
          agent_id,
          impersonated_service_account,
          agent_channel,
          xml_bucket,
          inspect_template,
          deidentify_template,
          audio_format,
      )
      print('Uploaded `{}` conversations'.format(
          len(operation_names)))

  else:
    print('Please provide audio files GCS bucket')