factories_v4.py

factories_v4.py

class PlayerFactory(DjangoModelFactory):

    # other declarations removed for brevity

    @post_generation
    def picture(self, create, extracted, **kwargs):
        if extracted:
            self.picture = extracted
        elif kwargs:
            assert 'filename' in kwargs, 'You need to pass in the filename'
            location = os.path.join(settings.BASE_DIR, f'fbd/championship/fixtures/{kwargs["filename"]}')
            assert os.path.isfile(location), f'{location} does not exist'
            self.picture = File(open(location, 'rb'))
        if create:
            self.save(update_fields=['picture'])