Question

I’ve got this model in my Django app:

class Image(models.Model):
    image_file = models.ImageField(
        upload_to='images/', 
        width_field='width',
        height_field='height'
    )
    width = models.PositiveIntegerField(
        blank = True, null = True,
        editable = False
    )
    height = models.PositiveIntegerField(
        blank = True, null = True,
        editable = False
    )

    sha1 = models.CharField(max_length=32, blank=True, editable=False)
    filesize = models.PositiveIntegerField(blank=True, null=True, editable=False)

I can now upload images through the Django admin site. And the width and height properties are saved in the database automatically when it’s uploaded, because of the special ImageField parameters.

But I’d also like it to automatically work out the uploaded file’s size and SHA-1 digest, and save those properties too. How would I do this?

Answer 1

Its been a while, but something like this should work:

import hashlib
class Image(models.Model):
#...
    def save(self, *args, **kwargs):
        super(Image, self).save(*args, **kwargs)
        f = self.image_file.open('rb')
        hash = hashlib.sha1()
        if f.multiple_chunks():
           for chunk in f.chunks():
              hash.update(chunk)
        else:    
              hash.update(f.read())
        f.close()
        self.sha1 =  hash.hexdigest()
        self.filesize = self.image_file.size 

EDIT:
Added suggestion by Dan on reading by chunk. Default chunk size is 64KB.