import logging

from awx.main.analytics import gather, ship
from dateutil import parser
from django.core.management.base import BaseCommand
from django.utils.timezone import now


class Command(BaseCommand):
    '''
    Gather AWX analytics data
    '''

    help = 'Gather AWX analytics data'

    def add_arguments(self, parser):
        parser.add_argument('--dry-run', dest='dry-run', action='store_true',
                            help='Gather analytics without shipping. Works even if analytics are disabled in settings.')
        parser.add_argument('--ship', dest='ship', action='store_true',
                            help='Enable to ship metrics to the Red Hat Cloud')
        parser.add_argument('--since', dest='since', action='store',
                            help='Start date for collection')
        parser.add_argument('--until', dest='until', action='store',
                            help='End date for collection')

    def init_logging(self):
        self.logger = logging.getLogger('awx.main.analytics')
        handler = logging.StreamHandler()
        handler.setLevel(logging.DEBUG)
        handler.setFormatter(logging.Formatter('%(message)s'))
        self.logger.addHandler(handler)
        self.logger.propagate = False

    def handle(self, *args, **options):
        self.init_logging()
        opt_ship = options.get('ship')
        opt_dry_run = options.get('dry-run')
        opt_since = options.get('since') or None
        opt_until = options.get('until') or None

        if opt_since:
            since = parser.parse(opt_since)
        else:
            since = None
        if opt_until:
            until = parser.parse(opt_until)
        else:
            until = now()

        if opt_ship and opt_dry_run:
            self.logger.error('Both --ship and --dry-run cannot be processed at the same time.')
            return
        tgzfiles = gather(collection_type='manual' if not opt_dry_run else 'dry-run', since = since, until = until)
        if tgzfiles:
            for tgz in tgzfiles:
                self.logger.info(tgz)
        else:
            self.logger.error('No analytics collected')
        if opt_ship:
            if tgzfiles:
                for tgz in tgzfiles:
                    ship(tgz)