Source code for galaxy.tools.error_reports.plugins.influxdb

"""The module describes the ``influxdb`` error plugin plugin."""
from __future__ import absolute_import

import datetime
import logging

try:
    import influxdb
except ImportError:
    # This middleware will never be used without influxdb.
    influxdb = None

from galaxy.util import unicodify
from . import ErrorPlugin

log = logging.getLogger(__name__)


[docs]class InfluxDBPlugin(ErrorPlugin):
    """Send error report to InfluxDB
    """
    plugin_type = "influxdb"

[docs]    def __init__(self, **kwargs):
        if not influxdb:
            raise ImportError("Could not find InfluxDB Client, this error reporter will be disabled; please pip install influxdb")

        self.app = kwargs['app']
        # Neither of these matter
        self.verbose = False
        self.user_submission = False
        # Anything with influxdb_ gets sent to the client initialization
        influx_args = {
            k[len('influxdb_'):]: v
            for (k, v)
            in kwargs.items()
            if k.startswith('influxdb_')
        }
        print(influx_args)
        self.client = influxdb.InfluxDBClient(**influx_args)

[docs]    def submit_report(self, dataset, job, tool, **kwargs):
        """Submit the error report to sentry
        """
        self.client.write_points([{
            'measurement': 'galaxy_tool_error',
            'time': datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%SZ'),
            'fields': {
                'value': 1
            },
            'tags': {
                'exit_code': job.exit_code,
                'tool_id': unicodify(job.tool_id),
                'tool_version': unicodify(job.tool_version),
                'tool_xml': unicodify(tool.config_file) if tool else None,
                'destination_id': unicodify(job.destination_id),
                'handler': unicodify(job.handler),
            }
        }])
        return ('Submitted to InfluxDB', 'success')


__all__ = ('InfluxDBPlugin', )