#save_image.py
import urllib2, urlparse
from twill import commands

def save_image(image_name):
    browser = commands.browser
    url = browser.get_url()

    f_in = urllib2.urlopen(urlparse.urljoin(url, image_name))
    image_data = f_in.read()
    f_in.close()

    f_out = file(image_name, 'wb')
    f_out.write(image_data)
    f_out.close()