def get_input_output_solr_url()

in ambari-infra-solr-client/src/main/python/migrationHelper.py [0:0]

• 15 lines of code
• 6 McCabe index (conditional complexity)

def get_input_output_solr_url(src_solr_urls, target_solr_urls):
  """
  Choose random solr urls for the source and target collections, prefer localhost and common urls
  """
  def intersect(a, b):
    return list(set(a) & set(b))
  input_solr_urls = src_solr_urls
  output_solr_urls = target_solr_urls
  hostname = socket.getfqdn()
  if any(hostname in s for s in input_solr_urls):
    input_solr_urls = filter(lambda x: hostname in x, input_solr_urls)
  if any(hostname in s for s in output_solr_urls):
    output_solr_urls = filter(lambda x: hostname in x, output_solr_urls)
  common_url_list = intersect(input_solr_urls, output_solr_urls)
  if common_url_list:
    input_solr_urls = common_url_list
    output_solr_urls = common_url_list

  return get_random_solr_url(input_solr_urls), get_random_solr_url(output_solr_urls)