web_shop_woocommerce/product.py

# The COPYRIGHT file at the top level of this repository contains the full
# copyright notices and license terms.
from trytond.pool import PoolMeta
from trytond.tools import slugify

from .web import ShopWooCommerceIdMixin


class Category(ShopWooCommerceIdMixin, metaclass=PoolMeta):
    __name__ = 'product.category'

    def get_woocommerce_entity(self):
        values = {
            'name': self.name,
            'slug': slugify(self.name).lower(),
            'parent': 0,
            }
        if self.parent:
            if self.parent.woocommerce_id:
                values['parent'] = self.parent.woocommerce_id
            else:
                return
        return values


class Product(ShopWooCommerceIdMixin, metaclass=PoolMeta):
    __name__ = 'product.product'

    def get_woocommerce_entity(self):
        short_description = description = (self.description or '')
        lines = description.splitlines()
        if len(lines) > 1:
            short_description = lines[0]
            description = '\n'.join(lines[1:])

        list_price = self.list_price
        sale_price = self.get_sale_price([self], 0)[self.id]
        if sale_price > list_price:
            list_price = sale_price
        values = {
            'name': self.name,
            'type': 'simple',
            'regular_price': str(list_price),
            'sale_price': str(sale_price),
            'description': description,
            'short_description': short_description,
            'status': 'publish',
        }
        if self.type != 'service':
            values['manage_stock'] = True
            values['stock_quantity'] = self.forecast_quantity
        if self.code:
            values['sku'] = self.code
        categories = []
        for category in self.categories:
            if category.woocommerce_id:
                categories.append({'id': category.woocommerce_id})
        values['categories'] = categories
        return values