old_contents: import os
  import yaml
  
  from dogapi import dog_stats_api, dog_http_api
  
  
  def setup():
      """
      Initialize connection to datadog during locust startup.
  
      Reads the datadog api key from (in order):
      1) An environment variable named DATADOG_API_KEY
      2) the DATADOG_API_KEY of a yaml file at
         2a) the environment variable ANSIBLE_VARS
         2b) /edx/app/edx_ansible/server-vars.yaml
      """
  
      api_key = os.environ.get('DATADOG_API_KEY')
  
      if api_key is None:
          server_vars_path = os.environ.get('ANSIBLE_VARS', '/edx/app/edx_ansible/server-vars.yaml')
          server_vars = yaml.safe_load(server_vars_path)
          api_key = server_vars.get('DATADOG_API_KEY')
  
      # By default use the statsd agent
      dog_stats_api.start(
          statsd=True,
          api_key=api_key,
      )
  
new_contents: import os
  import yaml
  
  from dogapi import dog_stats_api, dog_http_api
  
  
  def setup():
      """
      Initialize connection to datadog during locust startup.
  
      Reads the datadog api key from (in order):
      1) An environment variable named DATADOG_API_KEY
      2) the DATADOG_API_KEY of a yaml file at
         2a) the environment variable ANSIBLE_VARS
         2b) /edx/app/edx_ansible/server-vars.yaml
      """
  
      api_key = os.environ.get('DATADOG_API_KEY')
  
      if api_key is None:
          server_vars_path = os.environ.get('ANSIBLE_VARS', '/edx/app/edx_ansible/server-vars.yml')
          with open(server_vars_path, 'r') as server_vars_file:
              server_vars = yaml.safe_load(server_vars_file)
          api_key = server_vars.get('DATADOG_API_KEY')
  
      # By default use the statsd agent
      dog_stats_api.start(
          statsd=True,
          api_key=api_key,
      )
  
subject: Fix reading of API key from yaml file.