media/metadata.py

import logging
import os
from PIL import Image
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]
                value = tp(exif[key])
                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]
    return int(value)


def __num_denum_conv__(data):
    num, denum = data
    return num / denum


def __gps_conv__(data):
    def lat_lon_cal(lon_or_lat):
        lon_lat = 0.
        fac = 1.
        for num, denum in lon_or_lat:
            lon_lat += float(num) / float(denum) * fac
            fac *= 1. / 60.
        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)