mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2024-11-14 09:30:21 -08:00
121 lines
3.4 KiB
Python
121 lines
3.4 KiB
Python
from __future__ import (
|
|
absolute_import,
|
|
division,
|
|
print_function,
|
|
unicode_literals,
|
|
)
|
|
|
|
import os.path
|
|
|
|
from six import text_type
|
|
|
|
import core
|
|
from core import logger
|
|
from core.utils.files import list_media_files, move_file
|
|
from core.utils.paths import clean_directory, flatten_dir
|
|
|
|
|
|
def flatten(output_destination):
|
|
return flatten_dir(output_destination, list_media_files(output_destination))
|
|
|
|
|
|
def clean_dir(path, section, subsection):
|
|
cfg = dict(core.CFG[section][subsection])
|
|
min_size = int(cfg.get('minSize', 0))
|
|
delete_ignored = int(cfg.get('delete_ignored', 0))
|
|
try:
|
|
files = list_media_files(path, min_size=min_size, delete_ignored=delete_ignored)
|
|
except Exception:
|
|
files = []
|
|
return clean_directory(path, files)
|
|
|
|
|
|
def process_dir(path, link):
|
|
folders = []
|
|
|
|
logger.info('Searching {0} for mediafiles to post-process ...'.format(path))
|
|
dir_contents = os.listdir(text_type(path))
|
|
|
|
# search for single files and move them into their own folder for post-processing
|
|
|
|
# Generate list of sync files
|
|
sync_files = (
|
|
item for item in dir_contents
|
|
if os.path.splitext(item)[1] in ['.!sync', '.bts']
|
|
)
|
|
|
|
# Generate a list of file paths
|
|
filepaths = (
|
|
os.path.join(path, item) for item in dir_contents
|
|
if item not in ['Thumbs.db', 'thumbs.db']
|
|
)
|
|
|
|
# Generate a list of media files
|
|
mediafiles = (
|
|
item for item in filepaths
|
|
if os.path.isfile(item)
|
|
)
|
|
|
|
if any(sync_files):
|
|
logger.info('')
|
|
else:
|
|
for mediafile in mediafiles:
|
|
try:
|
|
move_file(mediafile, path, link)
|
|
except Exception as e:
|
|
logger.error('Failed to move {0} to its own directory: {1}'.format(os.path.split(mediafile)[1], e))
|
|
|
|
# removeEmptyFolders(path, removeRoot=False)
|
|
|
|
# Generate all path contents
|
|
path_contents = (
|
|
os.path.join(path, item)
|
|
for item in os.listdir(text_type(path))
|
|
)
|
|
|
|
# Generate all directories from path contents
|
|
directories = (
|
|
path for path in path_contents
|
|
if os.path.isdir(path)
|
|
)
|
|
|
|
for directory in directories:
|
|
dir_contents = os.listdir(directory)
|
|
sync_files = (
|
|
item for item in dir_contents
|
|
if os.path.splitext(item)[1] in ['.!sync', '.bts']
|
|
)
|
|
if not any(dir_contents) or any(sync_files):
|
|
continue
|
|
folders.append(directory)
|
|
|
|
return folders
|
|
|
|
|
|
def get_dirs(section, subsection, link='hard'):
|
|
to_return = []
|
|
|
|
watch_directory = core.CFG[section][subsection]['watch_dir']
|
|
directory = os.path.join(watch_directory, subsection)
|
|
|
|
if not os.path.exists(directory):
|
|
directory = watch_directory
|
|
|
|
try:
|
|
to_return.extend(process_dir(directory, link))
|
|
except Exception as e:
|
|
logger.error('Failed to add directories from {0} for post-processing: {1}'.format(watch_directory, e))
|
|
|
|
if core.USE_LINK == 'move':
|
|
try:
|
|
output_directory = os.path.join(core.OUTPUT_DIRECTORY, subsection)
|
|
if os.path.exists(output_directory):
|
|
to_return.extend(process_dir(output_directory, link))
|
|
except Exception as e:
|
|
logger.error('Failed to add directories from {0} for post-processing: {1}'.format(core.OUTPUT_DIRECTORY, e))
|
|
|
|
if not to_return:
|
|
logger.debug('No directories identified in {0}:{1} for post-processing'.format(section, subsection))
|
|
|
|
return list(set(to_return))
|