import factory
from faker import Factory as FakerFactory

from outbo.apps.organizations import models

faker = FakerFactory.create()


class AgencyOrganizationFactory(factory.django.DjangoModelFactory):
    name = factory.LazyAttribute(lambda x: faker.company())

    class Meta:
        model = models.AgencyOrganization


class ClientOrganizationFactory(factory.django.DjangoModelFactory):
    name = factory.LazyAttribute(lambda x: faker.company())

    class Meta:
        model = models.ClientOrganization


class SupplierOrganizationFactory(factory.django.DjangoModelFactory):
    name = factory.LazyAttribute(lambda x: faker.company())

    class Meta:
        model = models.SupplierOrganization


class CategoryFactory(factory.django.DjangoModelFactory):
    name = factory.LazyAttribute(lambda x: faker.word())

    class Meta:
        model = models.Category


class BrandFactory(factory.django.DjangoModelFactory):
    name = factory.LazyAttribute(lambda x: faker.word())
    client = factory.SubFactory(ClientOrganizationFactory)

    class Meta:
        model = models.Brand