summaryrefslogtreecommitdiff
path: root/impattern/urlDownload.py
diff options
context:
space:
mode:
Diffstat (limited to 'impattern/urlDownload.py')
-rw-r--r--impattern/urlDownload.py33
1 files changed, 33 insertions, 0 deletions
diff --git a/impattern/urlDownload.py b/impattern/urlDownload.py
new file mode 100644
index 0000000..37e7909
--- /dev/null
+++ b/impattern/urlDownload.py
@@ -0,0 +1,33 @@
+#!/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])