Browse Source

Added optional parameter 'include_identifier' to the 'searcher.get_search_title' event handler.

pull/2570/head
Dean Gardiner 12 years ago
parent
commit
2fd54901e7
  1. 2
      couchpotato/core/media/movie/searcher/main.py
  2. 6
      couchpotato/core/media/show/searcher/main.py
  3. 4
      couchpotato/core/providers/base.py
  4. 4
      couchpotato/core/providers/nzb/newznab/main.py
  5. 4
      couchpotato/core/providers/torrent/sceneaccess/main.py

2
couchpotato/core/media/movie/searcher/main.py

@ -343,7 +343,7 @@ class MovieSearcher(SearcherBase, MovieTypeBase):
log.error('Failed searching for next release: %s', traceback.format_exc()) log.error('Failed searching for next release: %s', traceback.format_exc())
return False return False
def getSearchTitle(self, library): def getSearchTitle(self, library, include_identifier = False):
if library['type'] == 'movie': if library['type'] == 'movie':
return getTitle(library) return getTitle(library)

6
couchpotato/core/media/show/searcher/main.py

@ -133,7 +133,7 @@ class ShowSearcher(Plugin):
return ret return ret
def getSearchTitle(self, library): def getSearchTitle(self, library, include_identifier = False):
if library['type'] not in ['show', 'season', 'episode']: if library['type'] not in ['show', 'season', 'episode']:
return return
@ -172,6 +172,10 @@ class ShowSearcher(Plugin):
else: else:
return None return None
# Return show title if we aren't including the identifier
if not include_identifier:
return title
# Add the identifier to search title # Add the identifier to search title
identifier = fireEvent('library.identifier', library, single = True) identifier = fireEvent('library.identifier', library, single = True)

4
couchpotato/core/providers/base.py

@ -207,7 +207,9 @@ class YarrProvider(Provider):
self._search(media, quality, results) self._search(media, quality, results)
# Search possible titles # Search possible titles
else: else:
for title in possibleTitles(fireEvent('searcher.get_search_title', media['library'], single = True)): media_title = fireEvent('searcher.get_search_title', media['library'], include_identifier = True, single = True)
for title in possibleTitles(media_title):
self._searchOnTitle(title, media, quality, results) self._searchOnTitle(title, media, quality, results)
return results return results

4
couchpotato/core/providers/nzb/newznab/main.py

@ -191,7 +191,7 @@ class Movie(MovieProvider, Base):
class Season(SeasonProvider, Base): class Season(SeasonProvider, Base):
def buildUrl(self, media, api_key): def buildUrl(self, media, api_key):
search_title = fireEvent('searcher.get_search_title', media['library']['root_library']) search_title = fireEvent('searcher.get_search_title', media['library'])
identifier = fireEvent('library.identifier', media['library']) identifier = fireEvent('library.identifier', media['library'])
query = tryUrlencode({ query = tryUrlencode({
@ -206,7 +206,7 @@ class Season(SeasonProvider, Base):
class Episode(EpisodeProvider, Base): class Episode(EpisodeProvider, Base):
def buildUrl(self, media, api_key): def buildUrl(self, media, api_key):
search_title = fireEvent('searcher.get_search_title', media['library']['root_library']) search_title = fireEvent('searcher.get_search_title', media['library'])
identifier = fireEvent('library.identifier', media['library']) identifier = fireEvent('library.identifier', media['library'])
query = tryUrlencode({ query = tryUrlencode({

4
couchpotato/core/providers/torrent/sceneaccess/main.py

@ -122,7 +122,7 @@ class Season(SeasonProvider, Base):
] ]
def buildUrl(self, media, quality): def buildUrl(self, media, quality):
return self._buildUrl(fireEvent('searcher.get_search_title', media['library']['root_library']), quality['identifier']) return self._buildUrl(fireEvent('searcher.get_search_title', media['library']), quality['identifier'])
class Episode(EpisodeProvider, Base): class Episode(EpisodeProvider, Base):
@ -132,4 +132,4 @@ class Episode(EpisodeProvider, Base):
] ]
def buildUrl(self, media, quality): def buildUrl(self, media, quality):
return self._buildUrl(fireEvent('searcher.get_search_title', media['library']['root_library']), quality['identifier']) return self._buildUrl(fireEvent('searcher.get_search_title', media['library']), quality['identifier'])
Loading…
Cancel
Save