mirror of
https://github.com/sashacmc/photo-importer.git
synced 2024-11-24 08:02:14 +02:00
105 lines
3.4 KiB
Python
Executable File
105 lines
3.4 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
import os
|
|
import shutil
|
|
import logging
|
|
|
|
import fileprop
|
|
|
|
|
|
class Mover(object):
|
|
OUT_SUBDIR_CFG = {
|
|
fileprop.IMAGE: 'out_subdir_image',
|
|
fileprop.VIDEO: 'out_subdir_video',
|
|
fileprop.AUDIO: 'out_subdir_audio',
|
|
}
|
|
|
|
def __init__(self, config, input_path, output_path, filenames, dryrun):
|
|
self.__config = config
|
|
self.__input_path = input_path
|
|
self.__output_path = output_path
|
|
self.__filenames = filenames
|
|
self.__dryrun = dryrun
|
|
self.__move_mode = int(config['main']['move_mode'])
|
|
self.__remove_garbage = int(config['main']['remove_garbage'])
|
|
self.__stat = {'total': len(filenames)}
|
|
self.__file_prop = fileprop.FileProp(self.__config)
|
|
|
|
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:
|
|
prop = self.__file_prop.get(fname)
|
|
new_fname = self.__move_file(fname, prop)
|
|
if new_fname:
|
|
res.append((fname, new_fname, prop))
|
|
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):
|
|
if prop.type() == fileprop.GARBAGE:
|
|
if self.__remove_garbage:
|
|
if not self.__dryrun:
|
|
os.remove(fname)
|
|
logging.info('removed "%s"' % fname)
|
|
self.__stat['removed'] += 1
|
|
else:
|
|
self.__stat['skipped'] += 1
|
|
return None
|
|
|
|
if prop.type() == fileprop.IGNORE 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):
|
|
if not self.__dryrun:
|
|
os.makedirs(path)
|
|
logging.info('dir "%s" created' % path)
|
|
|
|
fullname = prop.out_name_full(path)
|
|
if self.__move_mode:
|
|
if not self.__dryrun:
|
|
shutil.move(fname, fullname)
|
|
logging.info('"%s" moved "%s"' % (fname, fullname))
|
|
self.__stat['moved'] += 1
|
|
else:
|
|
if not self.__dryrun:
|
|
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()
|
|
if not self.__dryrun:
|
|
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
|