mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-10 19:35:08 -05:00
9a3e7ab0a9
Added fanzub anime nzb provider. Fixed NyaaTorrents anime provider. This is in testing phase so bugs are to be expected.
362 lines
14 KiB
Python
362 lines
14 KiB
Python
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# This file is part of SickRage.
|
|
#
|
|
# SickRage 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 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# SickRage 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 SickRage. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import os.path
|
|
import operator
|
|
import platform
|
|
import re
|
|
import uuid
|
|
|
|
from sickbeard import version
|
|
|
|
INSTANCE_ID = str(uuid.uuid1())
|
|
USER_AGENT = ('SickRage/' + version.SICKBEARD_VERSION.replace(' ', '-') +
|
|
' (' + platform.system() + '; ' + platform.release() +
|
|
'; ' + INSTANCE_ID + ')')
|
|
|
|
mediaExtensions = ['avi', 'mkv', 'mpg', 'mpeg', 'wmv',
|
|
'ogm', 'mp4', 'iso', 'img', 'divx',
|
|
'm2ts', 'm4v', 'ts', 'flv', 'f4v',
|
|
'mov', 'rmvb', 'vob', 'dvr-ms', 'wtv',
|
|
'ogv', '3gp', 'webm']
|
|
|
|
subtitleExtensions = ['srt', 'sub', 'ass', 'idx', 'ssa']
|
|
|
|
cpu_presets = {'HIGH': 0.1,
|
|
'NORMAL': 0.05,
|
|
'LOW': 0.01
|
|
}
|
|
|
|
### Other constants
|
|
MULTI_EP_RESULT = -1
|
|
SEASON_RESULT = -2
|
|
|
|
### Notification Types
|
|
NOTIFY_SNATCH = 1
|
|
NOTIFY_DOWNLOAD = 2
|
|
NOTIFY_SUBTITLE_DOWNLOAD = 3
|
|
|
|
notifyStrings = {}
|
|
notifyStrings[NOTIFY_SNATCH] = "Started Download"
|
|
notifyStrings[NOTIFY_DOWNLOAD] = "Download Finished"
|
|
notifyStrings[NOTIFY_SUBTITLE_DOWNLOAD] = "Subtitle Download Finished"
|
|
|
|
### Episode statuses
|
|
UNKNOWN = -1 # should never happen
|
|
UNAIRED = 1 # episodes that haven't aired yet
|
|
SNATCHED = 2 # qualified with quality
|
|
WANTED = 3 # episodes we don't have but want to get
|
|
DOWNLOADED = 4 # qualified with quality
|
|
SKIPPED = 5 # episodes we don't want
|
|
ARCHIVED = 6 # episodes that you don't have locally (counts toward download completion stats)
|
|
IGNORED = 7 # episodes that you don't want included in your download stats
|
|
SNATCHED_PROPER = 9 # qualified with quality
|
|
SUBTITLED = 10 # qualified with quality
|
|
FAILED = 11 #episode downloaded or snatched we don't want
|
|
SNATCHED_BEST = 12 # episode redownloaded using best quality
|
|
|
|
NAMING_REPEAT = 1
|
|
NAMING_EXTEND = 2
|
|
NAMING_DUPLICATE = 4
|
|
NAMING_LIMITED_EXTEND = 8
|
|
NAMING_SEPARATED_REPEAT = 16
|
|
NAMING_LIMITED_EXTEND_E_PREFIXED = 32
|
|
|
|
multiEpStrings = {}
|
|
multiEpStrings[NAMING_REPEAT] = "Repeat"
|
|
multiEpStrings[NAMING_SEPARATED_REPEAT] = "Repeat (Separated)"
|
|
multiEpStrings[NAMING_DUPLICATE] = "Duplicate"
|
|
multiEpStrings[NAMING_EXTEND] = "Extend"
|
|
multiEpStrings[NAMING_LIMITED_EXTEND] = "Extend (Limited)"
|
|
multiEpStrings[NAMING_LIMITED_EXTEND_E_PREFIXED] = "Extend (Limited, E-prefixed)"
|
|
|
|
|
|
class Quality:
|
|
NONE = 0 # 0
|
|
SDTV = 1 # 1
|
|
SDDVD = 1 << 1 # 2
|
|
HDTV = 1 << 2 # 4
|
|
RAWHDTV = 1 << 3 # 8 -- 720p/1080i mpeg2 (trollhd releases)
|
|
FULLHDTV = 1 << 4 # 16 -- 1080p HDTV (QCF releases)
|
|
HDWEBDL = 1 << 5 # 32
|
|
FULLHDWEBDL = 1 << 6 # 64 -- 1080p web-dl
|
|
HDBLURAY = 1 << 7 # 128
|
|
FULLHDBLURAY = 1 << 8 # 256
|
|
|
|
# put these bits at the other end of the spectrum, far enough out that they shouldn't interfere
|
|
UNKNOWN = 1 << 15 # 32768
|
|
|
|
qualityStrings = {NONE: "N/A",
|
|
UNKNOWN: "Unknown",
|
|
SDTV: "SD TV",
|
|
SDDVD: "SD DVD",
|
|
HDTV: "HD TV",
|
|
RAWHDTV: "RawHD TV",
|
|
FULLHDTV: "1080p HD TV",
|
|
HDWEBDL: "720p WEB-DL",
|
|
FULLHDWEBDL: "1080p WEB-DL",
|
|
HDBLURAY: "720p BluRay",
|
|
FULLHDBLURAY: "1080p BluRay"}
|
|
|
|
statusPrefixes = {DOWNLOADED: "Downloaded",
|
|
SNATCHED: "Snatched",
|
|
SNATCHED_PROPER: "Snatched (Proper)",
|
|
FAILED: "Failed",
|
|
SNATCHED_BEST: "Snatched (Best)"}
|
|
|
|
@staticmethod
|
|
def _getStatusStrings(status):
|
|
toReturn = {}
|
|
for x in Quality.qualityStrings.keys():
|
|
toReturn[Quality.compositeStatus(status, x)] = Quality.statusPrefixes[status] + " (" + \
|
|
Quality.qualityStrings[x] + ")"
|
|
return toReturn
|
|
|
|
@staticmethod
|
|
def combineQualities(anyQualities, bestQualities):
|
|
anyQuality = 0
|
|
bestQuality = 0
|
|
if anyQualities:
|
|
anyQuality = reduce(operator.or_, anyQualities)
|
|
if bestQualities:
|
|
bestQuality = reduce(operator.or_, bestQualities)
|
|
return anyQuality | (bestQuality << 16)
|
|
|
|
@staticmethod
|
|
def splitQuality(quality):
|
|
anyQualities = []
|
|
bestQualities = []
|
|
for curQual in Quality.qualityStrings.keys():
|
|
if curQual & quality:
|
|
anyQualities.append(curQual)
|
|
if curQual << 16 & quality:
|
|
bestQualities.append(curQual)
|
|
|
|
return (sorted(anyQualities), sorted(bestQualities))
|
|
|
|
@staticmethod
|
|
def nameQuality(name, anime=False):
|
|
"""
|
|
Return The quality from an episode File renamed by Sickbeard
|
|
If no quality is achieved it will try sceneQuality regex
|
|
"""
|
|
|
|
name = os.path.basename(name)
|
|
|
|
# if we have our exact text then assume we put it there
|
|
for x in sorted(Quality.qualityStrings.keys(), reverse=True):
|
|
if x == Quality.UNKNOWN:
|
|
continue
|
|
|
|
if x == Quality.NONE: #Last chance
|
|
return Quality.sceneQuality(name, anime)
|
|
|
|
regex = '\W' + Quality.qualityStrings[x].replace(' ', '\W') + '\W'
|
|
regex_match = re.search(regex, name, re.I)
|
|
if regex_match:
|
|
return x
|
|
|
|
@staticmethod
|
|
def sceneQuality(name, anime=False):
|
|
"""
|
|
Return The quality from the scene episode File
|
|
"""
|
|
|
|
name = os.path.basename(name)
|
|
|
|
checkName = lambda list, func: func([re.search(x, name, re.I) for x in list])
|
|
|
|
if anime:
|
|
blueRayOptions = checkName(["bluray", "blu-ray"], any)
|
|
hdOptions = checkName(["720p", "1280x720", "960x720"], any)
|
|
fullHD = checkName(["1080p", "1920x1080"], any)
|
|
|
|
if checkName(["360p", "XviD"], any):
|
|
return Quality.SDTV
|
|
elif checkName(["dvd", "480p", "848x480"], any):
|
|
return Quality.SDDVD
|
|
elif hdOptions and not blueRayOptions and not fullHD:
|
|
return Quality.HDTV
|
|
elif hdOptions and not blueRayOptions and not fullHD:
|
|
return Quality.HDWEBDL
|
|
elif blueRayOptions and hdOptions and not fullHD:
|
|
return Quality.HDBLURAY
|
|
elif fullHD:
|
|
return Quality.FULLHDBLURAY
|
|
else:
|
|
return Quality.UNKNOWN
|
|
|
|
if checkName(["(pdtv|hdtv|dsr|tvrip).(xvid|x264|h.?264)"], all) and not checkName(["(720|1080)[pi]"], all):
|
|
return Quality.SDTV
|
|
elif checkName(["web.dl|webrip", "xvid|x264|h.?264"], all) and not checkName(["(720|1080)[pi]"], all):
|
|
return Quality.SDTV
|
|
elif checkName(["(dvdrip|b[r|d]rip)(.ws)?.(xvid|divx|x264)"], any) and not checkName(["(720|1080)[pi]"], all):
|
|
return Quality.SDDVD
|
|
elif checkName(["720p", "hdtv", "x264"], all) or checkName(["hr.ws.pdtv.x264"], any) and not checkName(
|
|
["(1080)[pi]"], all):
|
|
return Quality.HDTV
|
|
elif checkName(["720p|1080i", "hdtv", "mpeg-?2"], all) or checkName(["1080[pi].hdtv", "h.?264"], all):
|
|
return Quality.RAWHDTV
|
|
elif checkName(["1080p", "hdtv", "x264"], all):
|
|
return Quality.FULLHDTV
|
|
elif checkName(["720p", "web.dl", "h.?264"], all) or checkName(["720p", "itunes", "h.?264"], all):
|
|
return Quality.HDWEBDL
|
|
elif checkName(["1080p", "web.dl", "h.?264"], all) or checkName(["1080p", "itunes", "h.?264"], all):
|
|
return Quality.FULLHDWEBDL
|
|
elif checkName(["720p", "webrip", "x264"], all):
|
|
return Quality.HDWEBDL
|
|
elif checkName(["1080p", "webrip", "x264"], all):
|
|
return Quality.FULLHDWEBDL
|
|
elif checkName(["720p", "bluray|hddvd|b[r|d]rip", "x264"], all):
|
|
return Quality.HDBLURAY
|
|
elif checkName(["1080p", "bluray|hddvd|b[r|d]rip", "x264"], all):
|
|
return Quality.FULLHDBLURAY
|
|
else:
|
|
return Quality.UNKNOWN
|
|
|
|
@staticmethod
|
|
def assumeQuality(name):
|
|
if name.lower().endswith((".avi", ".mp4")):
|
|
return Quality.SDTV
|
|
# elif name.lower().endswith(".mkv"):
|
|
# return Quality.HDTV
|
|
elif name.lower().endswith(".ts"):
|
|
return Quality.RAWHDTV
|
|
else:
|
|
return Quality.UNKNOWN
|
|
|
|
@staticmethod
|
|
def compositeStatus(status, quality):
|
|
return status + 100 * quality
|
|
|
|
@staticmethod
|
|
def qualityDownloaded(status):
|
|
return (status - DOWNLOADED) / 100
|
|
|
|
@staticmethod
|
|
def splitCompositeStatus(status):
|
|
"""Returns a tuple containing (status, quality)"""
|
|
if status == UNKNOWN:
|
|
return (UNKNOWN, Quality.UNKNOWN)
|
|
|
|
for x in sorted(Quality.qualityStrings.keys(), reverse=True):
|
|
if status > x * 100:
|
|
return (status - x * 100, x)
|
|
|
|
return (status, Quality.NONE)
|
|
|
|
@staticmethod
|
|
def statusFromName(name, assume=True):
|
|
quality = Quality.nameQuality(name)
|
|
if assume and quality == Quality.UNKNOWN:
|
|
quality = Quality.assumeQuality(name)
|
|
return Quality.compositeStatus(DOWNLOADED, quality)
|
|
|
|
DOWNLOADED = None
|
|
SNATCHED = None
|
|
SNATCHED_PROPER = None
|
|
FAILED = None
|
|
SNATCHED_BEST = None
|
|
|
|
|
|
Quality.DOWNLOADED = [Quality.compositeStatus(DOWNLOADED, x) for x in Quality.qualityStrings.keys()]
|
|
Quality.SNATCHED = [Quality.compositeStatus(SNATCHED, x) for x in Quality.qualityStrings.keys()]
|
|
Quality.SNATCHED_PROPER = [Quality.compositeStatus(SNATCHED_PROPER, x) for x in Quality.qualityStrings.keys()]
|
|
Quality.FAILED = [Quality.compositeStatus(FAILED, x) for x in Quality.qualityStrings.keys()]
|
|
Quality.SNATCHED_BEST = [Quality.compositeStatus(SNATCHED_BEST, x) for x in Quality.qualityStrings.keys()]
|
|
|
|
SD = Quality.combineQualities([Quality.SDTV, Quality.SDDVD], [])
|
|
HD = Quality.combineQualities(
|
|
[Quality.HDTV, Quality.FULLHDTV, Quality.HDWEBDL, Quality.FULLHDWEBDL, Quality.HDBLURAY, Quality.FULLHDBLURAY],
|
|
[]) # HD720p + HD1080p
|
|
HD720p = Quality.combineQualities([Quality.HDTV, Quality.HDWEBDL, Quality.HDBLURAY], [])
|
|
HD1080p = Quality.combineQualities([Quality.FULLHDTV, Quality.FULLHDWEBDL, Quality.FULLHDBLURAY], [])
|
|
ANY = Quality.combineQualities(
|
|
[Quality.SDTV, Quality.SDDVD, Quality.HDTV, Quality.FULLHDTV, Quality.HDWEBDL, Quality.FULLHDWEBDL,
|
|
Quality.HDBLURAY, Quality.FULLHDBLURAY, Quality.UNKNOWN], []) # SD + HD
|
|
|
|
# legacy template, cant remove due to reference in mainDB upgrade?
|
|
BEST = Quality.combineQualities([Quality.SDTV, Quality.HDTV, Quality.HDWEBDL], [Quality.HDTV])
|
|
|
|
qualityPresets = (SD, HD, HD720p, HD1080p, ANY)
|
|
qualityPresetStrings = {SD: "SD",
|
|
HD: "HD",
|
|
HD720p: "HD720p",
|
|
HD1080p: "HD1080p",
|
|
ANY: "Any"}
|
|
|
|
|
|
class StatusStrings:
|
|
def __init__(self):
|
|
self.statusStrings = {UNKNOWN: "Unknown",
|
|
UNAIRED: "Unaired",
|
|
SNATCHED: "Snatched",
|
|
DOWNLOADED: "Downloaded",
|
|
SKIPPED: "Skipped",
|
|
SNATCHED_PROPER: "Snatched (Proper)",
|
|
WANTED: "Wanted",
|
|
ARCHIVED: "Archived",
|
|
IGNORED: "Ignored",
|
|
SUBTITLED: "Subtitled",
|
|
FAILED: "Failed",
|
|
SNATCHED_BEST: "Snatched (Best)"}
|
|
|
|
def __getitem__(self, name):
|
|
if name in Quality.DOWNLOADED + Quality.SNATCHED + Quality.SNATCHED_PROPER + Quality.SNATCHED_BEST:
|
|
status, quality = Quality.splitCompositeStatus(name)
|
|
if quality == Quality.NONE:
|
|
return self.statusStrings[status]
|
|
else:
|
|
return self.statusStrings[status] + " (" + Quality.qualityStrings[quality] + ")"
|
|
else:
|
|
return self.statusStrings[name] if self.statusStrings.has_key(name) else ''
|
|
|
|
def has_key(self, name):
|
|
return name in self.statusStrings or name in Quality.DOWNLOADED or name in Quality.SNATCHED or name in Quality.SNATCHED_PROPER or name in Quality.SNATCHED_BEST
|
|
|
|
|
|
statusStrings = StatusStrings()
|
|
|
|
|
|
class Overview:
|
|
UNAIRED = UNAIRED # 1
|
|
QUAL = 2
|
|
WANTED = WANTED # 3
|
|
GOOD = 4
|
|
SKIPPED = SKIPPED # 5
|
|
|
|
# For both snatched statuses. Note: SNATCHED/QUAL have same value and break dict.
|
|
SNATCHED = SNATCHED_PROPER = SNATCHED_BEST # 9
|
|
|
|
overviewStrings = {SKIPPED: "skipped",
|
|
WANTED: "wanted",
|
|
QUAL: "qual",
|
|
GOOD: "good",
|
|
UNAIRED: "unaired",
|
|
SNATCHED: "snatched"}
|
|
|
|
# Get our xml namespaces correct for lxml
|
|
XML_NSMAP = {'xsi': 'http://www.w3.org/2001/XMLSchema-instance',
|
|
'xsd': 'http://www.w3.org/2001/XMLSchema'}
|
|
|
|
countryList = {'Australia': 'AU',
|
|
'Canada': 'CA',
|
|
'USA': 'US'
|
|
}
|
|
|