import factory

from factory.django import DjangoModelFactory

from django.utils import timezone

from psycopg2.extras import DateTimeTZRange

from utils.factories import UserFactory


class ProductCategoryFactory(DjangoModelFactory):
    class Meta:
        model = 'shop.ProductCategory'

    name = factory.Faker('word')


class ProductFactory(DjangoModelFactory):
    class Meta:
        model = 'shop.Product'

    name = factory.Faker('word')
    slug = factory.Faker('word')
    category = factory.SubFactory(ProductCategoryFactory)
    description = factory.Faker('paragraph')
    price = factory.Faker('pyint')
    available_in = factory.LazyFunction(
        lambda:
            DateTimeTZRange(
                lower=timezone.now(),
                upper=timezone.now() + timezone.timedelta(31)
            )
    )


class OrderFactory(DjangoModelFactory):
    class Meta:
        model = 'shop.Order'

    user = factory.SubFactory(UserFactory)


class OrderProductRelationFactory(DjangoModelFactory):
    class Meta:
        model = 'shop.OrderProductRelation'

    product = factory.SubFactory(ProductFactory)
    order = factory.SubFactory(OrderFactory)
    quantity = 1
    handed_out = False