datadog-sccree-time-board-converter.py

#! /bin/bash/python2.7
from datadog import initialize, api
import pdb
import sys

options = {
			'api_key': 'YOUR API KEY',
			'app_key': 'YOUR APP KEY'
		}
initialize(**options)

class converter(object):
	
	graphs = []
	board = []
	board_type = ""
	widgets =[]
	template_variables = []
	title = "Converted Widget"
	@classmethod
	def getdash(cls, dash):
		# Get the dashboard or the screenboard associated with the ID in the arg
		# Set the dashboard type

		try:
			cls.board = api.Timeboard.get(dash)
		except:
			pass
		print cls.board_type

		if 'errors' in cls.board:
			print "Reference ## is not in your timeboards"
			try:
				cls.board = api.Screenboard.get(dash)
				cls.template_variables = cls.board['template_variables']
			except:
				pass
		else: 
			cls.board_type = "timeboard"
			return cls.board_type

		if 'errors' in cls.board:
			print "Reference ## is not in your screenboards"
		else:
			cls.board_type = "screenboard"

	@classmethod
	def widget_transform(cls):
		# Transform the widget list so they are properly formatted 
		# Take off the Screenboard widgets not available in the Timeboards 
		# Returns the list of the different widgets

		if cls.board_type == "screenboard":
			## Get the widgets
			finaltimeboard_widgets=[]
			screenwidgets = cls.board['widgets']
			## Filter the illegal widgets
			forbidden_widget = ['free_text','alert_value','check_status','event_timeline','event_stream','image','note','alert_graph','iframe']
			tmp = [screenwidgets[x]['type'] not in forbidden_widget for x in range(len(screenwidgets))]
			
			## Add the valid widgets
			for x in range(len(tmp)):
				if tmp[x]:
					finaltimeboard_widgets.append(screenwidgets[x])
			return finaltimeboard_widgets

		else:
			print cls.board                    
			return cls.board['dash']['graphs']

		## Convert the widgets
	@classmethod
	def convert_s2t(cls, widgets):
		# Function to convert Screenboard to Timeboard.
		# Takes the widgets as input and output the widgets properly formatted.
		# Appens an additionnal attribute for the hostmap.
		# no output, just tranforms the cls.graphs
		
		for i in range(len(widgets)):
			
			if 'conditional_formats' not in widgets[i]['tile_def']['requests'][0]:
				widgets[i]['tile_def']['requests'][0]['conditional_formats'] = []
			#pdb.set_trace()
			
			if widgets[i]['type'] == 'hostmap':

				cls.graphs.append({
					"definition":{
					"style": widgets[i]['tile_def']['style'],
					"requests":widgets[i]['tile_def']['requests'],
					"viz":widgets[i]['type'],
					},
					"title":  widgets[i]['tile_def']['requests'][0]['q']
				})
			else:			
				cls.graphs.append({
					"definition":{
					"events": [],
					"requests":widgets[i]['tile_def']['requests'],
					"viz":widgets[i]['type'],
					},
					"title": widgets[i]['tile_def']['requests'][0]['q']
				})
		
		## Convert the widgets
	@classmethod
	def convert_t2s(cls, graphs):
		# Function to convert Timeboard to Screenboard.
		# Takes the widgets as input and output the widgets properly formatted.
		# pos_x, pos_y and tmp_y assure the right position of the widgets
		# margin, height and width are hardcoded values representing the default size of the widgets
		# hostmap, heatmap and distribution have specific treatment as their payload is different on a TB and on a SB

		pos_x = 0
		pos_y = 0
		height = 13
		width =  47
		margin = 5
		tmp_y = 0
		for i in range(len(graphs)):

			if i % 2 == 0 and i != 0:
				pos_x = 0
				tmp_y = pos_y

			elif i % 2 == 1 and i != 0:
				tmp_y = pos_y
				pos_y = pos_y + height + margin
				pos_x = width + margin

			if 'conditional_formats' not in graphs[i]['definition']['requests'][0]:
				graphs[i]['definition']['requests'][0]['conditional_formats'] = []

			if graphs[i]['definition']['viz'] not in ['hostmap',"distribution","heatmap"]:
				cls.widgets.append({
					'height': height, 
					'width': width, 
					'timeframe': '4h',
					'x' : pos_x,
					'y' : tmp_y,
					"tile_def":{
					"requests":graphs[i]['definition']['requests'],
					"viz":graphs[i]['definition']['viz'],
					},
					"title": graphs[i]['definition']['requests'][0]['q'],
					"type":graphs[i]['definition']['viz']
				})

			elif graphs[i]['definition']['viz'] == "heatmap" or graphs[i]['definition']['viz'] == "distribution":
				graphs[i]['definition']['requests'][0]['type'] = 'line'
				graphs[i]['definition']['requests'][0]['aggregator'] = 'avg'
				cls.widgets.append({
					'height': height, 
					'width': width, 
					'timeframe': '4h',
					'x' : pos_x,
					'y' : tmp_y,
					"tile_def":{
					"requests":graphs[i]['definition']['requests'],
					"viz":graphs[i]['definition']['viz'],
					},
					"title": graphs[i]['definition']['requests'][0]['q'],
					"type":"timeseries"
				})

			elif graphs[i]['definition']['viz'] == "hostmap":
				cls.widgets.append({
					'height': height, 
					'width': width, 
					'timeframe': '4h',
					'x' : pos_x,
					'y' : tmp_y,
					"tile_def":graphs[i]['definition'],
					"title": graphs[i]['definition']['requests'][0]['q'],
					"type":"hostmap"
				})			

	@classmethod
	def main(cls, dash):
		# Main fuction to fetch the dashboards, extract the widgets, transform the widgets and push the result
		# Takes the dash to convert as an input
		# Outputs the url of the new dash

		cls.getdash(dash)
		if cls.board_type == "screenboard":
			widgets = cls.widget_transform()
			cls.convert_s2t(widgets)
			output = api.Timeboard.create(title=cls.title, description='description', graphs=cls.graphs, template_variables=cls.template_variables, read_only=False)
			print 'http://app.datadoghq.com'+output['url']

		else:
			graphs = cls.widget_transform()
			cls.convert_t2s(graphs)
			output = api.Screenboard.create(board_title=cls.title, description='description', widgets=cls.widgets, template_variables=cls.template_variables)			
			print "http://app.datadoghq.com/screen/" + str(output['id'])

converter().main(sys.argv[1])