Added optional parameter 'include_identifier' to the 'searcher.get_search_title' event handler.
This commit is contained in:
@@ -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)
|
||||||
|
|
||||||
|
|||||||
@@ -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)
|
||||||
|
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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({
|
||||||
|
|||||||
@@ -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'])
|
||||||
Reference in New Issue
Block a user