"""Contract Exclusion logic."""

from owsresponse import response

from abacus_contract import models
from abacus_contract.constants.constants import COUNTRIES_POST_LIMIT
from abacus_contract.constants.error import ERROR_CONTRACT_EXCLUSION_ALREADY_EXISTS
from abacus_contract.constants.error import ERROR_CONTRACT_EXCLUSION_DOES_NOT_EXIST
from abacus_contract.constants.error import ERROR_COUNTRIES_POST_LIMIT
from abacus_contract.constants.error import ERROR_OA_CONTRACT_EXISTS
from abacus_contract.schemas.contract_exclusion import ContractExclusionSchema
from abacus_contract.utils.validations import validate_country_code


def create_contract_exclusions(contract_id, exclusions):
    """Create contract exclusions.

    @param contract_id: Contract id
    @param exclusions: Dict containing excluded countries and/or stores.
    For example: {"countries": ["USA", "AFG"], "stores": ["123"]}
    """
    contract = models.Contract.get_by_id_or_error(contract_id)

    if contract.contract_exclusion:
        return response.create_error_response(
            'error',
            ERROR_CONTRACT_EXCLUSION_ALREADY_EXISTS.format(contract_id=contract_id),
            status=400
        )

    countries = exclusions.get('countries', [])

    if len(countries) > COUNTRIES_POST_LIMIT:
        return response.create_error_response(
            'error',
            ERROR_COUNTRIES_POST_LIMIT.format(limit=COUNTRIES_POST_LIMIT),
            status=400
        )

    if contract.legacy_contract and contract.legacy_contract.oa_contract_id:
        return response.create_error_response(
            'error',
            ERROR_OA_CONTRACT_EXISTS,
            status=400
        )

    try:
        for country_code in countries:
            validate_country_code(country_code)

        exclusion_params = {
            'contract_id': contract_id,
            'exclusions': exclusions
        }
        exclusion = models.ContractExclusion.create(**exclusion_params)
        message = ContractExclusionSchema().dump(exclusion)
        return response.Response(message=message, status=201)
    except Exception as e:
        return response.create_error_response('error', str(e), status=400)


def get_exclusions_by_contract(contract_id):
    """GET exclusions object for a specified contract."""
    result = models.Contract.get_by_id_or_error(contract_id)
    if not result.contract_exclusion:
        return response.Response(message={}, status=200)
    message = ContractExclusionSchema().dump(result.contract_exclusion)
    return response.Response(message=message, status=200)


def update_contract_exclusions(contract_id, exclusions):
    """Update contract exclusions.

    @param contract_id: Contract id
    @param exclusions: Dict containing excluded countries and/or stores.
    For example: {"countries": ["USA", "AFG"], "stores": ["123"]}
    """
    contract = models.Contract.get_by_id_or_error(contract_id)

    if not contract.contract_exclusion:
        return response.create_error_response(
            'error',
            ERROR_CONTRACT_EXCLUSION_DOES_NOT_EXIST.format(contract_id=contract_id),
            status=404
        )

    countries = exclusions.get('countries', [])

    try:
        for country_code in countries:
            validate_country_code(country_code)

        exclusion_params = {
            'exclusions': exclusions
        }
        exclusion = contract.contract_exclusion.update_attributes(**exclusion_params)
        models.ContractExclusion.commit_changes()
        message = ContractExclusionSchema().dump(exclusion)
        return response.Response(message=message, status=200)
    except Exception as e:
        return response.create_error_response('error', str(e), status=400)