Limit the format and size (width and height) of the uploaded image in Django.

limit_uploaded_image

from cStringIO import StringIO

from PIL import Image
from django.http import HttpResponse


ALLOWED_FORMAT = (
    'JPEG',
    'PNG',
)
ALLOWED_SIZE = (
    (80， 60),
    (100， 85),
)


def upload(request):
    """The handler of the upload request."""
    if request.method == 'POST' and request.FILES:
        f = request.FILES['file']
        data = f.read()  # save file data for reuse
        
        fimage = StringIO(data)
        if check_image(StringIO(data)):
            return allow(fimage)
        else:
            return deny(fimage)
            

def check_image(fimage):
    """Check the format and size (width and height) of `fimage`."""
    image = Image.open(fimage)
    return image.format in ALLOWED_FORMAT and image.size in ALLOWED_SIZE
            
            
def allow(fimage):
    """The action when check successful."""
    return HttpResponse('true')
                
            
def deny(fimage):
    """The action when check failed."""
    return HttpResponse('false')