The django-images from mysociety

django-images's People

Contributors

Stargazers

Watchers

django-images's Issues

How can I use the HasImageMixin?

Hello there:
I'm using this mixin as follow, but I'm not entirely sure if I'm doing it right
This is the test

def test_organization_has_images(self):
        organization = Organization.objects.create(name='La Cosa Nostra')
        image = Image.objects.create(content_object=organization)
        self.assertEqual(organization.primary_image(), image.image)

And when I run it I get this error:

Traceback (most recent call last):
  File "/home/falvarez/Workspace/votainteligente-portal-electoral/popular_proposal/tests/organization_tests.py", line 15, in test_organization_has_images
    self.assertEqual(organization.primary_image(), image.image)
  File "/home/falvarez/.virtualenvs/vota/src/django-images/images/models.py", line 59, in primary_image
    primary_image_model = self.primary_image_model()
  File "/home/falvarez/.virtualenvs/vota/src/django-images/images/models.py", line 65, in primary_image_model
    primary_images = [i for i in self.images.all() if i.is_primary]
AttributeError: 'Organization' object has no attribute 'images'

So, I decided to write a property for the Organization model

class Organization(PopoloOrganization, HasImageMixin):                                                                   
    _id = models.AutoField(primary_key=True)                                                                             

    @property                                                                                                            
    def images(self):                                                                                                    
        return Image.objects.filter(object_id=self._id)

Which does the job, but I'm not entirely sure if you meant the creation of that property or am I getting it all wrong?

Thanks for this code BTW. 😄

Recommend Projects