def __init__(self):
self.config = CanapeConfig()
self.db = Canapedb(self.config['VIDEOS_DB'],
self.config['tvshow']['subtitles'],
self.config['tvshow']['quality'])
#Load searcher object
self.video = Video(self.config['sources']['video'])
self.subtitle = Subtitle(self.config['sources']['subtitle'])
self.information = Information()
#Load chooser object
self.videochooser = VideoChooser(self.config['QUALITIES_DB'])
self.subtitlechooser = SubtitleChooser()
#Load downloader object
self.downloader = Downloader(config=self.config.get('downloader', {}))
class Canape(object):
def __init__(self):
self.config = CanapeConfig()
self.db = Canapedb(self.config['VIDEOS_DB'],
self.config['tvshow']['subtitles'],
self.config['tvshow']['quality'])
#Load searcher object
self.video = Video(self.config['sources']['video'])
self.subtitle = Subtitle(self.config['sources']['subtitle'])
self.information = Information()
#Load chooser object
self.videochooser = VideoChooser(self.config['QUALITIES_DB'])
self.subtitlechooser = SubtitleChooser()
#Load downloader object
self.downloader = Downloader(config=self.config.get('downloader', {}))
def daemon_run(self):
"""
Launch periodically :meth:`Canape.check`.
The launch interval is determined by the config file::
[tvshow]
check_interval = TIME IN MINUTES
"""
if self.config['xmlrpc']['start']:
server = CanapeXMLRPCServer(self.config['xmlrpc']['hostname'],
self.config['xmlrpc']['port'])
server.start()
t = self.config['tvshow']['check_interval'] * 60
while True:
self.check()
time.sleep(t)
def check(self):
"""
Process all series from database:
1. First search for new episode with :meth:`Canape.updateEpisodes`
2. Secondly search and download them with :meth:`Canape.getEpisodeDownload`
3. Thirdly search and download subtitles with :meth:`Canape.getEpisodeSubtitles`
"""
LOGGER.info("Start checking")
self.downloader.check()
for serie in self.db.get_series():
LOGGER.info('Process serie: %s' % serie.name)
serie = self.updateEpisodes(serie)
updated_episodes = []
for episode in serie.episodes:
LOGGER.info('Process episode: S{0:0>2}E{1:0>2}'.format(episode.snum, episode.enum) )
if episode.is_downloading():
isfinished=False
id_ = "%sS%sE%s" % (serie.id_, episode.snum, episode.enum)
LOGGER.debug("Episode %s is downloading check that" % id_)
try:
isfinished = self.downloader.is_finished(id_)
except UnknownDownload:
LOGGER.error("Episode %s unknown from downloader guess it's downloaded" % id_ )
episode.set_downloaded()
if isfinished:
LOGGER.debug("Episode %s download finished" % id_)
episode.set_downloaded()
elif not episode.is_downloaded():
LOGGER.info('Search for video')
video = self.getEpisodeDownload(serie, episode)
if video is not None:
video.id_ = "%sS%sE%s" % (serie.id_, episode.snum, episode.enum)
self.downloader.addVideo(video)
episode.set_downloading()
if not episode.subtitle_downloaded():
LOGGER.info('Search for subtitles')
subtitle = self.getEpisodeSubtitles(serie, episode, video)
if subtitle is not None:
self.downloader.addSubtitle(subtitle)
episode.set_subtitle_downloaded()
updated_episodes.append(episode)
serie.episodes = updated_episodes
self.db.update_serie(serie)
def updateEpisodes(self, serieObj):
"""
Update serieObj with new avalaible episodes return updated serieObj
serieObj must be an instance of the :class:`canape.object.Serie` class.
Get information by :py:class:`canape.information.Searcher` and test airdate.
"""
serieObj.sort()
lastest_ep = serieObj.episodes[-1]
season_episodes= self.information.get_episodes(serieObj, lastest_ep.snum)
time_delta = datetime.timedelta( hours=self.config['tvshow']['check_timedelta'] )
for enum in season_episodes[lastest_ep.enum:]:
airdate = self.information.get_airdate(serieObj, lastest_ep.snum, enum)
if airdate + time_delta <= datetime.datetime.now():
serieObj.episodes.append( Episode(lastest_ep.snum, enum) )
return serieObj
def getEpisodeDownload(self, serieObj, episodeObj):
"""
Try to retrieve a download link for a specified serie's episode.
Return an instance of the :class:`canape.video.video.Video` class or
None.
* serieObj must be an instance of the :class:`canape.object.Serie` class.
* episodeObj must be an instance of the :class:`canape.object.Episode` class.
Use :py:class:`canape.video.Searcher` for find video link and
:py:class:`canape.chooser.VideoChooser` to choose the best.
"""
vresults = self.video.tvshow_search(serieObj, episodeObj)
if len(vresults):
return self.videochooser.choose(vresults)
else:
return None
def getEpisodeSubtitles(self, serieObj, episodeObj, videoObj=None):
"""
Try to retrieve subtitles for a specified serie's episode.
Return an instance of the :class:`canape.subtitle.subtitle.Subtitle` class or
None.
* serieObj must be an instance of the :py:class:`canape.object.Serie` class.
* episodeObj must be an instance of the :py:class:`canape.object.Episode` class.
* videoObj must be an instance of the :py:class:`canape.chooser.VideoChooser`
Use :py:class:`canape.subtitle.Searcher` for find subtitle and
:py:class:`canape.chooser.SubtitleChooser` to choose the best.
"""
subtitles = self.subtitle.tvshow_search(serieObj, episodeObj)
if len(subtitles):
return self.subtitlechooser.choose(subtitles, videoObj)
else:
return None