mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-07-16 02:02:53 -07:00
310 lines
No EOL
12 KiB
Python
310 lines
No EOL
12 KiB
Python
import os
|
|
import re
|
|
import time
|
|
import datetime
|
|
import urllib
|
|
import shutil
|
|
import sys
|
|
import platform
|
|
import nzbtomedia
|
|
from lib import requests
|
|
from nzbtomedia.Transcoder import Transcoder
|
|
from nzbtomedia.nzbToMediaSceneExceptions import process_all_exceptions
|
|
from nzbtomedia.nzbToMediaUtil import convert_to_ascii, delete, create_torrent_class
|
|
from nzbtomedia import logger
|
|
from nzbtomedia.transmissionrpc.client import Client as TransmissionClient
|
|
|
|
class autoProcessMovie:
|
|
def find_release_info(self, baseURL, download_id, dirName, nzbName, clientAgent):
|
|
imdbid = None
|
|
release_id = None
|
|
media_id = None
|
|
release_status = None
|
|
downloader = None
|
|
|
|
while(True):
|
|
# find imdbid in nzbName
|
|
m = re.search('(tt\d{7})', nzbName)
|
|
if m:
|
|
imdbid = m.group(1)
|
|
logger.postprocess("Found imdbid %s in name", imdbid)
|
|
break
|
|
|
|
# find imdbid in dirName
|
|
m = re.search('(tt\d{7})', dirName)
|
|
if m:
|
|
imdbid = m.group(1)
|
|
logger.postprocess("Found movie id %s in directory", imdbid)
|
|
break
|
|
break
|
|
|
|
url = baseURL + "/media.list/?release_status=snatched"
|
|
|
|
logger.debug("Opening URL: %s", url)
|
|
|
|
try:
|
|
r = requests.get(url)
|
|
except requests.ConnectionError:
|
|
logger.error("Unable to open URL")
|
|
return
|
|
|
|
results = r.json()
|
|
|
|
def search_results(results, clientAgent):
|
|
last_edit = {}
|
|
try:
|
|
for movie in results['movies']:
|
|
if imdbid:
|
|
if imdbid != movie['identifiers']['imdb']:
|
|
continue
|
|
|
|
for i, release in enumerate(movie['releases']):
|
|
if release['status'] != 'snatched':
|
|
continue
|
|
|
|
if download_id:
|
|
if release['download_info']['id'] == download_id:
|
|
return release
|
|
|
|
# store releases by datetime just incase we need to use this info
|
|
last_edit.update({datetime.datetime.fromtimestamp(release['last_edit']):release})
|
|
except:pass
|
|
|
|
if last_edit:
|
|
last_edit = sorted(last_edit.items())
|
|
if clientAgent != 'manual':
|
|
for item in last_edit:
|
|
release = item[1]
|
|
if release['download_info']['downloader'] == clientAgent:
|
|
return release
|
|
|
|
release = last_edit[0][1]
|
|
return release
|
|
|
|
matched_release = search_results(results, clientAgent)
|
|
|
|
if matched_release:
|
|
try:
|
|
release_id = matched_release['_id']
|
|
media_id = matched_release['media_id']
|
|
release_status = matched_release['status']
|
|
download_id = matched_release['download_info']['id']
|
|
downloader = matched_release['download_info']['downloader']
|
|
except:pass
|
|
|
|
return media_id, download_id, release_id, imdbid, release_status, downloader
|
|
|
|
def get_status(self, baseURL, media_id, release_id):
|
|
logger.debug("Attempting to get current status for movie:%s", media_id)
|
|
|
|
url = baseURL + "/media.get"
|
|
logger.debug("Opening URL: %s", url)
|
|
|
|
try:
|
|
r = requests.get(url, params={'id':media_id})
|
|
except requests.ConnectionError:
|
|
logger.error("Unable to open URL")
|
|
return
|
|
|
|
try:
|
|
result = r.json()
|
|
for release in result["media"]['releases']:
|
|
if release['_id'] == release_id:
|
|
return release["status"]
|
|
except:pass
|
|
|
|
def process(self, dirName, nzbName=None, status=0, clientAgent = "manual", download_id = "", inputCategory=None):
|
|
if dirName is None:
|
|
logger.error("No directory was given!")
|
|
return 1 # failure
|
|
|
|
# auto-detect correct section
|
|
section = nzbtomedia.CFG.findsection(inputCategory)
|
|
if not section:
|
|
logger.error(
|
|
"We were unable to find a section for category %s, please check your autoProcessMedia.cfg file.", inputCategory)
|
|
return 1
|
|
|
|
logger.postprocess("#########################################################")
|
|
logger.postprocess("## ..::[%s]::.. :: CATEGORY:[%s]", section, inputCategory)
|
|
logger.postprocess("#########################################################")
|
|
|
|
status = int(status)
|
|
|
|
host = nzbtomedia.CFG[section][inputCategory]["host"]
|
|
port = nzbtomedia.CFG[section][inputCategory]["port"]
|
|
apikey = nzbtomedia.CFG[section][inputCategory]["apikey"]
|
|
method = nzbtomedia.CFG[section][inputCategory]["method"]
|
|
delete_failed = int(nzbtomedia.CFG[section][inputCategory]["delete_failed"])
|
|
wait_for = int(nzbtomedia.CFG[section][inputCategory]["wait_for"])
|
|
|
|
try:
|
|
ssl = int(nzbtomedia.CFG[section][inputCategory]["ssl"])
|
|
except:
|
|
ssl = 0
|
|
try:
|
|
web_root = nzbtomedia.CFG[section][inputCategory]["web_root"]
|
|
except:
|
|
web_root = ""
|
|
try:
|
|
transcode = int(nzbtomedia.CFG["Transcoder"]["transcode"])
|
|
except:
|
|
transcode = 0
|
|
try:
|
|
remote_path = nzbtomedia.CFG[section][inputCategory]["remote_path"]
|
|
except:
|
|
remote_path = None
|
|
|
|
nzbName = str(nzbName) # make sure it is a string
|
|
|
|
if ssl:
|
|
protocol = "https://"
|
|
else:
|
|
protocol = "http://"
|
|
|
|
baseURL = protocol + host + ":" + port + web_root + "/api/" + apikey
|
|
|
|
media_id, download_id, release_id, imdbid, release_status, downloader = self.find_release_info(baseURL, download_id, dirName, nzbName, clientAgent)
|
|
|
|
if release_status:
|
|
if release_status != "snatched":
|
|
logger.warning("%s is marked with a status of %s on CouchPotato, skipping ...", nzbName, release_status)
|
|
return 0
|
|
elif imdbid and not (download_id or media_id or release_id):
|
|
logger.error("Could only find a imdbID for %s, sending folder name to be post-processed by CouchPotato ...", nzbName)
|
|
else:
|
|
logger.error("Could not find a release status for %s on CouchPotato, skipping ...", nzbName)
|
|
return 1
|
|
|
|
process_all_exceptions(nzbName.lower(), dirName)
|
|
nzbName, dirName = convert_to_ascii(nzbName, dirName)
|
|
|
|
if status == 0:
|
|
if transcode == 1:
|
|
result = Transcoder().Transcode_directory(dirName)
|
|
if result == 0:
|
|
logger.debug("Transcoding succeeded for files in %s", dirName)
|
|
else:
|
|
logger.warning("Transcoding failed for files in %s", dirName)
|
|
|
|
if method == "manage":
|
|
command = "/manage.update"
|
|
else:
|
|
command = "/renamer.scan"
|
|
|
|
params = {}
|
|
if download_id:
|
|
params['downloader'] = downloader
|
|
params['download_id'] = download_id
|
|
|
|
params['media_folder'] = urllib.quote(dirName)
|
|
if remote_path:
|
|
dirName_new = os.path.join(remote_path, os.path.basename(dirName)).replace("\\", "/")
|
|
params['media_folder'] = urllib.quote(dirName_new)
|
|
|
|
url = baseURL + command
|
|
|
|
logger.debug("Opening URL: %s", url)
|
|
|
|
try:
|
|
r = requests.get(url, params=params)
|
|
except requests.ConnectionError:
|
|
logger.error("Unable to open URL")
|
|
return 1 # failure
|
|
|
|
result = r.json()
|
|
|
|
logger.postprocess("CouchPotatoServer returned %s", result)
|
|
|
|
if result['success']:
|
|
logger.postprocess("%s scan started on CouchPotatoServer for %s", method, nzbName)
|
|
else:
|
|
logger.error("%s scan has NOT started on CouchPotatoServer for %s. Exiting", method, nzbName)
|
|
return 1 # failure
|
|
|
|
else:
|
|
logger.postprocess("Download of %s has failed.", nzbName)
|
|
|
|
if delete_failed and not os.path.dirname(dirName) == dirName:
|
|
logger.postprocess("Deleting failed files and folder %s", dirName)
|
|
delete(dirName)
|
|
|
|
if not download_id:
|
|
logger.warning("Cound not find a movie in the database for release %s", nzbName)
|
|
logger.warning("Please manually ignore this release and refresh the wanted movie")
|
|
logger.error("Exiting autoProcessMovie script")
|
|
return 1 # failure
|
|
|
|
logger.postprocess("Ignoring current failed release %s ...", nzbName)
|
|
|
|
url = baseURL + "/release.ignore"
|
|
logger.debug("Opening URL: %s", url)
|
|
|
|
try:
|
|
r = requests.get(url, params={'id':release_id})
|
|
except requests.ConnectionError:
|
|
logger.error("Unable to open URL")
|
|
return 1 # failure
|
|
|
|
result = r.json()
|
|
if result['success']:
|
|
logger.postprocess("%s has been set to ignored on CouchPotato", nzbName)
|
|
else:
|
|
logger.warning("Failed to ignore %s on CouchPotato ...", nzbName)
|
|
|
|
logger.postprocess("Snatching next highest ranked release on CouchPotato ...")
|
|
|
|
url = baseURL + "/movie.searcher.try_next"
|
|
logger.debug("Opening URL: %s", url)
|
|
|
|
try:
|
|
r = requests.get(url, params={'media_id':media_id})
|
|
except requests.ConnectionError:
|
|
logger.error("Unable to open URL")
|
|
return 1 # failure
|
|
|
|
result = r.json()
|
|
if result['success']:
|
|
logger.postprocess("CouchPotato successfully snatched the next highest release ...", nzbName)
|
|
return 0
|
|
else:
|
|
logger.postprocess("CouchPotato was unable to find a higher release then %s to snatch ...", nzbName)
|
|
return 1
|
|
|
|
if not (download_id or media_id or release_id) and imdbid:
|
|
url = baseURL + "/media.get"
|
|
logger.debug("Opening URL: %s", url)
|
|
|
|
# we will now check to see if CPS has finished renaming before returning to TorrentToMedia and unpausing.
|
|
timeout = time.time() + 60 * int(wait_for)
|
|
while (time.time() < timeout): # only wait 2 (default) minutes, then return.
|
|
try:
|
|
r = requests.get(url, params={id:imdbid})
|
|
except requests.ConnectionError:
|
|
logger.error("Unable to open URL")
|
|
return 1 # failure
|
|
|
|
result = r.json()
|
|
if result['success']:
|
|
logger.postprocess("CouchPotato successfully added %s to it's database ...", nzbName)
|
|
return 0
|
|
|
|
logger.postprocess("CouchPotato was unable to add %s to its database ...", nzbName)
|
|
return 1
|
|
elif not (download_id or media_id or release_id):
|
|
return 1
|
|
|
|
# we will now check to see if CPS has finished renaming before returning to TorrentToMedia and unpausing.
|
|
timeout = time.time() + 60 * int(wait_for)
|
|
while (time.time() < timeout): # only wait 2 (default) minutes, then return.
|
|
current_status = self.get_status(baseURL, media_id, release_id)
|
|
if current_status is not None and current_status != release_status: # Something has changed. CPS must have processed this movie.
|
|
logger.postprocess("SUCCESS: This release is now marked as status [%s] in CouchPotatoServer", current_status.upper())
|
|
return 0 # success
|
|
|
|
# pause and let CouchPotatoServer catch its breath
|
|
time.sleep(10 * int(wait_for))
|
|
|
|
# The status hasn't changed. we have waited 2 minutes which is more than enough. uTorrent can resule seeding now.
|
|
logger.warning("The movie does not appear to have changed status after %s minutes. Please check CouchPotato Logs", wait_for)
|
|
return 1 # failure |