Random Photo Colors and Sizes

I see that I can create a photo with a certain color and size like this:

class MyFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.MyModel

    the_image = factory.django.ImageField(color='blue')

Is it possible to randomize that color and also the dimensions?

More generally, this means using one declaration as the input to another declaration in generation.

[kaushal235]

Custom workaround for this issue is override ImageField & if any callable pass call it run time

class CustomImageField(factory.django.ImageField):

    def _make_data(self, params):
        color = params.pop('color', 'blue')
        if callable(color):
            color = color()
        params['color'] = color
        return super(CustomImageField, self)._make_data(params)

& use it as

class MediaItemFactory(factory.DjangoModelFactory):
    image = CustomImageField(color=faker.safe_color_name)
   
    class Meta:
         model = ModelName

here faker.safe_color_name is function that generate random color name

[x-yuri]

I’ve found the same can be done by overriding generate method:

class RandomImageField(factory.django.ImageField):
    def generate(self, step, params):
        for n in ['width', 'height']:
            if not n in params:
                continue
            v = params[n]
            if callable(v):
                v = v()
            params[n] = v
        return super(RandomImageField, self).generate(step, params)

class PhotoFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Photo
    file = RandomImageField(
        width=faker().image_width,
        height=faker().image_height)

def faker():
    return factory.Faker._get_faker()

class MyProvider(BaseProvider):
    def image_width(self):
        return self.random_int(300, 400)

    def image_height(self):
        return self.random_int(400, 500)

factory.Faker.add_provider(MyProvider)

Seeing how generate doesn’t start with an underscore make it feel like a more legitimate/safe way. But if you want to persist the generated values in the model (it has corresponding fields), you can replace:

            params[n] = v

with:

            step.attributes[n] = params[n] = v

Or better using evaluate:

from collections import ChainMap

class RandomImageField(factory.django.ImageField):
    def evaluate(self, instance, step, extra):
        return super(RandomImageField, self).evaluate(instance, step,
            ChainMap(extra,
                {k: getattr(instance, v) for k, v in self.defaults.items()
                    if k in ['width', 'height']}))

class PhotoFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Photo
    width = factory.Faker('image_width')
    height = factory.Faker('image_height')
    file = RandomImageField(width='width', height='height')