"""Product store logic."""
from oto import response

from availability.models import product as product_model
from availability.models import product_in_store


def import_store_links(data):
    """Import product store links.

    Args:
        data (list): List of product links.
    Returns:
        Response
    """
    msg = {}

    for item in data:
        result = product_model.create_product(
            product_id=item['product_id'],
            orchard_product_id=item['upc'],
            upc=item['upc'],
            itunes_vendor_id=item.get('itunes_vendor_id', None),
            provider=item.get('provider', '')
        )
        product = result.message
        msg[product['product_id']] = None

        if not product_in_store.exists_in_store(
                product['product_id'], item['store_id']):
            pis = product_in_store.create_product_store_link(item)
            msg[product['product_id']] = pis

    return response.Response(msg)