github/nzbToMedia
1
0
Fork 0
mirror of https://github.com/clinton-hall/nzbToMedia.git synced 2025-08-21 05:43:16 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

logging and QoQ close #76

Browse source
This commit is contained in:
Clinton Hall 2013-02-27 18:33:17 -08:00
commit 2304c6c9be
1 changed files with 50 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

74
autoProcessMovie.py
View file

@ -1,11 +1,15 @@
import sys import sys
import urllib import urllib
import os.path import os
import shutil import shutil
import ConfigParser import ConfigParser
import time import time
import json import json
from pprint import pprint import logging
from nzbToMediaEnv import *
Logger = logging.getLogger()
class AuthURLOpener(urllib.FancyURLopener): class AuthURLOpener(urllib.FancyURLopener):
def __init__(self, user, pw): def __init__(self, user, pw):
@ -25,16 +29,32 @@ class AuthURLOpener(urllib.FancyURLopener):
self.numTries = 0 self.numTries = 0
return urllib.FancyURLopener.open(self, url) return urllib.FancyURLopener.open(self, url)
def custom_groups(group, dirName):
mediaContainer = ['.mkv', '.avi', '.divx', '.xvid', '.mov', '.wmv', '.mp4', '.mpg', '.mpeg', '.iso']
if group == "QoQ": # for my NL friends :) we want to reverse the file names for the video files.
for dirpath, dirnames, filenames in os.walk(dirName):
for file in filenames:
filePath = os.path.join(dirpath, file)
fileExtention = os.path.splitext(file)[1]
if fileExtention in mediaContainer: # If the file is a video file
Logger.debug("Reversing the file name for a QoQ release %s", file)
newname = os.path.splitext(file)[0][::-1]
newfile = newname + fileExtention
newfilePath = os.path.join(dirpath, newfile)
os.rename(filePath, newfilePath)
Logger.debug("New file name is %s", newfile)
else: # we can add more customizations here.
pass
def process(dirName, nzbName=None, status=0): def process(dirName, nzbName=None, status=0):
status = int(status) status = int(status)
config = ConfigParser.ConfigParser() config = ConfigParser.ConfigParser()
configFilename = os.path.join(os.path.dirname(sys.argv[0]), "autoProcessMedia.cfg") configFilename = os.path.join(os.path.dirname(sys.argv[0]), "autoProcessMedia.cfg")
print "Loading config from", configFilename Logger.info("Loading config from %s", configFilename)
if not os.path.isfile(configFilename): if not os.path.isfile(configFilename):
print "ERROR: You need an autoProcessMedia.cfg file - did you rename and edit the .sample?" Logger.error("You need an autoProcessMedia.cfg file - did you rename and edit the .sample?")
sys.exit(-1) sys.exit(-1)
config.read(configFilename) config.read(configFilename)
@ -69,6 +89,13 @@ def process(dirName, nzbName=None, status=0):
# don't delay when we are calling this script manually. # don't delay when we are calling this script manually.
if nzbName == "Manual Run": if nzbName == "Manual Run":
delay = 0 delay = 0
# check for custom groups
customgroups = ['QoQ'] # we can add more to this list
for index in range(len(customgroups)):
if customgroups[index].lower() in nzbName.lower(): # match the group in the nzbname
custom_groups(customgroups[index], dirName) # files have been renamned
break
if status == 0: if status == 0:
if method == "manage": if method == "manage":
@ -78,41 +105,40 @@ def process(dirName, nzbName=None, status=0):
url = protocol + host + ":" + port + web_root + "/api/" + apikey + "/" + command url = protocol + host + ":" + port + web_root + "/api/" + apikey + "/" + command
print "waiting for", str(delay), "seconds to allow CPS to process newly extracted files" Logger.info("waiting for %s seconds to allow CPS to process newly extracted files", str(delay))
time.sleep(delay) time.sleep(delay)
print "Opening URL:", url Logger.debug("Opening URL: %s", url)
try: try:
urlObj = myOpener.openit(url) urlObj = myOpener.openit(url)
except IOError, e: except IOError, e:
print "Unable to open URL: ", str(e) Logger.error("Unable to open URL: %s", str(e))
sys.exit(1) sys.exit(1)
result = json.load(urlObj) result = json.load(urlObj)
print "CouchPotatoServer returned", result Logger.info("CouchPotatoServer returned %s", result)
if result['success']: if result['success']:
print command, "started on CouchPotatoServer for", nzbName1 Logger.info("%s started on CouchPotatoServer for %s", command, nzbName1)
else: else:
print "Error", command, "has NOT started on CouchPotatoServer for", nzbName1 Logger.error("%s has NOT started on CouchPotatoServer for %s", command, nzbName1)
else: else:
print "download of", nzbName1, "has failed." Logger.info("download of %s has failed.", nzbName1)
print "trying to re-cue the next highest ranked release" Logger.info("trying to re-cue the next highest ranked release")
a=nzbName1.find('.cp(')+4 a=nzbName1.find('.cp(')+4
b=nzbName1[a:].find(')')+a b=nzbName1[a:].find(')')+a
imdbid=nzbName1[a:b] imdbid=nzbName1[a:b]
#print imdbid
url = protocol + host + ":" + port + web_root + "/api/" + apikey + "/movie.list" url = protocol + host + ":" + port + web_root + "/api/" + apikey + "/movie.list"
print "Opening URL:", url Logger.debug("Opening URL: %s", url)
try: try:
urlObj = myOpener.openit(url) urlObj = myOpener.openit(url)
except IOError, e: except IOError, e:
print "Unable to open URL: ", str(e) Logger.error("Unable to open URL: %s", str(e))
sys.exit(1) sys.exit(1)
n=0 n=0
@ -123,31 +149,31 @@ def process(dirName, nzbName=None, status=0):
for index in range(len(movieid)): for index in range(len(movieid)):
if identifier[index] == imdbid: if identifier[index] == imdbid:
movid = str(movieid[index]) movid = str(movieid[index])
print "found movie id", movid, "in database for release", nzbName1 Logger.info("found movie id %s in database for release %s", movid, nzbName1)
n = n + 1 n = n + 1
break break
if n == 0: if n == 0:
print "cound not find a movie in the database for release", nzbName1 Logger.warning("cound not find a movie in the database for release %s", nzbName1)
print "please manually ignore this release and refresh the wanted movie" Logger.warning("please manually ignore this release and refresh the wanted movie")
print "exiting postprocessing script" Logger.error("exiting postprocessing script")
sys.exit(1) sys.exit(1)
url = protocol + host + ":" + port + web_root + "/api/" + apikey + "/searcher.try_next/?id=" + movid url = protocol + host + ":" + port + web_root + "/api/" + apikey + "/searcher.try_next/?id=" + movid
print "Opening URL:", url Logger.debug("Opening URL: %s", url)
try: try:
urlObj = myOpener.openit(url) urlObj = myOpener.openit(url)
except IOError, e: except IOError, e:
print "Unable to open URL: ", str(e) Logger.error("Unable to open URL: %s", str(e))
sys.exit(1) sys.exit(1)
result = urlObj.readlines() result = urlObj.readlines()
for line in result: for line in result:
print line Logger.info("%s", line)
print "movie", movid, "set to try the next best release on CouchPotatoServer" Logger.info("movie %s set to try the next best release on CouchPotatoServer", movid)
if delete_failed: if delete_failed:
print "Deleting failed files and folder", dirName Logger.info("Deleting failed files and folder %s", dirName)
shutil.rmtree(dirName) shutil.rmtree(dirName)