getting ready

1 files changed, 33 insertions, 0 deletions

diff --git a/frontend/imdither/urlDownload.py b/frontend/imdither/urlDownload.py
new file mode 100755
index 0000000..f806650
--- /dev/null
+++ b/frontend/imdither/urlDownload.py
@@ -0,0 +1,33 @@
+#!/usr/bin/python2.7
+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])