# Selective cleanup (deletion) of files (based on category and extension)
# For use by a SABnzbd+ external post-processing script
# Version: 1.05
# Date: 2009/07/19
# License: As-is; public domain
# Requirements: Python 3.1, SABnzbd+ 0.4.11
# Description: This script clean's up (deletes) files with specific extensions, e.g. .sfv, .nzb, but only for downloads belonging to a particular category.
# Remarks:
# The cleanup is performed only in the job directory. Its subdirectories, if any, are not affected.
# Extensions are case-insensitive.
# Usage syntax:
# C:\Python31\python.exe selective_cleanup.py job_directory job_category category_specified ext1 ext2 ... extLast
# Usage examples:
# C:\Python31\python.exe "D:\SABnzbd scripts\selective_cleanup.py" %1 %5 movies sfv
# C:\Python31\python.exe "D:\SABnzbd scripts\selective_cleanup.py" %1 %5 "movies (hd)" sfv nzb
# Keywords:
# sabnzbd+, sabnzbd, post-processing, post-processing script,
# delete, deletion, file deletion, extension, file cleanup, cleanup list
import os, sys
# Parse input arguments
job_dir = sys.argv[1]
job_cat = sys.argv[2]
cleanup_cat = sys.argv[3]
exts = sys.argv[4:]
exts = [ext.lower() for ext in exts]
os.chdir(job_dir)
# Selectively delete files
if job_cat == cleanup_cat:
files = [i for i in os.listdir(job_dir) if os.path.isfile(i)]
files = [f for f in files if os.path.splitext(f)[1][1:].lower() in exts]
for f in files:
os.remove(f)
print('Deleted {}'.format(f))
