Browse Source

[TV] Added "full_search" placeholder methods to avoid errors on startup

pull/3730/merge
Dean Gardiner 11 years ago
parent
commit
bda44848a1
  1. 17
      couchpotato/core/media/show/searcher/episode.py
  2. 18
      couchpotato/core/media/show/searcher/season.py
  3. 19
      couchpotato/core/media/show/searcher/show.py

17
couchpotato/core/media/show/searcher/episode.py

@ -1,6 +1,6 @@
from couchpotato import fireEvent, get_db, Env from couchpotato import fireEvent, get_db, Env
from couchpotato.api import addApiView from couchpotato.api import addApiView
from couchpotato.core.event import addEvent from couchpotato.core.event import addEvent, fireEventAsync
from couchpotato.core.logger import CPLog from couchpotato.core.logger import CPLog
from couchpotato.core.media._base.searcher.base import SearcherBase from couchpotato.core.media._base.searcher.base import SearcherBase
from couchpotato.core.media._base.searcher.main import SearchSetupError from couchpotato.core.media._base.searcher.main import SearchSetupError
@ -19,11 +19,26 @@ class EpisodeSearcher(SearcherBase, ShowTypeBase):
def __init__(self): def __init__(self):
super(EpisodeSearcher, self).__init__() super(EpisodeSearcher, self).__init__()
addEvent('%s.searcher.all' % self.getType(), self.searchAll)
addEvent('%s.searcher.single' % self.getType(), self.single) addEvent('%s.searcher.single' % self.getType(), self.single)
addEvent('searcher.correct_release', self.correctRelease) addEvent('searcher.correct_release', self.correctRelease)
addApiView('%s.searcher.full_search' % self.getType(), self.searchAllView, docs = {
'desc': 'Starts a full search for all wanted shows',
})
addApiView('%s.searcher.single' % self.getType(), self.singleView) addApiView('%s.searcher.single' % self.getType(), self.singleView)
def searchAllView(self, **kwargs):
fireEventAsync('%s.searcher.all' % self.getType(), manual = True)
return {
'success': not self.in_progress
}
def searchAll(self, manual = False):
pass
def singleView(self, media_id, **kwargs): def singleView(self, media_id, **kwargs):
db = get_db() db = get_db()
media = db.get('id', media_id) media = db.get('id', media_id)

18
couchpotato/core/media/show/searcher/season.py

@ -1,4 +1,5 @@
from couchpotato.core.event import addEvent from couchpotato.api import addApiView
from couchpotato.core.event import addEvent, fireEventAsync
from couchpotato.core.logger import CPLog from couchpotato.core.logger import CPLog
from couchpotato.core.media._base.searcher.base import SearcherBase from couchpotato.core.media._base.searcher.base import SearcherBase
from couchpotato.core.media.show import ShowTypeBase from couchpotato.core.media.show import ShowTypeBase
@ -16,8 +17,23 @@ class SeasonSearcher(SearcherBase, ShowTypeBase):
def __init__(self): def __init__(self):
super(SeasonSearcher, self).__init__() super(SeasonSearcher, self).__init__()
addEvent('%s.searcher.all' % self.getType(), self.searchAll)
addEvent('%s.searcher.single' % self.getType(), self.single) addEvent('%s.searcher.single' % self.getType(), self.single)
addApiView('%s.searcher.full_search' % self.getType(), self.searchAllView, docs = {
'desc': 'Starts a full search for all wanted seasons',
})
def searchAllView(self, **kwargs):
fireEventAsync('%s.searcher.all' % self.getType(), manual = True)
return {
'success': not self.in_progress
}
def searchAll(self, manual = False):
pass
def single(self, media, show, profile): def single(self, media, show, profile):
# Check if any episode is already snatched # Check if any episode is already snatched

19
couchpotato/core/media/show/searcher/show.py

@ -1,5 +1,6 @@
from couchpotato import get_db from couchpotato import get_db
from couchpotato.core.event import fireEvent, addEvent from couchpotato.api import addApiView
from couchpotato.core.event import fireEvent, addEvent, fireEventAsync
from couchpotato.core.helpers.variable import getTitle from couchpotato.core.helpers.variable import getTitle
from couchpotato.core.logger import CPLog from couchpotato.core.logger import CPLog
from couchpotato.core.media._base.searcher.base import SearcherBase from couchpotato.core.media._base.searcher.base import SearcherBase
@ -19,10 +20,24 @@ class ShowSearcher(SearcherBase, ShowTypeBase):
def __init__(self): def __init__(self):
super(ShowSearcher, self).__init__() super(ShowSearcher, self).__init__()
addEvent('%s.searcher.all' % self.getType(), self.searchAll)
addEvent('%s.searcher.single' % self.getType(), self.single) addEvent('%s.searcher.single' % self.getType(), self.single)
addEvent('searcher.get_search_title', self.getSearchTitle) addEvent('searcher.get_search_title', self.getSearchTitle)
addApiView('%s.searcher.full_search' % self.getType(), self.searchAllView, docs = {
'desc': 'Starts a full search for all wanted episodes',
})
def searchAllView(self, **kwargs):
fireEventAsync('%s.searcher.all' % self.getType(), manual = True)
return {
'success': not self.in_progress
}
def searchAll(self, manual = False):
pass
def single(self, media, search_protocols = None, manual = False): def single(self, media, search_protocols = None, manual = False):
# Find out search type # Find out search type
try: try:

Loading…
Cancel
Save