photo-importer/mover.py

#!/usr/bin/python3

import os
import shutil
import logging

import fileprop


class Mover(object):
    OUT_SUBDIR_CFG = {
        fileprop.FileProp.IMAGE: 'out_subdir_image',
        fileprop.FileProp.VIDEO: 'out_subdir_video',
        fileprop.FileProp.AUDIO: 'out_subdir_audio',
    }

    def __init__(self, config, input_path, output_path, filenames):
        self.__config = config
        self.__input_path = input_path
        self.__output_path = output_path
        self.__filenames = filenames
        self.__move_mode = int(config['main']['move_mode'])
        self.__remove_garbage = int(config['main']['remove_garbage'])
        self.__stat = {'total': len(filenames)}

    def run(self):
        self.__stat['moved'] = 0
        self.__stat['copied'] = 0
        self.__stat['removed'] = 0
        self.__stat['skipped'] = 0
        self.__stat['processed'] = 0
        self.__stat['errors'] = 0
        res = []
        for fname in self.__filenames:
            try:
                new_fname = self.__move_file(fname)
                if new_fname:
                    res.append(new_fname)
            except Exception as ex:
                logging.error('Move files exception: %s' % ex)
                self.__stat['errors'] += 1

            self.__stat['processed'] += 1
        return res

    def __move_file(self, fname):
        prop = fileprop.FileProp(self.__config, fname)

        if prop.type() == prop.GARBAGE:
            if self.__remove_garbage:
                os.remove(fname)
                logging.info('removed "%s"' % fname)
                self.__stat['removed'] += 1
            else:
                self.__stat['skipped'] += 1
            return None

        if prop.type() == prop.OTHER or prop.time() is None:
            self.__stat['skipped'] += 1
            return None

        if self.__output_path:
            type_subdir = \
                self.__config['main'][self.OUT_SUBDIR_CFG[prop.type()]]

            date_subdir = prop.time().strftime(
                self.__config['main']['out_date_format'])

            path = os.path.join(self.__output_path, type_subdir, date_subdir)
            if not os.path.isdir(path):
                os.makedirs(path)

            fullname = prop.out_name_full(path)
            if self.__move_mode:
                shutil.move(fname, fullname)
                logging.info('"%s" moved "%s"' % (fname, fullname))
                self.__stat['moved'] += 1
            else:
                shutil.copy2(fname, fullname)
                logging.info('"%s" copied "%s"' % (fname, fullname))
                self.__stat['copied'] += 1

            return fullname
        else:
            if prop.ok():
                self.__stat['skipped'] += 1
                return None
            else:
                new_fname = prop.out_name_full()
                os.rename(fname, new_fname)
                logging.info('"%s" renamed "%s"' % (fname, new_fname))
                self.__stat['moved'] += 1
                return new_fname

    def status(self):
        return self.__stat