#!/usr/bin/python
import pbutils
from os import getpid
class UrlDownload:
  def __init__(self, url):
    self.dir = "/tmp/";
    self.url = url
    self.filename = self.makeTempname()
    try: 
      self.download();
    except Exception as E:
      print E
      pbutils.error("Could not download image")

  def makeTempname(self):
    self.url = self.url.split("?")[0]
    parts = self.url.split(".")
    namepart = pbutils.sanitize(parts[-2].split("/")[-1])
    extension = pbutils.sanitize(parts[-1])
    return str(getpid()) + namepart + "." + extension;
   

  def getFilename(self):
    return self.filename
  def download(self):
    pbutils.download_image(self.url, self.filename, self.dir)

from sys import argv

if __name__ == "__main__":
  if len(argv) < 2:
    pbutils.error("supply a url as an argument")
  download = UrlDownload(argv[1])