diff options
Diffstat (limited to 'lib/Pb/__init__.py')
| -rw-r--r-- | lib/Pb/__init__.py | 127 |
1 files changed, 88 insertions, 39 deletions
diff --git a/lib/Pb/__init__.py b/lib/Pb/__init__.py index 4d8a8d4..eab9878 100644 --- a/lib/Pb/__init__.py +++ b/lib/Pb/__init__.py @@ -1,27 +1,33 @@ import re from Config import * import time -import urllib -import urllib2 -import sys -import os +import urllib, urllib2 +import sys, os import random from subprocess import Popen,PIPE,call from Params import Params -import time +import sha + Request = urllib2.Request urlencode = urllib.urlencode urlopen = urllib2.urlopen _max_filename_length = 20; +class PbProcessError(Exception): + pass + class Pb(object): - def __init__(self): + def __init__(self, **kwargs): + self._input_kwargs = kwargs self._now = str(int(time.time())); self.params = Params(classname=self.__class__.__name__, now=self._now); self._files_created = [] self.commands = []; self._working_dir = WORKING_DIR + self._tag = self.__class__.__name__ + self._hashdir = None + self._db_url_param = "NULL" def _filename_create(self, url=None, namepart="", extension=""): if url: @@ -47,6 +53,9 @@ class Pb(object): _filename = self._filename_create(namepart=namepart, extension=extension) return self._filepath_create(_filename, directory=directory) + def _hashdir_create(self): + self._hashdir = sha.new(self.filename).hexdigest()[:2] + def _url_sanitize (self, s): return re.sub(r'\W+', '', s) @@ -58,28 +67,26 @@ class Pb(object): except Exception: raise Exception("Unable to call cmd {}".format(str(cmd))) + def _dimensions(self, filepath): + ident = (Popen([BIN_IDENTIFY, filepath], stdout=PIPE).communicate()[0]).split(" ") + return ident[2].split("x") - @staticmethod - def dimensions (filepath): - #works in lieu of a mimetype check (it reads the header as well) - ident = (Popen([BIN_IDENTIFY, filepath], stdout=PIPE).communicate()[0]).split(" ") - return ident[2].split("x") - + def file_dimensions (self): + return self._dimensions(self.filename) + def _width_and_height_set(self, filepath=None, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT): if filepath: - self.width, self.height = Pb.dimensions(filepath) + self.width, self.height = self._dimensions(filepath) return self.width = width self.height = height - @staticmethod - def file_size (filepath): - try: - return os.stat(filepath)[6] - except Exception as e: - sys.stderr.write("Couldn't determine filesize\n") - sys.stderr.write(str(e)+"\n") - raise; + def file_size(self): + try: + return os.stat(self.filepath)[6] + except Exception as e: + self.err_warn("Couldn't determine filesize of %s\n" % self.filepath) + self.err_warn(str(e)) def _file_read(self, filepath): f = open(filepath, 'r'); @@ -89,14 +96,17 @@ class Pb(object): def err_warn(self, s): sys.stderr.write("ERROR:{} - {}\n".format(self.__class__.__name__, s)) - + raise PbProcessError + def _cleanup(self): - if not self._files_created: return - cmd = ["rm"]+self._files_created - self._call_cmd(cmd) + if not self._files_created: return + map(lambda n: os.remove(n), self._files_created) + + def _file_clean_local(self): + os.remove(self.filepath) def err_fatal(self, s): - sys.stderr.write("ERROR[FATAL]:{} - {}\n".format(self.__class__.__name__, s)) + sys.stderr.write("ERROR[FATAL]:%s - %s\n".format(self.__class__.__name__, s)) sys.exit(1); @classmethod @@ -113,18 +123,57 @@ class Pb(object): @staticmethod def gif_frames(filepath): - try: - info = Popen([BIN_IDENTIFY,filepath], stdout=PIPE).communicate()[0] - frames = filter((lambda x: x), map( - (lambda x: x.split(" ")[0]), - (info).split('\n') - )) - return frames - except Exception as e: - self.err_warn("couldn't get gif frames") - raise e; + try: + info = Popen([BIN_IDENTIFY,filepath], stdout=PIPE).communicate()[0] + frames = filter((lambda x: x), map( + (lambda x: x.split(" ")[0]), + (info).split('\n') + )) + return frames + except Exception as e: + self.err_warn("couldn't get gif frames") + raise e; def _choose_gif_frame(self, filepath): - _gif_frames = Pb.gif_frames(filepath) - frame = random.choice(_gif_frames) - self._call_cmd([BIN_CONVERT, frame, filepath]) + _gif_frames = Pb.gif_frames(filepath) + frame = random.choice(_gif_frames) + self._call_cmd([BIN_CONVERT, frame, filepath]) + + def db_send(self, remote_addr="NULL"): + try: + db.insert_cmd(**{ + 'date' : self._now, + 'remote_addr' : remote_addr, + 'username' : self.params.username or "NULL", + 'url' : self._db_url_param, + 'directory' : self._hashdir, + 'oldfile' : "NULL", #weird + 'newfile' : im.filename, + 'dataobj' : json.dumps(dict(self._input_kwargs)), + 'cmd' : ";".join(self.commands), + 'tag' : self._tag, + }) + except Exception as e: + self.err_warn("Problem sending to database") + + def file_s3move(self): + self._hashdir_create() + s3move(im.filepath, "im/{}/{}".format(self._hashdir, im.filename)) + self._files_created.append(self.filepath) + self._file_clean_local(); + + def file_json(self): + dimensions = self.file_dimensions(); + return json.dumps({ + 'url' : "im/%s/%s".format(BASE_URL, self._hashdir, self.filename), + 'size' : self.file_size(), + 'width' : "{}px".format(dimensions[0]), + 'height' : "{}px".format(dimensions[1]), + }) + +from Pb.Grid import PbGrid +from Pb.Break import PbBreak +from Pb.Pattern import PbPattern +from Pb.Generate import PbGenerate +from Pb.Landscape import PbLandscape +from Pb.Gradient import PbGradient |