Support for dd_url through env vars and hostvars

This allow customer to use the ansible callback with the Datadog EU
intake.

Author: hush-hush

CHANGELOG.md

@@ -1,6 +1,9 @@
 CHANGELOG
 =========
 
+# 2.5.0 / 2019-02-07
+* [FEATURE] Add support for "site" configuration.
+
 # 2.4.2 / 2018-06-08
 - [BUGFIX] Fix yaml import broken by Ansible 2.5.
 

README.md

@@ -28,6 +28,12 @@ if it doesn't exist.
 
 Set the environment variable `DATADOG_API_KEY`.
 
+Optionally to send data to Datadog EU, you can set the environment
+variable `DATADOG_SITE=datadoghq.eu`.
+
+To send data to a custom URL you can set the environment
+variable `DATADOG_URL=<custom URL>`.
+
 ##### Using a yaml file
 
 Create a `datadog_callback.yml` file alongside `datadog_callback.py`,
@@ -36,6 +42,11 @@ as following:
 
 ```
 api_key: <your-api-key>
+
+# optionally to send data to Datadog EU add the following setting
+site: datadoghq.eu
+# optionally to send data to a custom URL add the following setting
+url: <custom URl>
 ```
 
 You can specify a custom location for the configuration file using the
@@ -51,6 +62,12 @@ ANSIBLE_DATADOG_CALLBACK_CONF_FILE=/etc/datadog/callback_conf.yaml ansible-playb
 Alternatively you can use the hostvars of the host ansible is being run from (preferably in the vault file):
 ```
 datadog_api_key: <your-api-key>
+
+# Optionally to send data to Datadog EU add the following setting
+datadog_site: datadoghq.eu
+
+# Optionally to send data to a custom URL add the following setting
+datadog_url: <custom URL>
 ```
 
 3. Be sure to whitelist the plugin in your ansible.cfg

datadog_callback.py

@@ -16,6 +16,7 @@
 from ansible.plugins.callback import CallbackBase
 from __main__ import cli
 
+DEFAULT_DD_URL = "https://api.datadoghq.com"
 
 class CallbackModule(CallbackBase):
     def __init__(self):
@@ -57,7 +58,10 @@ def _load_conf(self, file_path):
             with open(file_path, 'r') as conf_file:
                 conf_dict = yaml.load(conf_file)
 
-        return os.environ.get('DATADOG_API_KEY', conf_dict.get('api_key', '')), conf_dict.get('url', 'https://app.datadoghq.com')
+        api_key = os.environ.get('DATADOG_API_KEY', conf_dict.get('api_key', ''))
+        dd_url = os.environ.get('DATADOG_URL', conf_dict.get('url', ''))
+        dd_site = os.environ.get('DATADOG_SITE', conf_dict.get('site', ''))
+        return api_key, dd_url, dd_site
 
     # Send event to Datadog
     def _send_event(self, title, alert_type=None, text=None, tags=None, host=None, event_type=None, event_object=None):
@@ -232,7 +236,7 @@ def v2_playbook_on_play_start(self, play):
 
         # Read config and hostvars
         config_path = os.environ.get('ANSIBLE_DATADOG_CALLBACK_CONF_FILE', os.path.join(os.path.dirname(__file__), "datadog_callback.yml"))
-        api_key, url = self._load_conf(config_path)
+        api_key, dd_url, dd_site = self._load_conf(config_path)
 
         # If there is no api key defined in config file, try to get it from hostvars
         if api_key == '':
@@ -244,13 +248,23 @@ def v2_playbook_on_play_start(self, play):
             else:
                 try:
                     api_key = hostvars['localhost']['datadog_api_key']
+                    if not dd_url:
+                        dd_url = hostvars['localhost'].get('datadog_url')
+                    if not dd_site:
+                        dd_site = hostvars['localhost'].get('datadog_site')
                 except Exception as e:
                     print('No "api_key" found in the config file ({0}) and "datadog_api_key" is not set in the hostvars: disabling Datadog callback plugin'.format(config_path))
                     self.disabled = True
 
+        if not dd_url:
+            if dd_site:
+                dd_url = "https://api."+ dd_site
+            else:
+                dd_url = DEFAULT_DD_URL # default to Datadog US
+
         # Set up API client and send a start event
         if not self.disabled:
-            datadog.initialize(api_key=api_key, api_host=url)
+            datadog.initialize(api_key=api_key, api_host=dd_url)
 
             self.send_playbook_event(
                 'Ansible play "{0}" started in playbook "{1}" by "{2}" against "{3}"'.format(