"""
This file holds the functions necesary to:

- Connect to seller's shop API
- Load information on seller [??]
- Load product information

"""
import logging
from io import BytesIO
import datetime

from woocommerce import API
import requests
from PIL import Image
from django.core.files import File

from companies.models import Company
from products.models import Product
from products.serializers import ProductSerializer
from history.models import HistorySync


logging.basicConfig(
    filename='logs/woocommerce.log',
    filemode='w',
    format='%(levelname)s:%(message)s',
    level=logging.INFO,
    )


PRODUCT_FIELDS = [f.name for f in Product._meta.get_fields()]


def get_wcapi_instance(url, key, secret, version="wc/v3"):
    wcapi = API(
        url=url,
        consumer_key=key,
        consumer_secret=secret,
        wp_api=True,
        version=version
    )
    return wcapi


def create_imported_product(info, company, history, user):
    # extract m2m field data
    tags = [t.get('name') for t in info.pop('tags')]
    attributes = [t.get('name') for t in info.pop('attributes')]
    # prepare instance data
    instance_data = {
        'company':company.id,
        'creator': user.id if user is not None else None,
        'history': history.id,
        'url': info['permalink'],
        }
    # parse the product info
    for key in info:
        if key in PRODUCT_FIELDS:
            instance_data[key] = info[key]
    # remove unwanted fields
    instance_data.pop('id')

    # alternative method
    serializer = ProductSerializer(data=instance_data)
    if serializer.is_valid():
        try:
            new = Product.objects.create(**serializer.validated_data)
            new.tags = tags
            new.attributes = attributes
            new.save()
        except Exception as e:
            logging.error(f"Could not create product instance: {str(e)}")
            return None
        try:
            # get image
            headers={"User-Agent" : "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.103 Safari/537.36"}
            image_url = info['images'][0]['src']
            response = requests.get(image_url, stream=True, headers=headers)
            response.raw.decode_content = True
            image = Image.open(response.raw)
            # save using File object
            img_io = BytesIO()
            image.save(img_io, format='JPEG')
            new.image.save(f"{new.name}-{new.sku}.jpg", File(img_io), save=False)
            new.save()
        except Exception as e:
            logging.error(f"Could not add image to product: {str(e)}")
        return new
    else:
        logging.error(f"{serializer.errors}")
        return []


def migrate_shop_products(url, key, secret, user=None, version="wc/v3"):
    """Tries to connect to WooCommerce site @ url with given credentials

    If succesful, returns list of Product instances created

    """
    # get wcapi
    wcapi = get_wcapi_instance(url, key, secret, version)

    consumer_key = 'ck_565539bb25b472b1ff7a209eb157ca11c0a26397'
    consumer_secret = 'cs_9c1690ba5da0dd70f51d61c395628fa14d1a104c'

    # get company fom url
    company = Company.objects.filter(web_link=url).first()

    if not company:
        logging.error(f"Could not find Company with URL: {url}")
        print(f"Could not find Company with URL: {url}")
        return None

    new_products = []
    page = 1
    # list products
    while True:
        try:
            response = wcapi.get('products/', params={'per_page': 10, 'page': page})
            page += 1
            if response.status_code == 200:
                products = response.json()
            elif response.status_code == 401:
                logging.error(f"{response.status_code} [{response.url}]: {response.json()}")
                return None
            else:
                logging.error(f"Could not load products from {response.url}: [{response.status_code}]")
                print(f"Could not load products fom {response.url}: [{response.status_code}]")
                return None
        except requests.exceptions.ReadTimeout as e:
            logging.error(f"Timeout reading backend: {str(e)}")
            # skip and try next
            continue
        # exit loop if no more products
        if not products:
            break

        # create HistorySync instance and link to every product created
        history = HistorySync.objects.create(
            company=company,
            sync_date=datetime.datetime.now(),
        )

        counter = 0
        for product in products:
            new = create_imported_product(product, company, history, user)
            new_products.append(new)
            counter += 1
            logging.info(f"Product '{new.name}' created")
            print(f"Product '{new.name}' created")

        # update history.quantity
        history.quantity = counter
        history.save()

    print(f"Products created: {len(new_products)}")
    return new_products