import logging
import pandas as pd

from store.models import (
    ProductTemplate,
    ProductCategoryParamValue,
    Product
)


logger = logging.getLogger(__name__)


class BaseLoader:
    def __init__(self, path):
        self.path = path

    def load_data(self):
        return pd.read_csv(self.path)
        

class ProductLoader(BaseLoader):
    
    def _process_row(self, row):
        template = ProductTemplate.objects.get(code=row["template"])
        price = float(row["price"])
        name = row["name"]
        available = bool(row["available"])
        params = []
        for param in row["params"]:
            key, value = param
            param = ProductCategoryParamValue.objects.get(param__key=key, value=value)
            params.append(param)
        product = Product.objects.get_or_create_by_params(template=template, params=params)
        product.price = price
        product.name = name
        product.available = available
        product.save()
        return product

    def process(self):
        data = self.load_data()
        products = []
        for _, row in data.iterrows():
            try:
                product = self._process_row(row)
            except Exception as e:
                # catch any error and log it, GlitchTip will catch it
                logger.exception(str(e))
            else:
                products.append(product)
        logger.info(f"Loaded {len(products)} products")