Read all metadata at once.

Make _read_image no-op if argument is an image.
Add gathering metadata function that returns a dict
This commit is contained in:
Glandos
2018-12-13 22:37:13 +01:00
committed by Simon Conseil
parent 972f0f167d
commit 89e3edf2f5
2 changed files with 48 additions and 3 deletions

View File

@@ -41,7 +41,7 @@ from click import get_terminal_size, progressbar
from . import image, signals, video
from .image import (get_exif_data, get_exif_tags, get_iptc_data, get_size,
process_image)
process_image, get_image_metadata)
from .settings import get_thumb
from .utils import (Devnull, cached_property, check_or_create_dir, copy,
get_mime, is_valid_html5_video, read_markdown,
@@ -156,6 +156,8 @@ class Media:
self.description = ''
"""Description extracted from the Markdown <imagename>.md file."""
self.file_metadata = {}
self.title = ''
"""Title extracted from the Markdown <imagename>.md file."""
@@ -196,6 +198,8 @@ class Image(Media):
def _get_metadata(self):
super()._get_metadata()
src_metadata = get_image_metadata(self.src_path)
self.file_metadata[self.src_path] = src_metadata
# If a title or description hasn't been obtained by other means, look
# for the information in IPTC fields
if self.title and self.description:
@@ -203,7 +207,7 @@ class Image(Media):
return
try:
iptc_data = get_iptc_data(self.src_path)
iptc_data = src_metadata.get('iptc', {})
except Exception as e:
self.logger.warning('Could not read IPTC data from %s: %s',
self.src_path, e)
@@ -217,7 +221,7 @@ class Image(Media):
def raw_exif(self):
"""If not `None`, contains the raw EXIF tags."""
try:
return (get_exif_data(self.src_path)
return (self.file_metadata[self.src_path]['exif']
if self.ext in ('.jpg', '.jpeg') else None)
except Exception as e:
self.logger.warning('Could not read EXIF data from %s: %s',

View File

@@ -56,6 +56,10 @@ def _has_exif_tags(img):
def _read_image(file_path):
# The image is already opened
if isinstance(file_path, PILImage.Image):
return file_path
with warnings.catch_warnings(record=True) as caught_warnings:
im = PILImage.open(file_path)
@@ -271,6 +275,43 @@ def get_iptc_data(filename):
return iptc_data
def get_image_metadata(filename):
logger = logging.getLogger(__name__)
try:
img = _read_image(filename)
except Exception as e:
logger.error('Could not open image %s metadata: %s',
filename, e)
exif = None
iptc = None
size = None
try:
exif = get_exif_data(img)
except Exception as e:
logger.warning('Could not read EXIF data from %s: %s',
filename, e)
try:
iptc = get_iptc_data(img)
except Exception as e:
logger.warning('Could not read IPTC data from %s: %s',
filename, e)
try:
size = get_size(img)
except Exception as e:
logger.warning('Could not read size from %s: %s',
filename, e)
return {
'exif': exif,
'iptc': iptc,
'size': size,
}
def dms_to_degrees(v):
"""Convert degree/minute/second to decimal degrees."""