import logging import os from PIL import Image import math import media import subprocess import time logger = media.logger __KEY_CAMERA_VENDOR__ = 'camera_vendor' __KEY_CAMERA_MODEL__ = 'camera_model' def get_audio_data(full_path): conv_key_dict = {} conv_key_dict['album'] = (str, media.KEY_ALBUM) conv_key_dict['TAG:album'] = (str, media.KEY_ALBUM) conv_key_dict['TAG:artist'] = (str, media.KEY_ARTIST) conv_key_dict['artist'] = (str, media.KEY_ARTIST) conv_key_dict['bit_rate'] = (__int_conv__, media.KEY_BITRATE) conv_key_dict['duration'] = (float, media.KEY_DURATION) conv_key_dict['TAG:genre'] = (str, media.KEY_GENRE) conv_key_dict['genre'] = (str, media.KEY_GENRE) conv_key_dict['TAG:title'] = (str, media.KEY_TITLE) conv_key_dict['title'] = (str, media.KEY_TITLE) conv_key_dict['TAG:track'] = (__int_conv__, media.KEY_TRACK) conv_key_dict['track'] = (__int_conv__, media.KEY_TRACK) conv_key_dict['TAG:date'] = (__int_conv__, media.KEY_YEAR) conv_key_dict['date'] = (__int_conv__, media.KEY_YEAR) return __adapt__data__(__get_xxprobe_data__(full_path, conv_key_dict), full_path) def get_video_data(full_path): conv_key_dict = {} conv_key_dict['creation_time'] = (__vid_datetime_conv__, media.KEY_TIME) conv_key_dict['TAG:creation_time'] = (__vid_datetime_conv__, media.KEY_TIME) conv_key_dict['bit_rate'] = (__int_conv__, media.KEY_BITRATE) conv_key_dict['duration'] = (float, media.KEY_DURATION) conv_key_dict['height'] = (__int_conv__, media.KEY_HEIGHT) conv_key_dict['width'] = (__int_conv__, media.KEY_WIDTH) conv_key_dict['display_aspect_ratio'] = (__ratio_conv__, media.KEY_RATIO) return __adapt__data__(__get_xxprobe_data__(full_path, conv_key_dict), full_path) def get_image_data(full_path): return __adapt__data__(__get_exif_data__(full_path), full_path) def __adapt__data__(data, full_path): data[media.KEY_SIZE] = os.path.getsize(full_path) # Join Camera Vendor and Camera Model if __KEY_CAMERA_MODEL__ in data and __KEY_CAMERA_VENDOR__ in data: model = data.pop(__KEY_CAMERA_MODEL__) vendor = data.pop(__KEY_CAMERA_VENDOR__) data[media.KEY_CAMERA] = '%s: %s' % (vendor, model) # Add time if not exists if media.KEY_TIME not in data: if media.KEY_YEAR in data and media.KEY_TRACK in data: if data[media.KEY_YEAR] != 0: # ignore year 0 - must be wrong # Use a date where track 1 is the newest in the given year minute = int(data[media.KEY_TRACK] / 60) second = (data[media.KEY_TRACK] - 60 * minute) % 60 # data[media.KEY_TIME] = int(time.mktime((data[media.KEY_YEAR], 1, 1, 0, 59 - minute, 59 - second, 0, 0, 0))) data[media.KEY_TIME_IS_SUBSTITUTION] = True else: data[media.KEY_TIME] = int(os.path.getmtime(full_path)) data[media.KEY_TIME_IS_SUBSTITUTION] = True return data def __get_xxprobe_data__(full_path, conv_key_dict): def _ffprobe_command(full_path): return ['ffprobe', '-v', 'quiet', '-show_format', '-show_streams', full_path] def _avprobe_command(full_path): return ['avprobe', '-v', 'quiet', '-show_format', '-show_streams', full_path] try: xxprobe_text = subprocess.check_output(_avprobe_command(full_path)) except FileNotFoundError: try: xxprobe_text = subprocess.check_output(_ffprobe_command(full_path)) except FileNotFoundError: logger.warning('ffprobe and avprobe seem to be not installed') return {} # rv = {} for line in xxprobe_text.decode('utf-8').splitlines(): try: key, val = [snippet.strip() for snippet in line.split('=')] except ValueError: continue else: if key in conv_key_dict: tp, name = conv_key_dict[key] try: rv[name] = tp(val) except ValueError: logger.log(logging.WARNING if val else logger.INFO, 'Can\'t convert %s (%s) for %s', repr(val), name, name) return rv def __get_exif_data__(full_path): rv = {} im = Image.open(full_path) try: exif = dict(im._getexif().items()) except AttributeError: logger.debug('%s does not have any exif information', full_path) else: conv_key_dict = {} # IMAGE conv_key_dict[0x9003] = (__datetime_conv__, media.KEY_TIME) conv_key_dict[0x8822] = (__exposure_program_conv__, media.KEY_EXPOSURE_PROGRAM) conv_key_dict[0x829A] = (__num_denum_conv__, media.KEY_EXPOSURE_TIME) conv_key_dict[0x9209] = (__flash_conv__, media.KEY_FLASH) conv_key_dict[0x829D] = (__num_denum_conv__, media.KEY_APERTURE) conv_key_dict[0x920A] = (__num_denum_conv__, media.KEY_FOCAL_LENGTH) conv_key_dict[0x8825] = (__gps_conv__, media.KEY_GPS) conv_key_dict[0xA003] = (__int_conv__, media.KEY_HEIGHT) conv_key_dict[0x8827] = (__int_conv__, media.KEY_ISO) conv_key_dict[0x010F] = (str, __KEY_CAMERA_VENDOR__) conv_key_dict[0x0110] = (str, __KEY_CAMERA_MODEL__) conv_key_dict[0x0112] = (__int_conv__, media.KEY_ORIENTATION) conv_key_dict[0xA002] = (__int_conv__, media.KEY_WIDTH) for key in conv_key_dict: if key in exif: tp, name = conv_key_dict[key] raw_value = exif[key] logger.debug("Converting %s out of %s", name, repr(raw_value)) value = tp(raw_value) if value is not None: rv[name] = value return rv # TODO: Join datetime converter __datetime_conv__ and __vid_datetime_conv_ def __datetime_conv__(dt): format_string = "%Y:%m:%d %H:%M:%S" return int(time.mktime(time.strptime(dt, format_string))) def __vid_datetime_conv__(dt): try: dt = dt[:dt.index('.')] except ValueError: pass # time string seems to have no '.' dt = dt.replace('T', ' ').replace('/', '').replace('\\', '') if len(dt) == 16: dt += ':00' format_string = '%Y-%m-%d %H:%M:%S' return int(time.mktime(time.strptime(dt, format_string))) def __exposure_program_conv__(n): return { 0: 'Unidentified', 1: 'Manual', 2: 'Program Normal', 3: 'Aperture Priority', 4: 'Shutter Priority', 5: 'Program Creative', 6: 'Program Action', 7: 'Portrait Mode', 8: 'Landscape Mode' }.get(n, None) def __flash_conv__(n): return { 0: 'No', 1: 'Fired', 5: 'Fired (?)', # no return sensed 7: 'Fired (!)', # return sensed 9: 'Fill Fired', 13: 'Fill Fired (?)', 15: 'Fill Fired (!)', 16: 'Off', 24: 'Auto Off', 25: 'Auto Fired', 29: 'Auto Fired (?)', 31: 'Auto Fired (!)', 32: 'Not Available' }.get(n, None) def __int_conv__(value): try: return int(value) except ValueError: for c in ['.', '/', '-']: p = value.find(c) if p >= 0: value = value[:p] try: return int(value) except ValueError: return None def __num_denum_conv__(data): try: return float(data) except TypeError: num, denum = data return num / denum def __gps_conv__(data): def lat_lon_cal(lon_or_lat): lon_lat = 0. fac = 1. for data in lon_or_lat: try: lon_lat += float(data[0]) / float(data[1]) * fac except TypeError: lon_lat += data * fac except ZeroDivisionError: return 0. fac *= 1. / 60. if math.isnan(lon_lat): return 0. return lon_lat try: lon = lat_lon_cal(data[0x0004]) lat = lat_lon_cal(data[0x0002]) if lon != 0 or lat != 0: # do not use lon and lat equal 0, caused by motorola gps weakness return {'lon': lon, 'lat': lat} except KeyError: logger.warning('GPS data extraction failed for %s', repr(data)) def __ratio_conv__(ratio): ratio = ratio.replace('\\', '') num, denum = ratio.split(':') return float(num) / float(denum)