summaryrefslogtreecommitdiff
path: root/rapid/media.py
diff options
context:
space:
mode:
Diffstat (limited to 'rapid/media.py')
-rwxr-xr-xrapid/media.py180
1 files changed, 180 insertions, 0 deletions
diff --git a/rapid/media.py b/rapid/media.py
new file mode 100755
index 0000000..f4342c6
--- /dev/null
+++ b/rapid/media.py
@@ -0,0 +1,180 @@
+#!/usr/bin/python
+# -*- coding: latin1 -*-
+
+### Copyright (C) 2007 Damon Lynch <damonlynch@gmail.com>
+
+### This program is free software; you can redistribute it and/or modify
+### it under the terms of the GNU General Public License as published by
+### the Free Software Foundation; either version 2 of the License, or
+### (at your option) any later version.
+
+### This program is distributed in the hope that it will be useful,
+### but WITHOUT ANY WARRANTY; without even the implied warranty of
+### MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+### GNU General Public License for more details.
+
+### You should have received a copy of the GNU General Public License
+### along with this program; if not, write to the Free Software
+### Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+
+import os
+
+import config
+import common
+
+import operator
+
+def getDefaultPhotoLocation():
+ for default in config.DEFAULT_PHOTO_LOCATIONS:
+ path = common.getFullPath(default)
+ if os.path.isdir(path):
+ return path
+ return common.getFullPath('')
+
+def isImageMedia(path):
+ """ Returns true if directory specifies some media with photos on it """
+
+ if os.path.isdir(os.path.join(path, "DCIM")):
+ # is very likely a memory card, or something like that!
+ return True
+ else:
+ return False
+
+
+def isBackupMedia(path, identifier, writeable=True):
+ """ Test to see if path is used as a backup medium for storing images
+
+ If writeable is True, the directory must be writeable by the user """
+ suitable = False
+ if os.path.isdir(os.path.join(path, identifier)):
+ if writeable:
+ suitable = os.access(os.path.join(path, identifier), os.W_OK)
+ else:
+ suitable = True
+ return suitable
+
+def isImage(fileName):
+ ext = os.path.splitext(fileName)[1].lower()[1:]
+ return (ext in config.RAW_FILE_EXTENSIONS) or (ext in config.NON_RAW_IMAGE_FILE_EXTENSIONS)
+
+class Media:
+ """ Generic class for media holding images """
+ def __init__(self, path, volume = None):
+ """
+ volume is a gnomevfs volume
+ """
+
+ self.path = path
+ self.volume = volume
+
+
+ def prettyName(self, limit=config.MAX_LENGTH_DEVICE_NAME):
+ """
+ Returns a name for the media, useful for display.
+
+ If the media is from a gnomevfs volume, returns the gnome name.
+
+ Else. returns the last part of the mount point after stripping out
+ underscores.
+ """
+
+ if self.volume:
+ return self.volume.get_name(limit)
+ else:
+ name = os.path.split(self.path)[1]
+ name = name.replace('_', ' ')
+ v = name
+ if limit:
+ if len(v) > limit:
+ v = v[:limit] + '...'
+ return v
+
+ def getPath(self):
+ return self.path
+
+
+class CardMedia(Media):
+ """Compact Flash cards, etc."""
+ def __init__(self, path, volume = None, doNotScan=True):
+ """
+ volume is a gnomevfs volume
+ """
+ Media.__init__(self, path, volume)
+ if not doNotScan:
+ self.scanMedia()
+
+ def scanMedia(self):
+ """ creates a list of images on a path, recursively scanning
+
+ images are sorted by modification time"""
+
+ self.images = []
+ self.imageSizeSum = 0
+ for root, dirs, files in os.walk(self.path):
+ for name in files:
+ if isImage(name):
+ image = os.path.join(root, name)
+ size = os.path.getsize(image)
+ modificationTime = os.path.getmtime(image)
+ self.images.append((name, root, size, modificationTime),)
+ self.imageSizeSum += size
+ self.images.sort(key=operator.itemgetter(3))
+ self.noImages = len(self.images)
+
+ def setMedia(self, images, imageSizeSum, noImages):
+ self.images = images
+ self.imageSizeSum = imageSizeSum
+ self.noImages = noImages
+
+ def numberOfImages(self):
+ return self.noImages
+
+ def sizeOfImages(self, humanReadable = True):
+ if humanReadable:
+ return common.formatSizeForUser(self.imageSizeSum)
+ else:
+ return self.imageSizeSum
+
+ def firstImage(self):
+ if self.images:
+ return self.images[0]
+ else:
+ return None
+
+
+def scanForImageMedia(path):
+ """ returns a list of paths that contain images on media produced by a digital camera """
+
+ media = []
+ for i in os.listdir(path):
+ p = os.path.join(path, i)
+ if os.path.isdir(p):
+ if isImageMedia(p):
+ media.append(p)
+ return media
+
+def scanForBackupMedia(path, identifier):
+ """ returns a list of paths that contains backed up images """
+
+ media = []
+ for i in os.listdir(path):
+ p = os.path.join(path, i)
+ if os.path.isdir(p):
+ if isBackupMedia(p, identifier):
+ media.append(os.path.join(p, identifier))
+ return media
+
+
+if __name__ == '__main__':
+ print "Card media:"
+ for m in scanForImageMedia('/media'):
+ media = CardMedia(m)
+ print media.prettyName()
+ print media.numberOfImages()
+ print media.sizeOfImages()
+
+ print "\nBackup media:"
+ for m in scanForBackupMedia('/media', 'photos'):
+ print m
+
+ print "\nDefault download folder: ", getDefaultPhotoLocation()