github/nzbToMedia
1
0
Fork 0
mirror of https://github.com/clinton-hall/nzbToMedia.git synced 2025-08-22 06:13:19 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

add windows builder files for nzbget

Browse source
This commit is contained in:
clinton-hall 2013-03-21 19:46:35 +10:30
commit 61df083863
5 changed files with 769 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
nzbget-postprocessing-files/windows-builder/README.md Normal file
View file

@ -0,0 +1,9 @@
To compile for windows, add the following files into the same folder.
Compile with pyinstaller using options -F -w
Transcoder.py
Linktastic\linktastic.py folder (not used, but imported in nzbToMediaUtil.py)
nzbToMediaUtil.py
migratecfg.py
nzbToMediaEnv.py
nzbToMediaSceneExceptions.py
Autoprocess files

105
nzbget-postprocessing-files/windows-builder/autoProcessMedia.cfg Normal file
View file

@ -0,0 +1,105 @@
[CouchPotato]
cpsCategory = movie
apikey = ac8a2260b147487c9c88a00fc9dde601
host = localhost
port = 5050
username = clinton
password = dlo3822
ssl = 0
web_root =
delay = 65
method = renamer
delete_failed = 0
[SickBeard]
sbCategory = TV
host = localhost
port = 8081
username =
password =
web_root =
ssl = 0
watch_dir = C:\TV\new
failed_fork = 0
[HeadPhones]
hpCategory = music
apikey =
host = localhost
port = 8181
username =
password =
ssl = 0
web_root =
delay = 65
[Mylar]
mlCategory = comics
host = localhost
port = 8090
username =
password =
web_root =
ssl = 0
[Gamez]
gzCategory = movie
apikey =
host = localhost
port = 8085
username =
password =
ssl = 0
web_root =
[Torrent]
clientAgent = utorrent
useLink = 1
minSampleSize = 200
outputDirectory = C:\TV
categories = music_videos,pictures,software
uTorrentWEBui = http://localhost:8090/gui/
uTorrentUSR = your username
uTorrentPWD = your password
deleteOriginal = 0
[Extensions]
compressedExtensions = .zip,.rar,.7z,.gz,.bz,.tar,.arj
mediaExtensions = .mkv,.avi,.divx,.xvid,.mov,.wmv,.mp4,.mpg,.mpeg,.vob,.iso
metaExtensions = .nfo,.sub,.srt,.jpg,.gif
[Transcoder]
transcode = 0
duplicate = 1
ignoreExtensions = .avi
outputVideoExtension = .mp4
outputVideoCodec = libx264
outputVideoFramerate = 24
outputVideoBitrate = 800k
outputAudioCodec = libmp3lame
outputAudioBitrate = 128k
[loggers]
keys = root
[handlers]
keys = console
[formatters]
keys = generic
[logger_root]
level = NOTSET
handlers = console
qualname =
[handler_console]
class = StreamHandler
args = (sys.stdout,)
level = INFO
formatter = generic
[formatter_generic]
format = %(asctime)s|%(levelname)-7.7s %(message)s
datefmt = %H:%M:%S

23
nzbget-postprocessing-files/windows-builder/logging.cfg Normal file
View file

@ -0,0 +1,23 @@
# Logging configuration
[loggers]
keys = root
[handlers]
keys = console
[formatters]
keys = generic
[logger_root]
level = NOTSET
handlers = console
qualname =
[handler_console]
class = StreamHandler
args = (sys.stdout,)
level = INFO
formatter = generic
[formatter_generic]
format = [%(levelname)-7.7s] %(message)s

144
nzbget-postprocessing-files/windows-builder/nzbget-postprocess.conf Normal file
View file

@ -0,0 +1,144 @@
#
# This file if part of nzbget
#
# Template configuration file for post-processing script "nzbget-postprocess.sh".
# Please refer to "nzbget-postprocess.sh" for usage instructions.
#
# Copyright (C) 2008-2013 Andrey Prygunkov <hugbug@users.sourceforge.net>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
#
##############################################################################
### PATHS ###
# Set the full path to location to move failed downloads to.
Failed_Directory=/usr/local/downloads/failed
##############################################################################
### OPTIONS ###
# Rename img-files to iso (yes, no).
RenameIMG=yes
# Joint TS-files (yes, no).
JoinTS=yes
# Perform SickBeard's postprocessing (yes, no).
SickBeard=yes
# Category for SickBeard's postprocessing.
SickBeardCategory=tv
# Perform Couchpotato's postprocessing (yes, no).
CouchPotato=yes
# Category for Couchpotato's postprocessing.
CouchPotatoCategory=movies
# Perform HeadPhones' postprocessing (yes, no).
HeadPhones=yes
# Category for HeadPhones' postprocessing.
HeadPhonesCategory=music
# Perform Mylar's postprocessing (yes, no).
Mylar=yes
# Category for Mylar's postprocessing.
MylarCategory=comics
# Perform Gamez's postprocessing (yes, no).
Gamez=yes
# Category for Gamez's postprocessing.
GamezCategory=games
# Clean up list. space seperated, in single quotes, default '.nzb .sfv .1'
FileCleanUp='.nzb .sfv .1'
# Remove all files when download fails (yes, no).
Delete_Failed=no
# Toggle detailed output (yes, no).
Debug=no
##############################################################################
### POSTPROCESSING-PARAMETERS ###
# This section defines parameters, which can be set for each nzb-file
# individually using either web-interface or command line.
# Example command line for setting parameter "PostProcess" to value "no" for
# nzb-file with id=2:
# nzbget -E G O PostProcess=no 2
# Perform postprocessing (yes, no).
#
# Set to "no" to skip postprocessing for this nzb-file.
PostProcess=yes
# Destination directory.
DestDir=
##############################################################################
### EMAIL-PARAMETERS ###
# This section defines parameters, which can be set to allow notifications to
# be sent via email.
# This uses sendEmail as authored by Brandon Zehm <caspian@dotconf.net>
# List Categories for successful download email, default 'tv movies'
Email_successful='tv movies music comics games pictures'
# List Categories for failed download email, default 'tv movies'
Email_failed='tv movies music comics games pictures'
# Set the full path and file name for sendEmail application.
sendEmail=/usr/local/nzbget/var/sendEmail/sendEmail
# Enter the email address you want this email to be sent from.
Email_From=
# Enter the email address you want this email to be sent to.
Email_To=
# Enter smtp server and port. eg smtp.live.com:25
Email_Server=
# Server uses tsl (auto, yes, no).
Tsl=auto
# Enter your smtp server user name (if required)
Email_User=
# Enter your smtp server password (if required)
Email_Pass=
# Enter your email subject, in single quotes.
# Use the following tags between the less than and greater than symbols.
# Use status to add 'completed'/'failed'.
# Use name to add the nzb name.
# Use cat to add the download categoty.
# Use script to name the external script used.
Email_Subject='The download of <name> has <status>.'
# Enter your email message, in single quotes.
# Use the same substitutes as described above.
# Use \r\n for new line.
Email_Message='The download of <name> has <status>. \r\n This has been processed by the script <script> for category <cat>.'
# Append the potprocessing log to the email (yes, no).
Add_Log=no

488
nzbget-postprocessing-files/windows-builder/nzbget-postprocess.py Normal file
View file

@ -0,0 +1,488 @@
#!/usr/bin/env python
#
# This file if part of nzbget
#
# Example postprocessing script for NZBGet
#
# Copyright (C) 2008 Peter Roubos <peterroubos@hotmail.com>
# Copyright (C) 2008 Otmar Werner
# Copyright (C) 2008-2013 Andrey Prygunkov <hugbug@users.sourceforge.net>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
#
####################### Usage instructions #######################
# o Script will cleanup, join ts-files and rename img-files to iso.
#
# o To use this script with nzbget set the option "PostProcess" in
# nzbget configuration file to point to this script file. E.g.:
# PostProcess=/home/user/nzbget/nzbget-postprocess.sh
#
# o The script needs a configuration file. An example configuration file
# is provided in file "nzbget-postprocess.conf". Put the configuration file
# into the directory where nzbget's configuration file (nzbget.conf) is located.
# Then edit the configuration file in any text editor to adjust the settings.
#
# o You can also edit the script's configuration via web-interface.
#
# o There are few options, which can be ajdusted for each nzb-file individually.
#
####################### End of Usage instructions #######################
import os
import logging
import logging.config
import sys
import shutil
from configobj import ConfigObj
from subprocess import call
from glob import iglob
import migratecfg
import autoProcessComics
import autoProcessGames
import autoProcessMusic
import autoProcessTV
import autoProcessMovie
# Postprocessing function for nzbToCouchPotato and nzbToSickBeard
def nzbToMedia(nzbStatus):
script = ""
result = ""
if Debug == "yes":
Logger.debug("Post-Process: Executing external postprocessing with argument %s", nzbStatus)
PostProcessStatus = nzbStatus
# 200 MB in bytes
SIZE_CUTOFF = 200 * 1024 * 1024
# Ignore 'sample' in files unless 'sample' in Torrent Name
for dirpath, dirnames, filenames in os.walk(NZBPP_DIRECTORY):
for file in filenames:
filePath = os.path.join(dirpath, file)
if ('sample' in filePath.lower()) and (not 'sample' in NZBPP_NZBNAME) and (os.path.getsize(filePath) < SIZE_CUTOFF):
Logger.info("Post-Process: Deleting sample file %s", filePath)
os.unlink(filePath)
if NZBPP_CATEGORY == CouchPotatoCategory:
if CouchPotato == "yes":
script = "autoProcessMovie"
# Call Couchpotato's postprocessing script
Logger.info("Post-Process: Running CouchPotato's postprocessing script")
if Debug == "yes":
Logger.debug("Post-Process: CouchPotato-Script-ARGV1= %s", NZBPP_DIRECTORY)
Logger.debug("Post-Process: CouchPotato-Script-ARGV2= %s", NZBPP_NZBFILENAME)
Logger.debug("Post-Process: CouchPotato-Script-ARGV3= %s", PostProcessStatus)
result = autoProcessMovie.process(NZBPP_DIRECTORY, NZBPP_NZBFILENAME, PostProcessStatus)
else:
Logger.debug("Post-Process: Ignored to run CouchPotato's postprocessing script as it is disabled by user")
if NZBPP_CATEGORY == SickBeardCategory:
if SickBeard == "yes":
script = "autoProcessTv"
# Call SickBeard's postprocessing script
Logger.info("Post-Process: Running SickBeard's postprocessing script")
if Debug == "yes":
Logger.debug("Post-Process: SickBeard-Script-ARGV1= %s", NZBPP_DIRECTORY)
Logger.debug("Post-Process: SickBeard-Script-ARGV2= %s", NZBPP_NZBFILENAME)
Logger.debug("Post-Process: SickBeard-Script-ARGV3= %s", PostProcessStatus)
result = autoProcessTV.processEpisode(NZBPP_DIRECTORY, NZBPP_NZBFILENAME, PostProcessStatus)
else:
Logger.debug("Post-Process: Ignored to run SickBeard's postprocessing script as it is disabled by user")
if NZBPP_CATEGORY == HeadPhonesCategory:
if HeadPhones == "yes":
script = "autoProcessMusic"
# Call HeadPhones' postprocessing script
Logger.info("Post-Process: Running HeadPhones' postprocessing script")
if Debug == "yes":
Logger.debug("Post-Process: HeadPhones-Script-ARGV1= %s", NZBPP_DIRECTORY)
Logger.debug("Post-Process: HeadPhones-Script-ARGV2= %s", NZBPP_NZBFILENAME)
Logger.debug("Post-Process: HeadPhones-Script-ARGV3= %s", PostProcessStatus)
result = autoProcessMusic.process(NZBPP_DIRECTORY, NZBPP_NZBFILENAME, PostProcessStatus)
else:
Logger.debug("Post-Process: Ignored to run HeadPhones' postprocessing script as it is disabled by user")
if NZBPP_CATEGORY == MylarCategory:
if Mylar == "yes":
script = "autoProcessComics"
# Call Mylar's postprocessing script
Logger.info("Post-Process: Running Mylar's postprocessing script")
if Debug == "yes":
Logger.debug("Post-Process: Mylar-Script-ARGV1= %s", NZBPP_DIRECTORY)
Logger.debug("Post-Process: Mylar-Script-ARGV2= %s", NZBPP_NZBFILENAME)
Logger.debug("Post-Process: Mylar-Script-ARGV3= %s", PostProcessStatus)
result = autoProcessComics.processEpisode(NZBPP_DIRECTORY, NZBPP_NZBFILENAME, PostProcessStatus)
else:
Logger.debug("Post-Process: Ignored to run Mylar's postprocessing script as it is disabled by user")
if NZBPP_CATEGORY == GamezCategory:
if Gamez == "yes":
script = "autoProcessGames"
# Call Gamez's postprocessing script
Logger.info("Post-Process: Running Gamez's postprocessing script")
if Debug == "yes":
Logger.debug("Post-Process: Gamez-Script-ARGV1= %s", NZBPP_DIRECTORY)
Logger.debug("Post-Process: Gamez-Script-ARGV2= %s", NZBPP_NZBFILENAME)
Logger.debug("Post-Process: Gamez-Script-ARGV3= %s", PostProcessStatus)
result = autoProcessGames.process(NZBPP_DIRECTORY, NZBPP_NZBFILENAME, PostProcessStatus)
else:
Logger.debug("Post-Process: Ignored to run Gamez's postprocessing script as it is disabled by user")
return script, result
# Pass on postprocess exit codes to external scripts for handling failed downloads
def do_exit(Process_Code):
if Debug == "yes":
Logger.debug("Post-Process: Executing function 'do_exit' with argument %s", Process_Code)
nzbStatus = 0
if Process_Code != POSTPROCESS_SUCCESS:
if Delete_Failed == "yes":
os.chdir(os.path.split(NZBPP_DIRECTORY)[0])
shutil.rmtree(NZBPP_DIRECTORY)
else:
os.mkdir(Failed_Directory)
os.chdir(Failed_Directory)
for dirpath, dirnames, filenames in os.walk(NZBPP_DIRECTORY):
for file in filenames:
filePath = os.path.join(dirpath, file)
newPath = os.path,join(Failed_Directory, file)
shutil.move(filePath, newPath)
shutil.rmtree(NZBPP_DIRECTORY)
NZBPP_DIRECTORY = Failed_Directory
os.chdir(NZBPP_DIRECTORY)
nzbStatus=1
script = "none"
Email_Message2 = Email_Message
script, result = nzbToMedia(nzbStatus)
Logger.debug("after calling nzbToMedia")
Email_Subject.replace("<name>", NZBPP_NZBFILENAME)
Email_Subject.replace("<cat>", NZBPP_CATEGORY)
Email_Subject.replace("<script>", script)
Email_Message2.replace("<name>", NZBPP_NZBFILENAME)
Email_Message2.replace("<cat>", NZBPP_CATEGORY)
Email_Message2.replace("<script>", script)
if NZBPP_CATEGORY in Email_successful and nzbStatus == 0:
Email_Subject.replace("<status>", "completed")
Email_Message2.replace("<status>", "completed")
if Add_Log == "yes":
Email_Message2 = Email_Message2 + "\r\nLog Result"
f = open(logFile2)
lines = f.readlines()
for line in lines:
if line != "":
Email_Message2 = Email_Message2 + "\r\n" + line
f.close()
command = [sendEmail, "-f", Email_From, "-t", Email_To, "-s", Email_Server, "-o", "tsl=" + Tsl]
if Email_User != "" and Email_Pass != "":
command.append("-xu")
command.append(Email_User)
command.append("-xp")
command.append(Email_Pass)
command.append("-u")
command.append(Email_Subject)
command.append("-m")
command.append(Email_Message2)
call(command)
if NZBPP_CATEGORY in Email_failed and nzbStatus != 0:
Email_Subject.replace("<status>", "failed")
Email_Message.replace("<status>", "failed")
if Add_Log == "yes":
Email_Message2 = Email_Message2 + "\r\nLog Result"
f = open(logFile2)
lines = f.readlines()
for line in lines:
if line != "":
Email_Message2 = Email_Message2 + "\r\n" + line
f.close()
command = [sendEmail, "-f", Email_From, "-t", Email_To, "-s", Email_Server, "-o", "tsl=" + Tsl]
if Email_User != "" and Email_Pass != "":
command.append("-xu")
command.append(Email_User)
command.append("-xp")
command.append(Email_Pass)
command.append("-u")
command.append(Email_Subject)
command.append("-m")
command.append(Email_Message2)
call(command)
exit(Process_Code)
if os.path.isfile(os.path.join(os.path.dirname(sys.argv[0]), "autoProcessMedia.cfg.sample")):
migratecfg.migrate()
logFile = os.path.join(os.path.dirname(sys.argv[0]), "postprocess.log")
logging.config.fileConfig(os.path.join(os.path.dirname(sys.argv[0]), "logging.cfg"))
fileHandler = logging.handlers.RotatingFileHandler(logFile, mode='a', maxBytes=20000, backupCount=1, encoding='utf-8', delay=True)
fileHandler.formatter = logging.Formatter('%(asctime)s|%(levelname)-7.7s %(message)s', '%H:%M:%S')
fileHandler.level = logging.DEBUG
logging.getLogger().addHandler(fileHandler)
logFile2 = os.path.join(os.path.dirname(sys.argv[0]), "tmp.log")
fileHandler2 = logging.FileHandler(logFile2, mode='w', encoding='utf-8', delay=True)
fileHandler2.formatter = logging.Formatter('%(asctime)s|%(levelname)-7.7s %(message)s', '%H:%M:%S')
fileHandler2.level = logging.DEBUG
logging.getLogger().addHandler(fileHandler2)
Logger = logging.getLogger(__name__)
# NZBGet passes following arguments to postprocess-programm as environment
# variables:
# NZBPP_DIRECTORY - path to destination dir for downloaded files;
NZBPP_DIRECTORY = os.path.normpath(os.getenv('NZBPP_DIRECTORY'))
# NZBPP_NZBNAME - user-friendly name of processed nzb-file as it is displayed
# by the program. The file path and extension are removed.
# If download was renamed, this parameter reflects the new name;
NZBPP_NZBNAME = os.getenv('NZBPP_NZBNAME')
# NZBPP_NZBFILENAME - name of processed nzb-file. It includes file extension and also
# may include full path;
NZBPP_NZBFILENAME = os.getenv('NZBPP_NZBFILENAME')
# NZBPP_CATEGORY - category assigned to nzb-file (can be empty string);
NZBPP_CATEGORY = os.getenv('NZBPP_CATEGORY')
# NZBPP_PARSTATUS - result of par-check:
# 0 = not checked: par-check is disabled or nzb-file does
# not contain any par-files;
# 1 = checked and failed to repair;
# 2 = checked and successfully repaired;
# 3 = checked and can be repaired but repair is disabled.
NZBPP_PARSTATUS = os.getenv('NZBPP_PARSTATUS')
# NZBPP_UNPACKSTATUS - result of unpack:
# 0 = unpack is disabled or was skipped due to nzb-file
# properties or due to errors during par-check;
# 1 = unpack failed;
# 2 = unpack successful.
NZBPP_UNPACKSTATUS = os.getenv('NZBPP_UNPACKSTATUS')
NZBOP_CONFIGFILE = os.getenv('NZBOP_CONFIGFILE')
NZBOP_UNPACK = os.getenv('NZBOP_UNPACK')
NZBPR_PostProcess = os.getenv('NZBPR_PostProcess')
NZBPR_DestDir = os.getenv('NZBPR_DestDir')
NZBOP_ALLOWREPROCESS = os.getenv('NZBOP_ALLOWREPROCESS')
# Name of script's configuration file
SCRIPT_CONFIG_FILE = "nzbget-postprocess.conf"
# Exit codes
POSTPROCESS_PARCHECK_CURRENT=91
POSTPROCESS_PARCHECK_ALL=92
POSTPROCESS_SUCCESS=93
POSTPROCESS_ERROR=94
POSTPROCESS_NONE=95
# Check if the script is called from nzbget 10.0 or later
if NZBPP_DIRECTORY == "" or NZBOP_CONFIGFILE == "":
Logger.info("*** NZBGet post-processing script ***")
Logger.warning("This script is supposed to be called from nzbget (10.0 or later).")
exit(POSTPROCESS_ERROR)
if NZBOP_UNPACK == "":
Logger.error("This script requires nzbget version at least 10.0-testing-r555 or 10.0-stable.")
exit(POSTPROCESS_ERROR)
# Check if postprocessing was disabled in postprocessing parameters
# (for current nzb-file) via web-interface or via command line with
# "nzbget -E G O PostProcess=no <ID>"
if NZBPR_PostProcess == "no":
Logger.warning("Post-Process: Post-processing disabled for this nzb-file, exiting")
exit(POSTPROCESS_NONE)
Logger.info("Post-Process: Post-processing script successfully started")
#os.chdir(NZBPP_DIRECTORY)
# Determine the location of configuration file (it must be stored in
# the directory with nzbget.conf).
ConfigDir = os.path.dirname(NZBOP_CONFIGFILE)
ScriptConfigFile = os.path.join(ConfigDir, SCRIPT_CONFIG_FILE)
if not os.path.isfile(ScriptConfigFile):
Logger.error("Post-Process: Configuration file %s not found, exiting", ScriptConfigFile)
exit(POSTPROCESS_ERROR)
# Readg configuration file
config = ConfigObj(ScriptConfigFile)
Failed_Directory = config['Failed_Directory']
RenameIMG = config['RenameIMG']
JoinTS = config['JoinTS']
SickBeard = config['SickBeard']
SickBeardCategory = config['SickBeardCategory']
CouchPotato = config['CouchPotato']
CouchPotatoCategory = config['CouchPotatoCategory']
HeadPhones = config['HeadPhones']
HeadPhonesCategory = config['HeadPhonesCategory']
Mylar = config['Mylar']
MylarCategory = config['MylarCategory']
Gamez = config['Gamez']
GamezCategory = config['GamezCategory']
FileCleanUp = (config['FileCleanUp']).split(' ')
Delete_Failed = config['Delete_Failed']
Debug = config['Debug']
PostProcess = config['PostProcess']
DestDir = config['DestDir']
Email_successful = (config['Email_successful']).split(' ')
Email_failed = (config['Email_failed']).split(' ')
sendEmail = config['sendEmail']
Email_From = config['Email_From']
Email_To = config['Email_To']
Email_Server = config['Email_Server']
Tsl = config['Tsl']
Email_User = config['Email_User']
Email_Pass = config['Email_Pass']
Email_Subject = config['Email_Subject']
Email_Message = config['Email_Message']
Add_Log = config['Add_Log']
# Check nzbget.conf options
BadConfig = 0
if NZBOP_ALLOWREPROCESS == "yes":
Logger.error("Post-Process: Please disable option \"AllowReProcess\" in nzbget configuration file")
BadConfig = 1
if NZBOP_UNPACK != "yes":
Logger.error("Post-Process: Please enable option \"Unpack\" in nzbget configuration file")
BadConfig = 1
if BadConfig == 1:
Logger.error("Post-Process: Exiting due to incompatible nzbget configuration")
exit(POSTPROCESS_ERROR)
# Check par status
if NZBPP_PARSTATUS == 3:
Logger.warning("Post-Process: Par-check successful, but Par-repair disabled, exiting")
do_exit(POSTPROCESS_NONE)
if NZBPP_PARSTATUS == 1:
Logger.warning("Post-Process: Par-check failed, exiting")
do_exit(POSTPROCESS_NONE)
# Check unpack status
if NZBPP_UNPACKSTATUS == 1:
Logger.warning("Post-Process: Unpack failed, exiting")
do_exit(POSTPROCESS_NONE)
if NZBPP_UNPACKSTATUS == 0 and NZBPP_PARSTATUS != 2:
# Unpack is disabled or was skipped due to nzb-file properties or due to errors during par-check
for dirpath, dirnames, filenames in os.walk(NZBPP_DIRECTORY):
for file in filenames:
fileExtension = os.path.splitext(file)[1]
if fileExtension in ['.rar', '.7z', '.7z']:
Logger.warning("Post-Process: Archive files exist but unpack skipped, exiting")
exit(POSTPROCESS_NONE)
if fileExtension in ['.par2']:
Logger.warning("Post-Process: Unpack skipped and par-check skipped (although par2-files exist), exiting")
exit(POSTPROCESS_NONE)
if os.path.isfile(os.path.join(NZBPP_DIRECTORY, "_brokenlog.txt")):
Logger.warning("Post-Process: _brokenlog.txt exists, download is probably damaged, exiting")
exit(POSTPROCESS_NONE)
Logger.info("Post-Process: Neither archive- nor par2-files found, _brokenlog.txt doesn't exist, considering download successful")
# Check if destination directory exists (important for reprocessing of history items)
if not os.path.isdir(NZBPP_DIRECTORY):
Logger.error("Post-Process: Nothing to post-process: destination directory $NZBPP_DIRECTORY doesn't exist")
do_exit(POSTPROCESS_ERROR)
# All checks done, now processing the files
# If download contains only nzb-files move them into nzb-directory
# for further download
AllFilesCount = 0
NZBFilesCount = 0
for dirpath, dirnames, filenames in os.walk(NZBPP_DIRECTORY):
for file in filenames:
fileExtension = os.path.splitext(file)[1]
AllFilesCount = AllFilesCount + 1
if fileExtension in ['.nzb']:
NZBFilesCount = NZBFilesCount + 1
if AllFilesCount == NZBFilesCount:
Logger.info("Moving downloaded nzb-files into incoming nzb-directory for further download")
for dirpath, dirnames, filenames in os.walk(NZBPP_DIRECTORY):
for file in filenames:
filePath = os.path.join(dirpath, file)
shutil.move(filepath, os.path.join(NZBOP_NZBDIR, file))
# Clean up
Logger.info("Post-Process: Cleaning up")
#call(['chmod', '-R', 'a+w', NZBPP_DIRECTORY])
# Clean up list, space seperated array from GUI
for dirpath, dirnames, filenames in os.walk(NZBPP_DIRECTORY):
for file in filenames:
filePath = os.path.join(dirpath, file)
fileExtension = os.path.splitext(file)[1]
if fileExtension in FileCleanUp or fileExtension in ['.par']:
os.unlink(filePath)
if JoinTS == "yes":
# Join any split .ts files if they are named xxxx.0000.ts xxxx.0001.ts
# They will be joined together to a file called xxxx.ts
outputname = ""
for filename in iglob(os.path.join(NZBPP_DIRECTORY, '*.ts')):
file, ext = os.path.splitext(filename) # split off the .ts
file2, ext2 = os.path.splitext(file) # split off the .0001
outputname = file2 + ext
break
if outputname != "":
Logger.info("Post-Process: Joining ts-files")
destination = open('outputname', 'wb')
for filename in iglob(os.path.join(NZBPP_DIRECTORY, '*.ts')):
shutil.copyfileobj(open(filename, 'rb'), destination)
destination.close()
Logger.info("Post-Process: Deleting source ts-files")
for filename in iglob(os.path.join(NZBPP_DIRECTORY, '*.ts')):
if filename != outputname:
os.unlink(filename)
if RenameIMG == "yes":
# Rename img file to iso
# It will be renamed to .img.iso so you can see that it has been renamed
for dirpath, dirnames, filenames in os.walk(NZBPP_DIRECTORY):
for file in filenames:
filePath = os.path.join(dirpath, file)
fileExtension = os.path.splitext(file)[1]
if fileExtension in ['.img']:
Logger.info("Post-Process: Renaming img-files to iso")
newPath = os.path.join(dirpath, file + ".iso")
shutil.move(filePath, newPath)
# Check if destination directory was set in postprocessing parameters
# (for current nzb-file) via web-interface or via command line with
# "nzbget -E G O DestDir=/new/path <ID>"
if not (NZBPR_DestDir == None or NZBPR_DestDir == ""):
Logger.info("Post-Process: moving files to %s", NZBPR_DestDir)
os.mkdir(NZBPR_DestDir)
for dirpath, dirnames, filenames in os.walk(NZBPP_DIRECTORY):
for file in filenames:
filePath = os.path.join(dirpath, file)
newPath = os.path,join(NZBPR_DestDir, file)
shutil.move(filePath, newPath)
os.chdir(NZBPR_DestDir)
shutil.rmtree(NZBPP_DIRECTORY)
NZBPP_DIRECTORY = NZBPR_DestDir
os.chdir(NZBPP_DIRECTORY)
# All OK, requesting cleaning up of download queue
do_exit(POSTPROCESS_SUCCESS)