|
|
|
|
#!/usr/bin/env python2
|
|
|
|
|
# encoding:utf-8
|
|
|
|
|
# project:indexer_api
|
|
|
|
|
# license:unlicense (http://unlicense.org/)
|
|
|
|
|
|
|
|
|
|
__all__ = ['check_exception_type', 'ExceptionTuples',
|
|
|
|
|
'BaseTVinfoException', 'BaseTVinfoError', 'BaseTVinfoAuthenticationerror',
|
|
|
|
|
'BaseTVinfoUserabort', 'BaseTVinfoAttributenotfound', 'BaseTVinfoEpisodenotfound',
|
|
|
|
|
'BaseTVinfoSeasonnotfound', 'BaseTVinfoShownotfound']
|
|
|
|
|
|
|
|
|
|
"""Custom exceptions used or raised by indexer_api"""
|
|
|
|
|
|
|
|
|
|
from lib.tvdb_api.tvdb_exceptions import \
|
|
|
|
|
TvdbException, TvdbAttributenotfound, TvdbEpisodenotfound, TvdbError, \
|
|
|
|
|
TvdbSeasonnotfound, TvdbShownotfound, TvdbUserabort, TvdbTokenexpired
|
|
|
|
|
|
|
|
|
|
indexer_excepts = [
|
|
|
|
|
'tvinfo_exception', 'tvinfo_error', 'tvinfo_userabort',
|
|
|
|
|
'tvinfo_shownotfound', 'tvinfo_seasonnotfound', 'tvinfo_episodenotfound',
|
|
|
|
|
'tvinfo_attributenotfound', 'tvinfo_authenticationerror']
|
|
|
|
|
|
|
|
|
|
tvdb_excepts = [
|
|
|
|
|
'tvdb_exception', 'tvdb_error', 'tvdb_userabort', 'tvdb_shownotfound',
|
|
|
|
|
'tvdb_seasonnotfound', 'tvdb_episodenotfound', 'tvdb_attributenotfound',
|
|
|
|
|
'tvdb_tokenexpired']
|
|
|
|
|
|
|
|
|
|
tvdbv1_excepts = [
|
|
|
|
|
'tvdb_exception_v1', 'tvdb_error_v1', 'tvdb_userabort_v1', 'tvdb_shownotfound_v1',
|
|
|
|
|
'tvdb_seasonnotfound_v1', 'tvdb_episodenotfound_v1', 'tvdb_attributenotfound_v1']
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BaseTVinfoException(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BaseTVinfoError(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BaseTVinfoAuthenticationerror(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BaseTVinfoUserabort(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BaseTVinfoAttributenotfound(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BaseTVinfoEpisodenotfound(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BaseTVinfoSeasonnotfound(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BaseTVinfoShownotfound(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# link API exceptions to our exception handler
|
|
|
|
|
class ExceptionTuples:
|
|
|
|
|
tvinfo_exception = TvdbException, BaseTVinfoException
|
|
|
|
|
tvinfo_error = TvdbError, BaseTVinfoError
|
|
|
|
|
tvinfo_authenticationerror = TvdbTokenexpired, BaseTVinfoAuthenticationerror
|
|
|
|
|
tvinfo_userabort = TvdbUserabort, BaseTVinfoUserabort
|
|
|
|
|
tvinfo_attributenotfound = TvdbAttributenotfound, BaseTVinfoAttributenotfound
|
|
|
|
|
tvinfo_episodenotfound = TvdbEpisodenotfound, BaseTVinfoEpisodenotfound
|
|
|
|
|
tvinfo_seasonnotfound = TvdbSeasonnotfound, BaseTVinfoSeasonnotfound
|
|
|
|
|
tvinfo_shownotfound = TvdbShownotfound, BaseTVinfoShownotfound
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def check_exception_type(ex, ex_class, *args):
|
|
|
|
|
# type: (...) -> bool
|
|
|
|
|
if issubclass(ex.__class__, ex_class + args):
|
|
|
|
|
return True
|
|
|
|
|
return False
|
