import_video.py

#!/usr/bin/python

import argparse
import glob
import json
import os
import shutil
import subprocess
import time

from datetime import datetime

from utils import files


def copy_file(video_file, out_dir):
    """ Copy video file into %Y-%m-%d subfolder of the out_dir """

    cmnd = ['ffprobe', '-show_format', '-print_format', 'json', '-loglevel', 'quiet', '{0}'.format(video_file)]
    p = subprocess.Popen(cmnd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    out, _err = p.communicate()

    info = json.loads(out.decode('utf-8'))
    creation_time_found = False

    if 'format' in info and 'tags' in info['format']:
        tags = info['format']['tags']
        if 'creation_time' in tags:
            creation_time_found = True

            now_timestamp = time.time()
            utc_offset = datetime.fromtimestamp(now_timestamp) - datetime.utcfromtimestamp(now_timestamp)
            dt = datetime.strptime(tags['creation_time'], '%Y-%m-%dT%H:%M:%S.%fZ') + utc_offset

            dst_dir = os.path.join(out_dir, dt.strftime('%Y-%m-%d'))
            os.makedirs(dst_dir, exist_ok=True)

            dst_file = os.path.join(dst_dir, os.path.basename(video_file))
            if os.path.exists(dst_file):
                if files.equal(video_file, dst_file):
                    return
                if os.path.getsize(video_file) < os.path.getsize(dst_file):
                    return

            shutil.copy(video_file, dst_dir)

            imported.append(os.sep.join(
                [dt.strftime('%Y-%m-%d'), os.path.basename(dst_file)]))

    if not creation_time_found:
        print('[ERROR] Creation time not found in {0}'.format(video_file))


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="Import videos (*.mp4; *.mov; *.mpg) into specifed folder organizing them according to 'creation_time' information found in any of its media streams")

    parser.add_argument('--src-dir', type=str,
                        help='directory containing videos (*.mp4; *.mov; *.mpg)')
    parser.add_argument('--out-dir', type=str,
                        help='destination directory')
    parser.add_argument('--log-file', type=str,
                        help='output file for imported videos list')

    args = parser.parse_args()
    log_file = args.log_file
    src_dir = args.src_dir
    out_dir = args.out_dir

    imported = []

    for video_file in glob.glob(os.path.join(src_dir, '**/*.*'), recursive=True):
        if not video_file.lower().endswith('.mp4') and not video_file.lower().endswith('.mov') and not video_file.lower().endswith('.mpg'):
            continue

        copy_file(video_file, out_dir)

    with open(log_file, 'w', encoding='utf-8') as logfile:
        json.dump({'files': imported}, logfile, ensure_ascii=False, indent=4)