mediagoblin/media_types/image/__init__.py

   1 # GNU MediaGoblin -- federated, autonomous media hosting
   2 # Copyright (C) 2011, 2012 MediaGoblin contributors.  See AUTHORS.
   3 #
   4 # This program is free software: you can redistribute it and/or modify
   5 # it under the terms of the GNU Affero General Public License as published by
   6 # the Free Software Foundation, either version 3 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU Affero General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Affero General Public License
  15 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 import datetime
  17 import logging
  18
  19 from mediagoblin.media_types import MediaManagerBase
  20 from mediagoblin.media_types.image.processing import sniff_handler, \
  21         ImageProcessingManager
  22 from mediagoblin.tools import pluginapi
  23
  24 _log = logging.getLogger(__name__)
  25
  26
  27 ACCEPTED_EXTENSIONS = ["jpg", "jpeg", "png", "gif", "tiff"]
  28 MEDIA_TYPE = 'mediagoblin.media_types.image'
  29
  30
  31 def setup_plugin():
  32     config = pluginapi.get_config('mediagoblin.media_types.image')
  33
  34
  35 class ImageMediaManager(MediaManagerBase):
  36     human_readable = "Image"
  37     display_template = "mediagoblin/media_displays/image.html"
  38     default_thumb = "images/media_thumbs/image.png"
  39
  40     media_fetch_order = [u'medium', u'original', u'thumb']
  41
  42     def get_original_date(self):
  43         """
  44         Get the original date and time from the EXIF information. Returns
  45         either a datetime object or None (if anything goes wrong)
  46         """
  47         if not self.entry.media_data or not self.entry.media_data.exif_all:
  48             return None
  49
  50         try:
  51             # Try wrapped around all since exif_all might be none,
  52             # EXIF DateTimeOriginal or printable might not exist
  53             # or strptime might not be able to parse date correctly
  54             exif_date = self.entry.media_data.exif_all[
  55                 'EXIF DateTimeOriginal']['printable']
  56             original_date = datetime.datetime.strptime(
  57                 exif_date,
  58                 '%Y:%m:%d %H:%M:%S')
  59             return original_date
  60         except (KeyError, ValueError):
  61             return None
  62
  63
  64 def get_media_type_and_manager(ext):
  65     if ext in ACCEPTED_EXTENSIONS:
  66         return MEDIA_TYPE, ImageMediaManager
  67
  68
  69 hooks = {
  70     'setup': setup_plugin,
  71     'get_media_type_and_manager': get_media_type_and_manager,
  72     'sniff_handler': sniff_handler,
  73     ('media_manager', MEDIA_TYPE): lambda: ImageMediaManager,
  74     ('reprocess_manager', MEDIA_TYPE): lambda: ImageProcessingManager,
  75 }