How do I copy a remote image in python?

Question

I need to copy a remote image (for example http://site.com/image.jpg) to my server. Is this possible?

How do you verify that this is indeed an image?

Answer 1

Downloading stuff

import urllib
url = "http://site.com/image.jpg"
fname = "image.jpg"
urllib.urlretrieve( url, fname )

Verifying that it is a image can be done in many ways. The hardest check is opening the file with the Python Image Library and see if it throws an error.

If you want to check the file type before downloading, look at the mime-type the remote server gives.

import urllib
url = "http://site.com/image.jpg"
fname = "image.jpg"
opener = urllib.urlopen( url )
if opener.headers.maintype == 'image':
    # you get the idea
    open( fname, 'wb').write( opener.read() )

Answer 2

To download:

import urllib2
img = urllib2.urlopen("http://site.com/image.jpg").read()

To verify can use PIL

import StringIO
from PIL import Image
try:
    im = Image.open(StringIO.StringIO(img))
    im.verify()
except Exception, e:
    # The image is not valid

If you just want to verify this is an image even if the image data is not valid: You can use imghdr

import imghdr
imghdr.what('ignore', img)

The method checks the headers and determines the image type. It will return None if the image was not identifiable.

Answer 3

Same thing using httplib2...

from PIL import Image
from StringIO import StringIO
from httplib2 import Http

# retrieve image
http = Http()
request, content = http.request('http://www.server.com/path/to/image.jpg')
im = Image.open(StringIO(content))

# is it valid?
try:
    im.verify()
except Exception:
    pass  # not valid