diff options
Diffstat (limited to 'lib/pb/generate.py')
| -rwxr-xr-x | lib/pb/generate.py | 113 |
1 files changed, 62 insertions, 51 deletions
diff --git a/lib/pb/generate.py b/lib/pb/generate.py index 12e3de1..5fc1402 100755 --- a/lib/pb/generate.py +++ b/lib/pb/generate.py @@ -1,23 +1,21 @@ #!/usr/bin/python2.7 -import sys -import os -from config import * +from config import BIN_CONVERT, OUTPUT_IMAGE_TYPES, DEFAULT_FINALFORMAT from pb import Pb -_default_tag = "im" +_DEFAULT_TAG = "im" -_gravity_params = [ +_GRAVITY_PARAMS = [ "NorthWest", "North", "NorthEast", "West", "Center", "East", "SouthWest", "South", "SouthEast" ] -_gravity_default = "Center" -_compose_params = [ +_GRAVITY_DEFAULT = "Center" +_COMPOSE_PARAMS = [ "Over", "ATop", "Dst_Over", "Dst_In", "Dst_Out", "Multiply", "Screen", "Divide", "Plus", "Difference", "Exclusion", "Pin_Light", "Lighten", "Darken", "Overlay", "Hard_Light", "Soft_Light", "Linear_Dodge", "Linear_Burn", "Color_Dodge", "Color_Burn" ] -_dispose_params = ["None", "Previous", "Background"] -_dispose_default = "None" +_DISPOSE_PARAMS = ["None", "Previous", "Background"] +_DISPOSE_DEFAULT = "None" class PbGenerate(Pb): #{{{ example params example_params = { @@ -49,59 +47,68 @@ class PbGenerate(Pb): #}}} def __init__(self, **kwargs): super(PbGenerate, self).__init__(**kwargs) + """ + Used to assert the value-types of the incoming parameters. + Types are defined as in their individual params classes. + """ _definitions = { - """ - Used to assert the value-types of the incoming parameters. - Types are defined as in their individual params classes. - """ #IMAGES - "url": {'type': "img_url"}, - "background": {'type': "img_url"}, + 'url': {'type': 'img_url'}, + 'background': {'type': 'img_url'}, #BOOLS - "coalesce": {'type': "bool"}, - "nearest": {'type': "bool"}, - "merge_early": {'type': "bool"}, - "flip": {'type': "bool"}, - "flop": {'type': "bool"}, - "tile": {'type': "bool"}, - "transparent": {'type': "bool"}, + 'coalesce': {'type': 'bool'}, + 'nearest': {'type': 'bool'}, + 'merge_early': {'type': 'bool'}, + 'flip': {'type': 'bool'}, + 'flop': {'type': 'bool'}, + 'tile': {'type': 'bool'}, + 'transparent': {'type': 'bool'}, #COLORS - "black": {'type': "color", 'default': 'black'}, - "white": {'type': "color", 'default': 'white'}, - "subtract": {'type': "color"}, + 'black': {'type': 'color', 'default': 'black'}, + 'white': {'type': 'color', 'default': 'white'}, + 'subtract': {'type': 'color'}, #INTS - "fuzz": {'type': "int"}, - "width": {'type': "int"}, - "height": {'type': "int"}, - "brightness": {'type': "int"}, - "contrast": {'type': "int"}, - "saturation": {'type': "int"}, - "rotate": {'type': "int"}, - "hue": {'type': "int"}, + 'fuzz': {'type': 'int'}, + 'width': {'type': 'int'}, + 'height': {'type': 'int'}, + 'brightness': {'type': 'int'}, + 'contrast': {'type': 'int'}, + 'saturation': {'type': 'int'}, + 'rotate': {'type': 'int'}, + 'hue': {'type': 'int'}, #ENUMS - "compose": {'type': "enum", 'enum_values': _compose_params, 'default': "Atop"}, - "gravity": {'type': "enum", 'enum_values': _gravity_params, 'default': _gravity_default}, - "dispose": {'type': "enum", 'enum_values': _dispose_params, 'default': "None"}, - "format": {'type': "enum", 'enum_values': OUTPUT_IMAGE_TYPES, 'default': DEFAULT_FINALFORMAT}, + 'compose': {'type': 'enum', 'enum_values': _COMPOSE_PARAMS, 'default': 'Atop'}, + 'gravity': { + 'type': 'enum', 'enum_values': _GRAVITY_PARAMS, 'default': _GRAVITY_DEFAULT + }, + 'dispose': { + 'type': 'enum', 'enum_values': _DISPOSE_PARAMS, 'default': 'None' + }, + 'format': { + 'type': 'enum', 'enum_values': OUTPUT_IMAGE_TYPES, 'default': DEFAULT_FINALFORMAT + }, #STRINGS "username": {'type': "string"}, "callback": {'type': "string"}, } - """Definitions and arguments are merged into attributes of the params object""" - self.params.definitions_import(_definitions, kwargs, classname=self.__class__.__name__) + self.params.definitions_import( + _definitions, kwargs, classname=self.__class__.__name__ + ) - """Used for the database tag column. Allows for tracking of the type + """Used for the database tag column. Allows for tracking of the type of overlay method used.""" - self.tag = _default_tag - if self.params.background: self.tag = self.params.compose - if self.params.transparent: self.tag = self.params.transparent + self.tag = _DEFAULT_TAG + if self.params.background: + self.tag = self.params.compose + if self.params.transparent: + self.tag = self.params.transparent self.filename, self.filepath = self._filename_filepath_create( url=self.params.url['url'], extension=self.params.format @@ -109,7 +116,7 @@ class PbGenerate(Pb): self._db_url_param = str(self.params.url['url']) - def _composite (self): + def _composite(self): """Imagemagick composite command""" cmd = [ BIN_CONVERT, self.params.background['path'], @@ -124,22 +131,26 @@ class PbGenerate(Pb): def _convert(self): """Imagemagick convert command""" cmd = [BIN_CONVERT, self.params.url['path']] - if self.params.rotate: cmd += ["-rotate", self.params.rotate] - if self.params.flip: cmd += ["-flip"] - if self.params.flop: cmd += ["-flop"] + if self.params.rotate: + cmd += ["-rotate", self.params.rotate] + if self.params.flip: + cmd += ["-flip"] + if self.params.flop: + cmd += ["-flop"] if self.params.transparent: if self.params.fuzz: cmd += ["-fuzz", "{}%".format(self.params.fuzz)] cmd += ["-transparent", self.params.subtract] if self.params.width or self.params.height: if self.params.nearest and self.params.format == "gif": - cmd += ["-coalesce","+map","-interpolate","Nearest","-interpolative-resize"] + cmd += ["-coalesce", "+map", "-interpolate", "Nearest", "-interpolative-resize"] else: cmd.append("-resize") cmd += ["{}x{}".format(self.params.width or "", self.params.height or "")] if self.params.black != "black" or self.params.white != 'white': - cmd += ["+level-colors" , "{},{}".format(self.params.black, self.params.white)] - if self.params.contrast: cmd += ['-contrast-stretch', self.params.contrast] + cmd += ["+level-colors", "{},{}".format(self.params.black, self.params.white)] + if self.params.contrast: + cmd += ['-contrast-stretch', self.params.contrast] if self.params.brightness or self.params.saturation or self.params.hue: cmd += [ "-modulate", "{},{},{}".format( @@ -148,7 +159,7 @@ class PbGenerate(Pb): self.params.hue or 100 ) ] - cmd.append("-coalesce"); #why? #FIXME + cmd.append("-coalesce")#why? #FIXME cmd += [self.filepath] self._call_cmd(cmd) |