#!/usr/bin/python

# This script copies photos from a PTP camera into the photo directory and
# renames them using the Shutter Count EXIF tag (ignoring the camera's file
# name).
#
# Copyright (c) 2007 by Reid Priedhorsky, <reid@reidster.net>.
#
# This script is distributed under the terms of the GNU General Public
# License; see http://www.gnu.org/licenses/gpl.txt for more information.
#
# Usage: ptp-acquire [SUFFIX]



import cPickle as pickle
import glob
import os
import re
import subprocess
import shutil
import sys
import time

TARGET = '/storage/photos'

STATEFILE = TARGET + '/ptp-acquire.copied.txt'
RE_LIST = re.compile(r'^#(?P<id>\d+)\s+(?P<name>[\w.]+)', re.MULTILINE)
RE_FILE = re.compile(r'(?P<num>\d+)\.(?P<ext>\w{3})$')
RE_INFO = re.compile(r"""
^Information[ ]on[ ]file[ ]'(?P<name>[\w.]+)'
.*?
^[ ][ ]Time:\s+(?P<time>[\w :]+)$
""", re.VERBOSE | re.DOTALL | re.MULTILINE)

class Photo(object):
   __slots__  = ('id', 'name', 'timestamp')

def main():

   print 'Loading state'
   try:
      state = state_load(open(STATEFILE))
   except IOError, x:
      print "can't load state, starting fresh (%s)" % (x)
      state = set()

   print 'Listing photos'
   photos_all = photos_list()
   print '  found %d of them' % (len(photos_all))
   text = gphoto2_output('--show-info 1-%d' % (len(photos_all)))
   ms = RE_INFO.finditer(text)
   i = 0  # must manually manage i as findall() doesn't return match objects
   for m in ms:
      assert (photos_all[i].name == m.group('name'))
      photos_all[i].timestamp = time.strptime(m.group('time'))
      i += 1
   assert (i == len(photos_all))


   print 'Filtering photos'
   photos_wanted = list()
   for p in photos_all:
      if (needs_copy(p, state)):
         photos_wanted.append(p)
         record_copy(p, state)

   print 'Need %d images, skipping %d' % (len(photos_wanted),
                                          len(photos_all) - len(photos_wanted))
   if (len(photos_wanted) == 0):
      print 'Nothing to do, exiting'
      sys.exit()

   print 'Working in %s' % (TARGET)
   os.chdir('%s' % (TARGET))

   print 'Copying from camera'
   fail = os.system('gphoto2 --get-file %s'
                    % (",".join([str(p.id) for p in photos_wanted])))
   if (fail):
      print 'gphoto2 reported error, aborting'
      sys.exit(1)

   suffix = ''
   if (len(sys.argv) > 1):
      suffix = '_' + sys.argv[1]
   print 'Suffix is "%s"' % (suffix)

   print 'Renaming photos'
   for p in photos_wanted:
      dstdir = (time.strftime('%Y/%m/%d', p.timestamp) + suffix)
      m = RE_FILE.search(p.name)
      dstfile = "%s/%05d.%s" % (dstdir, shutter_count_get(p.name),
                                m.group('ext').lower())
      if (os.path.exists(dstfile)):
         print 'WARNING: %s unexpectedly exists, skipping' % (dstfile)
      else:
         if (not os.path.exists(dstdir)):
            print 'creating %s' % (dstdir)
            os.makedirs(dstdir)
         shutil.move(p.name, dstfile)

   print 'Saving state'
   state_dump(state, open(STATEFILE, 'w'))
   
   print 'Done'


def date_get(dirname):
   m = RE_DATE.search(dirname)
   return ('%s/%s' % (m.group(1), m.group(2)))

def gphoto2_output(arg_str):
   return os.popen('gphoto2 %s' % (arg_str)).read()

def photos_list():
   text = gphoto2_output('--list-files')
   photos = list()
   for m in RE_LIST.finditer(text):
      p = Photo()
      p.id = int(m.group('id'))
      p.name = m.group('name')
      photos.append(p)
   return photos

def needs_copy(photo, state):
   return (photo_hash(photo) not in state)

def photo_hash(photo):
   return '%s$%s' % (photo.name, time.asctime(photo.timestamp))

def record_copy(photo, state):
   state.add(photo_hash(photo))

def shutter_count_get(filename):
   return int(os.popen('exiftool -b -ShutterCount %s' % (filename)).read())

def state_dump(state, fp):
   for h in sorted(state):
      fp.write('%s\n' % (h))

def state_load(fp):
   s = set()
   for line in fp.readlines():
      s.add(line.replace('\n', ''))
   return s


if (__name__ == '__main__'):
   main()

# year=`date +'%Y'`

# cd $target/$year
# echo "working in $target/$year"

# for srcdir in $mount/$dcim/???_????; do
#    # extract the date; e.g. "/mnt/pentax/dcim/101_0131" -> "01/31"
#    dstdir=`echo $srcdir | sed -e 's/.*\/[0-9][0-9][0-9]_\([0-9][0-9]\)\([0-9][0-9]\)/\1\/\2/'`
#    if [ -e $dstdir ]; then
#       echo "$dstdir exists; skipping"
#    else
#       echo "copying photos to $dstdir"
#       mkdir -p $dstdir
#       for srcpath in $srcdir/imgp????.???; do
#          dstfile=`echo $srcpath | sed -e 's/.*\/imgp//'`
#          cp -uv $srcpath $dstdir/$dstfile
#       done
#    fi
# done

# umount $mount