ordigi/elodie/media/photo.py

100 lines
3.1 KiB
Python
Raw Normal View History

2015-10-07 10:48:01 +02:00
"""
2016-01-08 23:49:06 +01:00
The photo module contains the :class:`Photo` class, which is used to track
image objects (JPG, DNG, etc.).
.. moduleauthor:: Jaisen Mathai <jaisen@jmathai.com>
2015-10-07 10:48:01 +02:00
"""
2016-03-12 20:09:28 +01:00
from __future__ import print_function
from __future__ import absolute_import
2015-10-07 10:48:01 +02:00
import imghdr
2015-10-07 10:48:01 +02:00
import os
import re
import time
from datetime import datetime
from re import compile
2015-10-07 10:48:01 +02:00
2016-11-09 07:41:00 +01:00
from elodie import log
2016-03-12 20:09:28 +01:00
from .media import Media
2015-10-07 10:48:01 +02:00
2015-10-07 10:48:01 +02:00
class Photo(Media):
2016-01-08 23:49:06 +01:00
"""A photo object.
:param str source: The fully qualified path to the photo file
"""
__name__ = 'Photo'
2016-01-08 23:49:06 +01:00
#: Valid extensions for photo files.
2016-09-05 01:10:30 +02:00
extensions = ('arw', 'cr2', 'dng', 'gif', 'jpeg', 'jpg', 'nef', 'rw2')
2015-10-07 10:48:01 +02:00
def __init__(self, source=None):
super(Photo, self).__init__(source)
# We only want to parse EXIF once so we store it here
self.exif = None
def get_date_taken(self):
2016-01-08 23:49:06 +01:00
"""Get the date which the photo was taken.
The date value returned is defined by the min() of mtime and ctime.
:returns: time object or None for non-photo files or 0 timestamp
"""
if(not self.is_valid()):
return None
2016-06-24 06:31:58 +02:00
source = self.source
seconds_since_epoch = min(os.path.getmtime(source), os.path.getctime(source)) # noqa
exif = self.get_exiftool_attributes()
if not exif:
return seconds_since_epoch
# We need to parse a string from EXIF into a timestamp.
# EXIF DateTimeOriginal and EXIF DateTime are both stored
# in %Y:%m:%d %H:%M:%S format
2016-06-24 06:31:58 +02:00
# we split on a space and then r':|-' -> convert to int -> .timetuple()
# the conversion in the local timezone
# EXIF DateTime is already stored as a timestamp
# Sourced from https://github.com/photo/frontend/blob/master/src/libraries/models/Photo.php#L500 # noqa
for key in self.exif_map['date_taken']:
try:
if(key in exif):
if(re.match('\d{4}(-|:)\d{2}(-|:)\d{2}', exif[key]) is not None): # noqa
dt, tm = exif[key].split(' ')
dt_list = compile(r'-|:').split(dt)
dt_list = dt_list + compile(r'-|:').split(tm)
dt_list = map(int, dt_list)
time_tuple = datetime(*dt_list).timetuple()
seconds_since_epoch = time.mktime(time_tuple)
break
except BaseException as e:
2016-11-09 07:41:00 +01:00
log.error(e)
pass
if(seconds_since_epoch == 0):
return None
return time.gmtime(seconds_since_epoch)
def is_valid(self):
2016-01-08 23:49:06 +01:00
"""Check the file extension against valid file extensions.
The list of valid file extensions come from self.extensions. This
also checks whether the file is an image.
:returns: bool
"""
source = self.source
# gh-4 This checks if the source file is an image.
# It doesn't validate against the list of supported types.
if(imghdr.what(source) is None):
return False
return os.path.splitext(source)[1][1:].lower() in self.extensions