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

Added function for configuring logging

Browse source
This commit is contained in:
Berkona 2013-02-19 23:07:48 -05:00
commit b757f333ed
2 changed files with 37 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

47
TorrentToMedia.py
View file

@ -11,14 +11,12 @@ import linktastic.linktastic as linktastic
import autoProcessMovie import autoProcessMovie
import autoProcessTV import autoProcessTV
from nzbToMediaEnv import * from nzbToMediaEnv import *
from nzbToMediaUtil import *
nzbtomedia_configure_logging(os.path.dirname(sys.argv[0]))
Logger = logging.getLogger(__name__)
Logger = logging.getLogger()
logFile = os.path.join(os.path.dirname(sys.argv[0]), "postprocess.log")
logging.config.fileConfig(os.path.join(os.path.dirname(sys.argv[0]), "logger.conf"))
fileHandler = logging.FileHandler(logFile, encoding='utf-8', delay=True)
fileHandler.formatter = logging.Formatter('%(asctime)s|%(levelname)-7.7s %(message)s', '%H:%M:%S')
fileHandler.level = logging.DEBUG
Logger.addHandler(fileHandler)
def removeEmptyFolders(path): def removeEmptyFolders(path):
if not os.path.isdir(path): if not os.path.isdir(path):
@ -39,11 +37,12 @@ def removeEmptyFolders(path):
os.rmdir(path) os.rmdir(path)
#old_stdout = sys.stdout #backup the default stdout #old_stdout = sys.stdout #backup the default stdout
#log_file = open(os.path.join(os.path.dirname(sys.argv[0]), "postprocess.log"),"a+") #log_file = open(os.path.join(os.path.dirname(sys.argv[0]), "postprocess.log"),"a+")
#sys.stdout = log_file #create a local log file, and direct all "print" to the log. #sys.stdout = log_file #create a local log file, and direct all "print" to the log.
Logger.info("TorrentToMedia %s", VERSION) Logger.info("TorrentToMedia %s", VERSION)
if len(sys.argv) == 4: if len(sys.argv) == 4:
##You can use the following parameters (UTORRENT): ##You can use the following parameters (UTORRENT):
## ##
@ -73,7 +72,7 @@ if len(sys.argv) == 4:
##Finished - 11 ##Finished - 11
##Queued - 12 ##Queued - 12
##Stopped - 13 ##Stopped - 13
## We will pass in %D, %N, %L from uTorrent ## We will pass in %D, %N, %L from uTorrent
Logger.info("Script called from utorrent") Logger.info("Script called from utorrent")
Directory = sys.argv[1] ## %D -- Example output: F:\path\to\dir\My.Series.S01E01.720p.HDTV.x264-2HD Directory = sys.argv[1] ## %D -- Example output: F:\path\to\dir\My.Series.S01E01.720p.HDTV.x264-2HD
@ -100,7 +99,7 @@ else:
Logger.error("There was a problem loading variables from Transmission: Exiting") Logger.error("There was a problem loading variables from Transmission: Exiting")
sys.exit(-1) sys.exit(-1)
Category = '' #We dont have a category, so assume the last directory is the category for now. Category = '' #We dont have a category, so assume the last directory is the category for now.
Logger.debug("Received Directory: %s", Directory) Logger.debug("Received Directory: %s", Directory)
Logger.debug("Received Torrent Name: %s", Name) Logger.debug("Received Torrent Name: %s", Name)
Logger.debug("Received Category: %s", Category) Logger.debug("Received Category: %s", Category)
@ -136,7 +135,7 @@ if DirBase[1] == Name:
if not Category: if not Category:
Logger.info("Determined Category to be: %s", DirBase2[1]) Logger.info("Determined Category to be: %s", DirBase2[1])
Category = DirBase2[1] Category = DirBase2[1]
elif DirBase[1] == Movie_Cat or DirBase[1] == TV_Cat: elif DirBase[1] == Movie_Cat or DirBase[1] == TV_Cat:
if os.path.isdir(os.path.join(Directory, Name)): if os.path.isdir(os.path.join(Directory, Name)):
Logger.info("Found torrent directory %s in category directory %s", os.path.join(Directory, Name), Directory) Logger.info("Found torrent directory %s in category directory %s", os.path.join(Directory, Name), Directory)
@ -149,13 +148,13 @@ elif DirBase[1] == Movie_Cat or DirBase[1] == TV_Cat:
if not Category: if not Category:
Logger.info("Determined Category to be: %s", DirBase[1]) Logger.info("Determined Category to be: %s", DirBase[1])
Category = DirBase[1] Category = DirBase[1]
else: # no category found in directory. For Utorrent we can do a recursive scan. else: # no category found in directory. For Utorrent we can do a recursive scan.
Logger.info("The directory passed does not appear to include a category or the torrent name") Logger.info("The directory passed does not appear to include a category or the torrent name")
Logger.warn("You should change settings to download torrents to their own directory") Logger.warn("You should change settings to download torrents to their own directory")
Logger.info("We will try and determine which files to process, individually") Logger.info("We will try and determine which files to process, individually")
root = 1 root = 1
if Category == Movie_Cat: if Category == Movie_Cat:
destination = os.path.join(Movie_dest, Name) destination = os.path.join(Movie_dest, Name)
elif Category == TV_Cat: elif Category == TV_Cat:
@ -182,7 +181,7 @@ if root == 1:
break break
else: else:
continue continue
else: else:
ext = [os.path.splitext(file)[1] for file in f[1]] ext = [os.path.splitext(file)[1] for file in f[1]]
if set(ext).intersection(set(test)): if set(ext).intersection(set(test)):
Logger.info("Found compressed archives, extracting") Logger.info("Found compressed archives, extracting")
@ -198,7 +197,7 @@ else:
sys.exit(-1) sys.exit(-1)
if useLink == 0 and packed == 0 and video == 1: ## copy if useLink == 0 and packed == 0 and video == 1: ## copy
if root == 0: #move all files in tier own directory if root == 0: #move all files in tier own directory
Logger.info("Copying all files from %s to %s.", Directory, destination) Logger.info("Copying all files from %s to %s.", Directory, destination)
shutil.copytree(Directory, destination) shutil.copytree(Directory, destination)
else: #we only want to move files matching the torrent name when root directory is used. else: #we only want to move files matching the torrent name when root directory is used.
@ -225,9 +224,9 @@ elif useLink == 1 and packed == 0 and video == 1: ## hardlink
continue #ignore the other files continue #ignore the other files
source = os.path.join(dirpath, file) source = os.path.join(dirpath, file)
target = os.path.join(destination, file) target = os.path.join(destination, file)
linktastic.link(source, target) linktastic.link(source, target)
elif packed == 1: ## unpack elif packed == 1: ## unpack
## Using Windows? ## Using Windows?
if os.name == 'nt': if os.name == 'nt':
@ -235,7 +234,7 @@ elif packed == 1: ## unpack
ext_7zip = [".rar",".zip",".tar.gz","tgz",".tar.bz2",".tbz",".tar.lzma",".tlz",".7z",".xz"] ext_7zip = [".rar",".zip",".tar.gz","tgz",".tar.bz2",".tbz",".tar.lzma",".tlz",".7z",".xz"]
EXTRACT_COMMANDS = dict.fromkeys(ext_7zip, cmd_7zip) EXTRACT_COMMANDS = dict.fromkeys(ext_7zip, cmd_7zip)
Logger.info("We are using Windows") Logger.info("We are using Windows")
## Using linux? ## Using linux?
elif os.name == 'posix': elif os.name == 'posix':
required_cmds=["unrar", "unzip", "tar", "unxz", "unlzma", "7zr"] required_cmds=["unrar", "unzip", "tar", "unxz", "unlzma", "7zr"]
@ -252,7 +251,7 @@ elif packed == 1: ## unpack
".7z": ["7zr", "x"], ".7z": ["7zr", "x"],
} }
Logger.info("We are using *nix") Logger.info("We are using *nix")
## Need to add a check for which commands that can be utilized in *nix systems.. ## Need to add a check for which commands that can be utilized in *nix systems..
else: else:
Logger.error("Unknown OS, exiting") Logger.error("Unknown OS, exiting")
@ -263,7 +262,7 @@ elif packed == 1: ## unpack
if (Name in file) or (file in Name): if (Name in file) or (file in Name):
pass pass
else: else:
continue #ignore the other files continue #ignore the other files
ext = os.path.splitext(f) ext = os.path.splitext(f)
fp = os.path.join(Directory, os.path.normpath(f)) fp = os.path.join(Directory, os.path.normpath(f))
if ext[1] in (".gz", ".bz2", ".lzma"): if ext[1] in (".gz", ".bz2", ".lzma"):
@ -286,8 +285,8 @@ elif packed == 1: ## unpack
continue continue
Logger.info("Extracting to %s", destination) Logger.info("Extracting to %s", destination)
## Running.. ## Running..
Logger.info("Extracting %s %s %s %s", cmd[0], cmd[1], fp, destination) Logger.info("Extracting %s %s %s %s", cmd[0], cmd[1], fp, destination)
pwd = os.getcwd() # Get our Present Working Directory pwd = os.getcwd() # Get our Present Working Directory
os.chdir(destination) #not all unpack commands accept full paths, so just extract into this directory. os.chdir(destination) #not all unpack commands accept full paths, so just extract into this directory.
@ -314,7 +313,7 @@ elif packed == 1: ## unpack
else: else:
Logger.error("Extraction failed for %s. 7zip result was %s", fp, res) Logger.error("Extraction failed for %s. 7zip result was %s", fp, res)
except: except:
Logger.error("Extraction failed for %s. Could not call command %s %s %s %s", fp, cmd[0], cmd[1], fp) Logger.error("Extraction failed for %s. Could not call command %s %s %s %s", fp, cmd[0], cmd[1], fp)
os.chdir(pwd) # Go back to our Original Working Directory os.chdir(pwd) # Go back to our Original Working Directory
for dirpath, dirnames, filenames in os.walk(destination): #flatten out the directory to make postprocessing easier. for dirpath, dirnames, filenames in os.walk(destination): #flatten out the directory to make postprocessing easier.
@ -335,7 +334,7 @@ else:
## Now we pass off to CouchPotato or SickBeard. ## Now we pass off to CouchPotato or SickBeard.
old_stdout = sys.stdout #backup the default stdout old_stdout = sys.stdout #backup the default stdout
sys.stdout = Logger.info #Capture the print from the autoProcess scripts. sys.stdout = Logger.info #Capture the print from the autoProcess scripts.
if Category == Movie_Cat: if Category == Movie_Cat:
autoProcessMovie.process(destination, Name, status) autoProcessMovie.process(destination, Name, status)
elif Category == TV_Cat: elif Category == TV_Cat:
autoProcessTV.processEpisode(destination, Name, status) autoProcessTV.processEpisode(destination, Name, status)

14
nzbToMediaUtil.py Normal file
View file

@ -0,0 +1,14 @@
import logging
import logging.config
import os.path
def nzbtomedia_configure_logging(dirname):
logFile = os.path.join(dirname, "postprocess.log")
logging.config.fileConfig(os.path.join(dirname, "logger.conf"))
fileHandler = logging.FileHandler(logFile, 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)