from decimal import Decimal

from django.utils.timezone import now

from shop.models import Product, ProductPrice, Tax


class CreateProductsMixin:
    def create_product(
        self, sku='000001', name='Producto 1', description='Descripción', price=Decimal('10.00'), is_shipping=False
    ) -> Product:
        tax, created = Tax.objects.get_or_create(code='IVA', value=21)
        product = Product.objects.create(sku=sku, name=name, description=description, is_shipping_method=is_shipping)
        ProductPrice.objects.create(price=price, product=product, date=now(), tax=tax, current=True)
        return product