shane/recordingprocessing
1
0
Fork 0
Code Issues Pull Requests Releases 2 Wiki Activity

added settings file

Browse Source
This commit is contained in:
Shane Frischkorn
2013-07-08 17:21:42 +10:00
parent ccb103ad36
commit 7a34fd01c8
8 changed files with 418 additions and 230 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
libsickbeard.py
View File

@@ -12,13 +12,14 @@ from fuzzywuzzy import fuzz
from operator import itemgetter
class Sickbeard:
def __init__(self, address, port, apikey):
self.address = address
self.port = port
self.apikey = apikey
def __init__(self, settings):
self.__settings = settings
self.__address = settings.SickbeardAddress()
self.__port = settings.SickbeardPort()
self.__apikey = settings.SickbeardAPIKey()
def __GetApiURL(self):
return "http://{0}:{1}/api/{2}/".format(self.address, self.port, self.apikey)
return "http://{0}:{1}/api/{2}/".format(self.__address, self.__port, self.__apikey)
def FindShowId(self, showName):
jsonurl = urlopen(self.__GetApiURL()+"?cmd=shows")
@@ -42,10 +43,11 @@ class Sickbeard:
def FindEpisodeByDescription(self, showId, season, episode, description):
jsonEpisodeUrl = urlopen("{0}?cmd=episode&tvdbid={1}&season={2}&episode={3}".format(self.__GetApiURL(), showId, season, episode))
episodeResult = json.loads(jsonEpisodeUrl.read())
if fuzz.ratio(episodeResult['data']['description'].lower(), description.lower()) > 85:
return (season, episode, episodeResult['data']['name'])
sickbeardDescription = episodeResult['data']['description']
if fuzz.ratio(sickbeardDescription.lower(), description.lower()) > 85 or fuzz.ratio(sickbeardDescription.lower()[:len(description)], description.lower()) > 85 or fuzz.ratio(sickbeardDescription.lower(), description.lower()[:len(sickbeardDescription)]) > 85:
return (season, episode, episodeResult['data']['name'])
return None
def FindEpisode(self, showId, name=None, description=None):
@@ -54,8 +56,8 @@ class Sickbeard:
for season in result['data']:
for episode in result['data'][season]:
if name is not None and name.lower() == result['data'][season][episode]['name'].lower():
return (season, episode, name)
if name is not None and fuzz.partial_ratio(name.lower(), result['data'][season][episode]['name'].lower()) > 90:
return (season, episode, result['data'][season][episode]['name'])
elif description is not None:
descriptionQueryResult = self.FindEpisodeByDescription(showId, season, episode, description)
if descriptionQueryResult is not None:
@@ -63,8 +65,18 @@ class Sickbeard:
return (0, 0, '')
def GetEpisodeName(subtitle, showName):
if subtitle[:len(showName)].lower() == showName.lower():
return subtitle[len(showName + ' and the '):]
else:
return subtitle
#==============================================================================
# def GetEpisodeName(subtitle, showName):
# if subtitle[:len(showName)].lower() == showName.lower():
# return subtitle[len(showName + ' and the '):]
# else:
# return subtitle
#==============================================================================
def FixEpisodeTitle(self, showName, episodeTitle):
sickbeardPrefix = self.__settings.GetShowSickbeardEpisodePrefix(showName)
if sickbeardPrefix != "":
if not episodeTitle.lower.startswith(sickbeardPrefix.lower()):
return "{0} {1}".format(sickbeardPrefix.rstrip(), episodeTitle.lstrip())
return episodeTitle