mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-01-08 04:00:03 -08:00
bfb45c180a
Remove version checks and update logic Remove extraneous constants: SOURCE_ROOT, SYS_ARGV, APP_FILENAME, CONFIG_MOVIE_FILE, MY_APP, CONFIG_TV_FILE, GIT_* Remove nzb2media.utils.processes Update requirements Flatten project structure Keep settings close to code Refactor NZBget, torrent configs, torrents, transcoder, tools, constants and forks Refactor `nzbToMedia.main` to `nzb2media.app.main` Fix flake/lint issues
175 lines
7.0 KiB
Python
175 lines
7.0 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
import os
|
|
import re
|
|
import shlex
|
|
import subprocess
|
|
from subprocess import DEVNULL
|
|
|
|
import nzb2media
|
|
import nzb2media.tool
|
|
from nzb2media.utils.files import list_media_files
|
|
|
|
log = logging.getLogger(__name__)
|
|
log.addHandler(logging.NullHandler())
|
|
reverse_list = [r'\.\d{2}e\d{2}s\.', r'\.[pi]0801\.', r'\.p027\.', r'\.[pi]675\.', r'\.[pi]084\.', r'\.p063\.', r'\b[45]62[xh]\.', r'\.yarulb\.', r'\.vtd[hp]\.', r'\.ld[.-]?bew\.', r'\.pir.?(dov|dvd|bew|db|rb)\.', r'\brdvd\.', r'\.vts\.', r'\.reneercs\.', r'\.dcv\.', r'\b(pir|mac)dh\b', r'\.reporp\.', r'\.kcaper\.', r'\.lanretni\.', r'\b3ca\b', r'\.cstn\.']
|
|
reverse_pattern = re.compile('|'.join(reverse_list), flags=re.IGNORECASE)
|
|
season_pattern = re.compile(r'(.*\.\d{2}e\d{2}s\.)(.*)', flags=re.IGNORECASE)
|
|
word_pattern = re.compile(r'([^A-Z0-9]*[A-Z0-9]+)')
|
|
media_list = [r'\.s\d{2}e\d{2}\.', r'\.1080[pi]\.', r'\.720p\.', r'\.576[pi]', r'\.480[pi]\.', r'\.360p\.', r'\.[xh]26[45]\b', r'\.bluray\.', r'\.[hp]dtv\.', r'\.web[.-]?dl\.', r'\.(vod|dvd|web|bd|br).?rip\.', r'\.dvdr\b', r'\.stv\.', r'\.screener\.', r'\.vcd\.', r'\bhd(cam|rip)\b', r'\.proper\.', r'\.repack\.', r'\.internal\.', r'\bac3\b', r'\.ntsc\.', r'\.pal\.', r'\.secam\.', r'\bdivx\b', r'\bxvid\b']
|
|
media_pattern = re.compile('|'.join(media_list), flags=re.IGNORECASE)
|
|
garbage_name = re.compile(r'^[a-zA-Z0-9]*$')
|
|
char_replace = [[r'(\w)1\.(\w)', r'\1i\2']]
|
|
|
|
|
|
def process_all_exceptions(name, dirname):
|
|
par2(dirname)
|
|
rename_script(dirname)
|
|
for filename in list_media_files(dirname):
|
|
newfilename = None
|
|
parent_dir = os.path.dirname(filename)
|
|
head, file_extension = os.path.splitext(os.path.basename(filename))
|
|
if reverse_pattern.search(head) is not None:
|
|
exception = reverse_filename
|
|
elif garbage_name.search(head) is not None:
|
|
exception = replace_filename
|
|
else:
|
|
exception = None
|
|
newfilename = filename
|
|
if not newfilename:
|
|
newfilename = exception(filename, parent_dir, name)
|
|
if nzb2media.GROUPS:
|
|
newfilename = strip_groups(newfilename)
|
|
if newfilename != filename:
|
|
rename_file(filename, newfilename)
|
|
|
|
|
|
def strip_groups(filename):
|
|
if not nzb2media.GROUPS:
|
|
return filename
|
|
dirname, file = os.path.split(filename)
|
|
head, file_extension = os.path.splitext(file)
|
|
newname = head.replace(' ', '.')
|
|
for group in nzb2media.GROUPS:
|
|
newname = newname.replace(group, '')
|
|
newname = newname.replace('[]', '')
|
|
newfile = newname + file_extension
|
|
newfile_path = os.path.join(dirname, newfile)
|
|
return newfile_path
|
|
|
|
|
|
def rename_file(filename, newfile_path):
|
|
if os.path.isfile(newfile_path):
|
|
newfile_path = os.path.splitext(newfile_path)[0] + '.NTM' + os.path.splitext(newfile_path)[1]
|
|
log.error(f'Replacing file name {filename} with download name {newfile_path}')
|
|
try:
|
|
os.rename(filename, newfile_path)
|
|
except Exception as error:
|
|
log.error(f'Unable to rename file due to: {error}')
|
|
|
|
|
|
def replace_filename(filename, dirname, name):
|
|
head, file_extension = os.path.splitext(os.path.basename(filename))
|
|
if media_pattern.search(os.path.basename(dirname).replace(' ', '.')) is not None:
|
|
newname = os.path.basename(dirname).replace(' ', '.')
|
|
log.debug(f'Replacing file name {head} with directory name {newname}')
|
|
elif media_pattern.search(name.replace(' ', '.').lower()) is not None:
|
|
newname = name.replace(' ', '.')
|
|
log.debug(f'Replacing file name {head} with download name {newname}')
|
|
else:
|
|
log.warning(f'No name replacement determined for {head}')
|
|
newname = name
|
|
newfile = newname + file_extension
|
|
newfile_path = os.path.join(dirname, newfile)
|
|
return newfile_path
|
|
|
|
|
|
def reverse_filename(filename, dirname):
|
|
head, file_extension = os.path.splitext(os.path.basename(filename))
|
|
na_parts = season_pattern.search(head)
|
|
if na_parts is not None:
|
|
match = word_pattern.findall(na_parts.group(2))
|
|
if match:
|
|
new_words = ''
|
|
for group in match:
|
|
if group[0] == '.':
|
|
new_words += '.'
|
|
new_words += re.sub(r'\W', '', group)
|
|
else:
|
|
new_words = na_parts.group(2)
|
|
for each_char in char_replace:
|
|
new_words = re.sub(each_char[0], each_char[1], new_words)
|
|
newname = new_words[::-1] + na_parts.group(1)[::-1]
|
|
else:
|
|
newname = head[::-1].title()
|
|
newname = newname.replace(' ', '.')
|
|
log.debug(f'Reversing filename {head} to {newname}')
|
|
newfile = newname + file_extension
|
|
newfile_path = os.path.join(dirname, newfile)
|
|
return newfile_path
|
|
|
|
|
|
def rename_script(dirname):
|
|
rename_file = ''
|
|
for directory, _, files in os.walk(dirname):
|
|
for file in files:
|
|
if re.search(r'(rename\S*\.(sh|bat)$)', file, re.IGNORECASE):
|
|
rename_file = os.path.join(directory, file)
|
|
dirname = directory
|
|
break
|
|
if rename_file:
|
|
with open(rename_file, encoding='utf-8') as fin:
|
|
rename_lines = [line.strip() for line in fin]
|
|
for line in rename_lines:
|
|
if re.search('^(mv|Move)', line, re.IGNORECASE):
|
|
cmd = shlex.split(line)[1:]
|
|
else:
|
|
continue
|
|
if len(cmd) == 2 and os.path.isfile(os.path.join(dirname, cmd[0])):
|
|
orig = os.path.join(dirname, cmd[0])
|
|
dest = os.path.join(dirname, cmd[1].split('\\')[-1].split('/')[-1])
|
|
if os.path.isfile(dest):
|
|
continue
|
|
log.debug(f'Renaming file {orig} to {dest}')
|
|
try:
|
|
os.rename(orig, dest)
|
|
except Exception as error:
|
|
log.error(f'Unable to rename file due to: {error}')
|
|
|
|
|
|
def par2(dirname):
|
|
sofar = 0
|
|
parfile = ''
|
|
objects = []
|
|
if os.path.exists(dirname):
|
|
objects = os.listdir(dirname)
|
|
for item in objects:
|
|
if item.endswith('.par2'):
|
|
size = os.path.getsize(os.path.join(dirname, item))
|
|
if size > sofar:
|
|
sofar = size
|
|
parfile = item
|
|
if nzb2media.tool.PAR2CMD and parfile:
|
|
pwd = os.getcwd() # Get our Present Working Directory
|
|
os.chdir(dirname) # set directory to run par on.
|
|
log.info(f'Running par2 on file {parfile}.')
|
|
command = [nzb2media.tool.PAR2CMD, 'r', parfile, '*']
|
|
cmd = ''
|
|
for item in command:
|
|
cmd = f'{cmd} {item}'
|
|
log.debug(f'calling command:{cmd}')
|
|
try:
|
|
with subprocess.Popen(command, stdout=DEVNULL, stderr=DEVNULL) as proc:
|
|
proc.communicate()
|
|
result = proc.returncode
|
|
except Exception:
|
|
log.error(f'par2 file processing for {parfile} has failed')
|
|
if not result:
|
|
log.info('par2 file processing succeeded')
|
|
os.chdir(pwd)
|
|
|
|
# dict for custom groups
|
|
# we can add more to this list
|
|
# _customgroups = {'Q o Q': process_qoq, '-ECI': process_eci}
|