(root)/misc-scripts : /image-process.py (revision 75)

#!/usr/bin/python -tt

""" image-process.py (c) 2008 Matthew John Ernisse <mernisse@ub3rgeek.net>

Redistribution and use in source and binary forms,

with or without modification, are permitted provided

that the following conditions are met:

    * Redistributions of source code must retain the

      above copyright notice, this list of conditions

      and the following disclaimer.

    * Redistributions in binary form must reproduce

     the above copyright notice, this list of conditions

      and the following disclaimer in the documentation

      and/or other materials provided with the distribution.

THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS

FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE

COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,

INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,

BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS

OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND

ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR

TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE

USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

$Id: image-process.py,v 1.3 2008/12/31 06:30:48 mernisse Exp $

"""

import datetime

import gio

import getopt

import os

import pyexiv2

import random

import re

import sys

from stat import *

# directories... I really wish there was a better way to query this crap.

# I should be able to get this from Gnome/XDG

PHOTOS = "%s/%s/mine" % (os.environ["HOME"], "Pictures")

OTHER = "%s/%s/other" % (os.environ["HOME"], "Pictures")

# these are evil tags that fuck my term up.

BLACKLIST = [

	"Exif.Nikon3.DataDump",

	]

def Usage():

	binary = os.path.basename(sys.argv[0])

	print """Usage: %s [-v] FILE ... 

   or: %s -d [-v] DIRECTORY ...

Process images listed on the command line.

Options:

	-d 	treat command line argumetnts as directories.

	-v	display EXIF data prior to renaming

""" %  (binary, binary)

	return

def fileCopy(src, dst):

	fd = [None, None]

	buf = None

	bs = 16384

	count = 0

	try:

		fd[0] = open(src, "rb")

	except IOError, e:

		print "couldn't open src %s (%s)" % (src, str(e))

		return 127

	try:

		fd[1] = open(dst, "wb")

	except IOError, e:

		print "couldn't open dst %s (%s)" % (dst, str(e))

		return 127

	(st_mode, st_ino, st_dev, st_nlink, st_uid, st_gid, st_size, st_atime, 

	 st_mtime, st_ctime) = os.stat(src)

	sys.stdout.write("0%")

	while True:

		buf = fd[0].read(bs)

		if buf:

			fd[1].write(buf)

			count += 1

			pct = (st_size - (count * bs)) / 100

			if pct > 25 and pct < 50:

				sys.stdout.write("..25%")

			if pct > 50 and pct < 75:

				sys.stdout.write("..50%")

			if pct > 75 and pct < 100:

				sys.stdout.write("..75%")

		else:

			break

	fd[0].close()

	fd[1].flush()

	fd[1].close()

	try:

		os.utime(dst, (st_atime, st_mtime))

		os.chmod(dst, st_mode)

	except Exception, e:

		print "couldn't update dst's attributes. %s" % (str(e))

		os.unlink(dst)

	sys.stdout.write("..100%\n")

	return 0

def printExif(file):

	global BLACKLIST

	image = pyexiv2.Image(file)

	image.readMetadata()

	for key in image.exifKeys():

		if key in BLACKLIST:

			continue

		print "%s: %s" % (key, image[key])

def dateTimeRename(file):

	global PHOTOS

	image = pyexiv2.Image(file)

	image.readMetadata()

	make = "UNKN"

	model = "0000"

	try:

		capture_date = image['Exif.Photo.DateTimeOriginal']

	except:

		print "couldn't get EXIF information for %s." % (file)

		mtime = os.stat(file)[ST_MTIME]

		capture_date = datetime.date.fromtimestamp(mtime)

		pass

	try:

		make = image['Exif.Image.Make']

		model = image['Exif.Image.Model']

	except:

		# this isn't fatal.

		pass

	fname = "%s_%s-%s.jpg" % (

		capture_date.strftime("%Y-%m-%d_%H-%M-%S"),

		make,

		model

	)

	archive = "%s/%s" % ( 

		capture_date.strftime("%Y"),

		capture_date.strftime("%m_%B")

	)

	if not os.path.exists("%s/%s" % (PHOTOS, archive)):

		try:

			os.makedirs("%s/%s" % (PHOTOS, archive))

		except Exception, e:

			print "couldn't make %s/%s, %s" % (

				PHOTOS,

				archive,

				str(e)

			)

			return 127

	print "%s" % ( fname )

	prefix = fname

	while True:

		if os.path.exists("%s/%s/%s" % ( PHOTOS, archive, fname )):

			# The dest file exists, let us err on the side of 

			# safety -- try to find ourselves a fname that 

			# exists.

			fname = prefix + "_" + random.randrange(0,9999)

		else:

			break

	fileCopy(file, "%s/%s/%s" % ( PHOTOS, archive, fname ))

	return 0

def Main():

	global PHOTOS, OTHER

	if not os.path.exists(PHOTOS):

		print "%s: does not exist." % (PHOTOS)

		sys.exit(127)

	if not os.path.exists(OTHER):

		print "%s: does not exist." % (OTHER)

		sys.exit(127)

	dirs = None

	links = None

	verbose = None

	files = []

	try:

		opts, args = getopt.getopt(sys.argv[1:], "dv")

	except Exception, e:

		print "str(e)"

		Usage()

		return 1

	for o, a in opts:

		if o == "-v":

			verbose = True

		elif o == "-d":

			dirs = True

		else:

			Usage()

			return 1

	# process the args as dirnames, walk them and push the files up into

	# the files list.  else if dirs is false, then just copy args [] to

	# files []

	if dirs:

		for dir in args:

			for d in os.walk(dir):

				for file in d[2]:

					files.append("%s/%s" % ( d[0], file ))

	else:

		files = args

	for file in files:

		if re.search(r'^\.', os.path.basename(file)):

			# We don't touch hidden files or folders.

			continue

		if not re.search(r'\.jpg$', file, re.I):

			print "%s: moving non-jpeg" % (file)

			code = fileCopy(file, "%s/%s" % (

				OTHER,

				os.path.basename(file)

				)

			)

			if code > 0:

				return code

			gF = gio.File(file)

			if not gF.trash():

				print "Could not throw out %s" % (file)

			continue

		if not os.path.exists(file):

			print "%s: File does not exist" % (file)

			continue

		if verbose:

			code = printExif(file)

			if code > 0:

				return code

		code = dateTimeRename(file)

		if code > 0:

			return code

		# use Gnome's GIO bindings to throw the file

		# out, using the Gnome Trash, this seems nicer

		# than just calling os.unlink

		gF = gio.File(file)

		if not gF.trash():

			print "Could not throw out %s" % (file)

if __name__ == "__main__":

	if not len(sys.argv) > 1:

		Usage()

		sys.exit(1)

	sys.exit(Main())