"""
Appraise
"""
# pylint: disable=C0103,C0111,C0330,E1101
import sys
from json import loads
from zipfile import is_zipfile
from zipfile import ZipFile

from django.core.management.base import BaseCommand
from django.core.management.base import CommandError

from Campaign.models import Campaign


class Command(BaseCommand):
    help = 'Validates campaign data batches'

    def add_arguments(self, parser):
        parser.add_argument(
            'campaign_name',
            type=str,
            help='Name of the campaign you want to process data for',
        )

    def handle(self, *args, **options):
        # Identify Campaign instance for given name.
        try:
            campaign = Campaign.get_campaign_or_raise(options['campaign_name'])
        except LookupError as error:
            raise CommandError(error)

        _validate_campaign_data(campaign, self.stdout)


def _validate_campaign_data(campaign, stdout=None):
    """Validates campaign data batches."""
    validated_batches = 0
    batches_to_check = campaign.batches.filter(dataValid=False, dataReady=False)
    for batch in batches_to_check:
        batch_name = batch.dataFile.name
        batch_file = batch.dataFile

        stdout.write('Batch name: {}'.format(batch_name))
        try:
            # TODO: move validation code to CampaignData class.
            if batch_name.endswith('.zip'):
                if not is_zipfile(batch_file):
                    stdout.write('Batch {0} not a valid ZIP archive'.format(batch_name))
                    continue

                batch_zip = ZipFile(batch_file)
                batch_json_files = [
                    x for x in batch_zip.namelist() if x.endswith('.json')
                ]
                for batch_json_file in batch_json_files:
                    batch_data = batch_zip.read(batch_json_file).decode('utf-8')
                    # Python 3.9 removed 'encoding' from json.loads
                    if sys.version_info >= (3, 9, 0):
                        loads(batch_data)
                    else:
                        loads(batch_data, encoding='utf-8')

            else:
                loads(str(batch_file.read(), encoding='utf-8'))

            batch.dataValid = True
            batch.save()

            validated_batches += 1

        except Exception as error:
            raise CommandError(error)

    stdout.write('Validated {0} batches'.format(validated_batches))