import exceptions, body import os, sys, traceback, shutil import string import re import Image import codecs import functions from node import Node from functions import printTraceback, trim, xmlString, getYesNo from PIL import Image #################### # class picture #################### """ A model of a picture which inherits from node class. Opens picture and retrieves width and height Can print its own index file and the link to it in the main index file """ class Picture(Node): def __init__(self, props=None, absfilename=None): Node.__init__(self) self.absfilename = absfilename self.width = 0 self.height = 0 self.res_width = 0 self.thumb_width = 0 self.format = None self.res_height = 0 self.thumb_height = 0 self.date = "" self.where = "" self.who = "" self.what = "" self.exif = None self.im = None self.width = None self.height = None self.format = None self.mode = None self.info = None self.camera = None self.highQuality = 0 if props: self.res_width = props["size"] self.thumb_width = props["thumbsize"] self.camera = None #props["camerainfo"] // future property self.highQuality = not props["faster"] self.getImage() def tostring(self): return "Filename: " + self.absfilename + "\n" + "Width:" + str(self.width) + "\n" + "Height:" + str(self.height) + "\n" + "Resized width:" + str(self.res_width) + "\n" + "Resized height:" + str(self.res_height) +"\n" def getAbsfilename(self): return self.absFilename def getWidth(self): return self.width def getResizedWidth(self): return self.res_width def getThumbWidth(self): return self.thumb_width def getHeight(self): return self.height def getResizedHeight(self): return self.res_height def getThumbHeight(self): return self.thumb_height def getDate(self): "Return date, if empty get it from EXIF data in picture" if self.date == "": self.getExifDate() return self.date def getWho(self): return self.who def getWhere(self): return self.where def getWhat(self): return self.what def setAbsfilename(self, absfilename): self.absfilename = absfilename def setWidth(self, width): if width is None: self.width = 0 else: self.width = int(width) def setResizedWidth(self, res_width): if res_width is None: self.res_width = 0 else: self.res_width = int(res_width) def setThumbWidth(self, thumb_width): if thumb_width is None: self.thumb_width = 0 else: self.thumb_width = int(thumb_width) def setHeight(self, height): if height is None: self.height = 0 else: self.height = int(height) def setResizedHeight(self, res_height): if res_height is None: self.res_height = 0 else: self.res_height = int(res_height) def setThumbHeight(self, thumb_height): if thumb_height is None: self.thumb_height = 0 else: self.thumb_height = int(thumb_height) def setDate(self, date): self.date = date def setWho(self, who): self.who = who def setWhere(self, where): self.where = where def setWhat(self, what): self.what = what def unload(self): """Unload picture from memory""" self.im = None def getExifDate(self): "Extract timestamp from EXIF data" try: self.date = self.exif[306].replace(":","-",2) # date in format yyyy-MM-DD hh:mm:ss except: return def getExifCamera(self): "Extract camera from EXIF data" # see http://www.exif.org/Exif_2-1.PDF try: make = self.exif[271] model = self.exif[272] if make.split()[0] in model: result = model # make is in model name, so don't mention it twice else: result = make + ' ' + model t = self.exif[33434] # shutter time d = self.exif[33437] # diafragma focus = self.exif[37386] flash = getYesNo(self.exif[37385]) ep = self.exif[34850] # exposure program eps=['undefined (0)','manual','normal','aperture priority','shutter priority','max. depth','speed','portrait','landscape'] if ep in range(len(eps)): ep = eps[ep] else: ep = str(ep) result = result + ("
Exposure: 1/%d sec. F/%.1f")%(t[1]/t[0], float(d[0])/d[1]) result = result + ("   flash: %s   focus %d mm.   Exposure program: %s")%(flash, focus[0]/focus[1], ep) except: result = 'unknown' return result def getExifOrientation(self): orientation = self.exif[274] rotationOffsets = {3:180, 6:270, 8:90} # CW, upside-down, CCW return rotationOffsets.get(orientation, 0) def getSize(self): return (self.width, self.height) def getImage(self): "Return a PIL Image object of picture, set properties on self" try: if self.absfilename: if self.im is None: self.im = Image.open(open(self.absfilename, "rb")) self.width = int(self.im.size[0]) self.height = int(self.im.size[1]) self.format = self.im.format self.mode = self.im.mode if self.format == "JPEG": if self.highQuality: self.im.info["quality"] = 100 self.im.info["optimize"] = 1 else: self.im.info["quality"] = 1 self.im.info["progression"] = 1 self.info = self.im.info # if we have PIL lib less than 1.1.4 _getexif() fails, trap it try: self.exif = self.im._getexif() except: pass except: printTraceback() return self.im def calculateSize(self, requestedSize): """Return tuple describing (width,height) resized picture given a requested size. The requested size is the size of the longer edge""" _height = 0 _width = 0 try: if self.width > self.height: ratio = self.width * 1.0 / requestedSize _height = int(self.height / ratio) _width = requestedSize else: "If portrait - use width as height, must calculate width" ratio = self.height * 1.0 / requestedSize _height = requestedSize _width = int(self.width / ratio) return(_width,_height) except: printTraceback() def getResizedPicture(self, sizeTuple): """Return a PIL Image instance, where size is given in supplied tuple (width, heigth) Also check for orientation""" im = self.getImage() offsetDegree = self.getExifOrientation() if offsetDegree != 0: im = im.rotate(offsetDegree) if self.highQuality: return im.resize(sizeTuple,Image.ANTIALIAS) else: return im.resize(sizeTuple,Image.NEAREST) def putResized(self, destdirectory): "Put myself in destdirectory and resize to self.res_width" try: location = self.getLocation(destdirectory) sizeTuple = self.calculateSize(self.res_width) self.getResizedPicture(sizeTuple).save(open(location,"wb"), self.format) except: printTraceback() def putThumbnail(self, destdirectory): "Put myself in destdirectory and resize to self.thumb_width" try: location = self.getLocation(destdirectory) sizeTuple = self.calculateSize(self.thumb_width) self.getResizedPicture(sizeTuple).save(open(location, "wb"), self.format) except: printTraceback() def putInDir(self, destdirectory, size): """Put myself resized to size in descdirectory. Size is the length of the longer side""" location = self.getLocation(destdirectory) sizeTuple = self.calculateSize(size) self.getResizedPicture(sizeTuple).save(open(location, "wb"), self.format) def putFullsize(self, destdirectory): "Put self in destdirectory, no resizing" location = self.getLocation(destdirectory) shutil.copyfile(self.absfilename, location) def getLocation(self, destinationDir): "Construct directory location from given destinationDir and sourcefile tail" return os.path.join(destinationDir, os.path.split(self.absfilename)[1]) def getNameNoExtension(self): "Return filename stripped for extension" filename = os.path.basename(self.absfilename) return filename[0:filename.rfind(".")] def getTail(self): "Return filename stripped for extension" return os.path.split(self.absfilename)[1] def getLinkHTML(self, fullsize=None,): "Return HTML for inclusion in gallery index file" (xthumb_width, xthumb_height) = self.calculateSize(self.thumb_width) html = [] html.append("\n ") html.append("\""") if fullsize: html.append("
Full size") html.append("") return "".join(html) def getLinkHTMLForSinglePic(self, fullsize=None): """If a directory has only a single picture, resize it to 640 and center it""" html = [] html.append("
\n\n
") html.append("") html.append("
\n\n") html.append("") return "".join(html) def createPictureHTML(self, directory, stylesheet, fullsize, footer): "Create HTML file to display resized picture" html = [] html.append("\n \n \n " + self.getTail()) html.append("\n \n \n") html.append(r'') html.append("\n \n") html.append(self.navigationBar(fullsize)) html.append("\n
\n \n \n ") if self.height > self.width: # place date right of picture, after 30 html.append("\n \n ") else: # place date under picture html.append("\n \n \n \n \n \n ") html.append("\n \n
\n \n \n ") html.append(self.tagLabelContent()) html.append("\n
\n
") html.append(self.tagLabelContent()) html.append("\n
\n
") html.append(self.navigationBar(fullsize)) html.append(footer.getHTML()) raw = os.path.join(directory, self.getNameNoExtension()) + ".html" file = open(raw, "w") file.write("".join(html).encode('utf-8')) file.close() def navigationBar(self, fullsize): "Return navigation bar, only simple links at the moment" html = [] if self.next != None: html.append("\n

\n [Next]") if self.previous != None: html.append("\n [Previous]" ) if fullsize: html.append("\n [Fullsize]") html.append("\n [Gallery]\n

\n") return string.joinfields(html) def tagLabelContent(self): "Return picture descriptions" html = [] html.append("\n Date: " + self.getDate()) if self.who: html.append("
\n Who: " + self.who) if self.where: html.append("
\n Where: " + self.where) if self.what: html.append("
\n What: " + self.what) if self.camera: html.append("
\n Camera: " + self.getExifCamera()) return string.joinfields(html) def getXML(self, xml): "Return XML code for this picture" xml.append("\n ") xml.append("\n " + xmlString(self.getTail()) + "") xml.append("\n " + str(self.width) + "") xml.append("\n " + str(self.res_width) + "") xml.append("\n " + str(self.thumb_width) + "") xml.append("\n " + str(self.height) + "") xml.append("\n " + str(self.res_height) + "") xml.append("\n " + str(self.thumb_height) + "") xml.append("\n " + xmlString(self.date) + "") xml.append("\n " + xmlString(self.who) + "") xml.append("\n " + xmlString(self.where) + "") xml.append("\n " + xmlString(self.what) + "") xml.append("\n " + xmlString(self.getExifCamera()) + "") xml.append("\n ") return xml