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 ProcessImage, \
  21     sniff_handler
  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     processor = ProcessImage
  38     display_template = "mediagoblin/media_displays/image.html"
  39     default_thumb = "images/media_thumbs/image.png"
  40
  41     media_fetch_order = [u'medium', u'original', u'thumb']
  42
  43     def get_original_date(self):
  44         """
  45         Get the original date and time from the EXIF information. Returns
  46         either a datetime object or None (if anything goes wrong)
  47         """
  48         if not self.entry.media_data or not self.entry.media_data.exif_all:
  49             return None
  50
  51         try:
  52             # Try wrapped around all since exif_all might be none,
  53             # EXIF DateTimeOriginal or printable might not exist
  54             # or strptime might not be able to parse date correctly
  55             exif_date = self.entry.media_data.exif_all[
  56                 'EXIF DateTimeOriginal']['printable']
  57             original_date = datetime.datetime.strptime(
  58                 exif_date,
  59                 '%Y:%m:%d %H:%M:%S')
  60             return original_date
  61         except (KeyError, ValueError):
  62             return None
  63
  64
  65 def get_media_type_and_manager(ext):
  66     if ext in ACCEPTED_EXTENSIONS:
  67         return MEDIA_TYPE, ImageMediaManager
  68
  69
  70 hooks = {
  71     'setup': setup_plugin,
  72     'get_media_type_and_manager': get_media_type_and_manager,
  73     'sniff_handler': sniff_handler,
  74     ('media_manager', MEDIA_TYPE): lambda: ImageMediaManager,
  75     ('reprocess_action', 'image'): ProcessImage().reprocess_action,
  76     ('media_reprocess', 'image'): ProcessImage().media_reprocess,
  77 }