from boto.ec2.cloudwatch import CloudWatchConnection
from boto.exception import NoAuthHandlerFound

from . import MetricsBackend


# This limit is defined by Amazon. Don't change it unless they do.
AWS_MAX_BATCH_SIZE = 10


# This relies on boto being configured:
try:
    CLOUDWATCH_CONNECTION = CloudWatchConnection()
except NoAuthHandlerFound:
    raise


class CloudWatchMetricsBackend(MetricsBackend):

    # Send as many items as possible in each CloudWatch API call.
    BUFFER_SIZE = AWS_MAX_BATCH_SIZE

    # Set this per environment to keep metrics separated.
    ENVIRONMENT = None

    def _get_dimensions(self):
        """
        Creates a dimensions dictionary. Uses the hostname if "use_hostname"
        has been called. Uses the ENVIRONMENT variable to allow for
        differentiating between production, staging, dev, etc, because
        CloudWatch is a single pool of metric data.

        """

        if self.ENVIRONMENT:
            if self.hostname:
                return {
                    'Environment': self.ENVIRONMENT,
                    'HostName': self.hostname,
                }
            else:
                return {
                    'Environment': self.ENVIRONMENT,
                }
        else:
            if self.hostname:
                return {
                    'HostName': self.hostname,
                }
            else:
                return {}

    def publish(self, items):
        """Send the buffered metric data to CloudWatch."""

        # Read the items into three separate lists.
        # This is how CloudWatch accepts multiple items.
        names = []
        values = []
        units = []
        for (name, value, unit) in items:
            names.append(name)
            values.append(value)
            units.append(unit)

        CLOUDWATCH_CONNECTION.put_metric_data(
            namespace=self.namespace,
            name=names,
            value=values,
            unit=units,
            dimensions=self._get_dimensions(),
        )

import os

from boto.ec2 import cloudwatch
from boto.exception import NoAuthHandlerFound

from . import MetricsBackend


# This limit is defined by Amazon. Don't change it unless they do.
AWS_MAX_BATCH_SIZE = 10


# This relies on boto being configured:
try:
    CLOUDWATCH_CONNECTION = cloudwatch.connect_to_region(
        region_name=os.environ.get('AWS_DEFAULT_REGION')
    ) or cloudwatch.CloudWatchConnection()
except NoAuthHandlerFound:
    raise


class CloudWatchMetricsBackend(MetricsBackend):

    # Send as many items as possible in each CloudWatch API call.
    BUFFER_SIZE = AWS_MAX_BATCH_SIZE

    # Set this per environment to keep metrics separated.
    ENVIRONMENT = None

    def _get_dimensions(self):
        """
        Creates a dimensions dictionary. Uses the hostname if "use_hostname"
        has been called. Uses the ENVIRONMENT variable to allow for
        differentiating between production, staging, dev, etc, because
        CloudWatch is a single pool of metric data.

        """

        if self.ENVIRONMENT:
            if self.hostname:
                return {
                    'Environment': self.ENVIRONMENT,
                    'HostName': self.hostname,
                }
            else:
                return {
                    'Environment': self.ENVIRONMENT,
                }
        else:
            if self.hostname:
                return {
                    'HostName': self.hostname,
                }
            else:
                return {}

    def publish(self, items):
        """Send the buffered metric data to CloudWatch."""

        # Read the items into three separate lists.
        # This is how CloudWatch accepts multiple items.
        names = []
        values = []
        units = []
        for (name, value, unit) in items:
            names.append(name)
            values.append(value)
            units.append(unit)

        CLOUDWATCH_CONNECTION.put_metric_data(
            namespace=self.namespace,
            name=names,
            value=values,
            unit=units,
            dimensions=self._get_dimensions(),
        )

Use AWS region if set via environment variable