mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-08-21 22:03:13 -07:00
add windows builder files for nzbget
This commit is contained in:
clinton-hall
parent
e8dc2f8683
commit
61df083863
5 changed files with 769 additions and 0 deletions
9
nzbget-postprocessing-files/windows-builder/README.md
Normal file
9
nzbget-postprocessing-files/windows-builder/README.md
Normal 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
105
nzbget-postprocessing-files/windows-builder/autoProcessMedia.cfg
Normal 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
23
nzbget-postprocessing-files/windows-builder/logging.cfg
Normal 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
|
||||
|
|
@ -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
|
||||
|
|
@ -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)
|
||||
Loading…
Add table
Add a link
Reference in a new issue