Skip to content

Check for datadog and yaml libraries #6

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 3 commits into from
Closed

Check for datadog and yaml libraries #6

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
974b9ad
Checks existence of datadog_callback.yml before starting
Oct 26, 2015
59f5d7a
More descriptive error message
Oct 26, 2015
f38e46c
Check if datadog library exists before starting
Oct 26, 2015
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 18 additions & 5 deletions datadog_callback.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,15 +1,24 @@
import os.path
import time

import datadog
import yaml
try:
import datadog
import yaml
HAS_MODULES = True
except ImportError:
HAS_MODULES = False

config_file = "datadog_callback.yml"

class CallbackModule(object):
def __init__(self):
# Read config and set up API client
api_key, url = self._load_conf(os.path.join(os.path.dirname(__file__), "datadog_callback.yml"))
datadog.initialize(api_key=api_key, api_host=url)
if not HAS_MODULES:
self.disabled = True
print 'Datadog callback disabled.\nMake sure you call all required libraries.'

if not os.path.isfile(os.path.join(os.path.dirname(__file__),config_file)):
self.disabled = True
print 'Datadog callback disabled.\nMake sure you have "{0}" configuration file.'.format(config_file)

self._playbook_name = None
self._start_time = time.time()
Expand Down Expand Up @@ -144,6 +153,10 @@ def runner_on_unreachable(self, host, res):
)

def playbook_on_start(self):
# Read config and set up API client
api_key, url = self._load_conf(os.path.join(os.path.dirname(__file__), config_file))
datadog.initialize(api_key=api_key, api_host=url)

# Retrieve the playbook name from its filename
self._playbook_name, _ = os.path.splitext(
os.path.basename(self.playbook.filename))
Expand Down