def cleanup_list(wdir, skip_nzb):
""" Remove all files whose extension matches the cleanup list,
optionally ignoring the nzb extension """
if cfg.cleanup_list():
try:
files = os.listdir(wdir)
except:
files = ()
for file in files:
path = os.path.join(wdir, file)
if os.path.isdir(path):
cleanup_list(path, skip_nzb)
else:
if on_cleanup_list(file, skip_nzb):
try:
logging.info("Removing unwanted file %s", path)
os.remove(path)
except:
logging.error(Ta('Removing %s failed'), path)
logging.info("Traceback: ", exc_info = True)
if files:
try:
remove_dir(wdir)
except:
pass
def cleanup_list(wdir, skip_nzb):
""" Remove all files whose extension matches the cleanup list,
optionally ignoring the nzb extension
"""
if cfg.cleanup_list():
try:
files = os.listdir(wdir)
except:
files = ()
for filename in files:
path = os.path.join(wdir, filename)
if os.path.isdir(path):
cleanup_list(path, skip_nzb)
else:
if on_cleanup_list(filename, skip_nzb):
try:
logging.info("Removing unwanted file %s", path)
remove_file(path)
except:
logging.error(T('Removing %s failed'), clip_path(path))
logging.info("Traceback: ", exc_info=True)
if files:
try:
remove_dir(wdir)
except:
pass
def on_cleanup_list(filename, skip_nzb=False):
""" Return True if a filename matches the clean-up list
"""
lst = cfg.cleanup_list()
if lst:
ext = os.path.splitext(filename)[1].strip().strip('.').lower()
for k in lst:
item = k.strip().strip('.').lower()
if item == ext and not (skip_nzb and item == 'nzb'):
return True
return False
def on_cleanup_list(filename, skip_nzb=False):
""" Return True if a filename matches the clean-up list
"""
lst = cfg.cleanup_list()
if lst:
ext = os.path.splitext(filename)[1].strip().strip('.').lower()
for k in lst:
item = k.strip().strip('.').lower()
if item == ext and not (skip_nzb and item == 'nzb'):
return True
return False
def on_cleanup_list(filename, skip_nzb=False):
""" Return True if a filename matches the clean-up list """
lst = cfg.cleanup_list()
if lst:
name, ext = os.path.splitext(filename)
ext = ext.strip().lower()
name = name.strip()
for k in lst:
item = k.strip().strip('.').lower()
item = '.' + item
if (item == ext or (ext == '' and item == name)) and not (skip_nzb and item == '.nzb'):
return True
return False
def on_cleanup_list(filename, skip_nzb=False):
""" Return True if a filename matches the clean-up list """
lst = cfg.cleanup_list()
if lst:
name, ext = os.path.splitext(filename)
ext = ext.strip().lower()
name = name.strip()
for k in lst:
item = k.strip().strip(".").lower()
item = "." + item
if (item == ext or (ext == "" and item == name)) and not (skip_nzb and item == ".nzb"):
return True
return False