mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-03 01:43:37 +00:00
68fa29e77a
Add get_episode_data in tvdb_api v2 Usage: lINDEXER_API_PARMS = sickbeard.indexerApi(show.indexer).api_params.copy() t = sickbeard.indexerApi(show.indexer).indexer(**lINDEXER_API_PARMS) ep_data = t.get_episode_data(episodeid)
137 lines
4.2 KiB
Python
137 lines
4.2 KiB
Python
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# This file is part of SickGear.
|
|
#
|
|
# SickGear is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# SickGear is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with SickGear. If not, see <http://www.gnu.org/licenses/>.
|
|
import os
|
|
import sickbeard
|
|
import time
|
|
|
|
from indexer_config import initConfig, indexerConfig
|
|
from sickbeard.helpers import proxy_setting
|
|
|
|
|
|
class ShowContainer(dict):
|
|
"""Simple dict that holds a series of Show instances
|
|
"""
|
|
|
|
def __init__(self):
|
|
self._stack = []
|
|
self._lastgc = time.time()
|
|
|
|
def __setitem__(self, key, value):
|
|
self._stack.append(key)
|
|
|
|
# keep only the 100th latest results
|
|
if time.time() - self._lastgc > 20:
|
|
for o in self._stack[:-100]:
|
|
del self[o]
|
|
|
|
self._stack = self._stack[-100:]
|
|
|
|
self._lastgc = time.time()
|
|
|
|
super(ShowContainer, self).__setitem__(key, value)
|
|
|
|
|
|
class DummyIndexer:
|
|
def __init__(self, *args, **kwargs):
|
|
self.config = {
|
|
'apikey': '',
|
|
'debug_enabled': False,
|
|
'custom_ui': None,
|
|
'proxy': None,
|
|
'cache_enabled': False,
|
|
'cache_location': '',
|
|
'valid_languages': [],
|
|
'langabbv_to_id': {},
|
|
'language': 'en',
|
|
'base_url': '',
|
|
}
|
|
|
|
self.corrections = {}
|
|
self.shows = ShowContainer()
|
|
|
|
def __getitem__(self, key):
|
|
return None
|
|
|
|
def __repr__(self):
|
|
return str(self.shows)
|
|
|
|
def search(self, series):
|
|
return []
|
|
|
|
|
|
class indexerApi(object):
|
|
def __init__(self, indexerID=None):
|
|
self.indexerID = int(indexerID) if indexerID else None
|
|
|
|
def __del__(self):
|
|
pass
|
|
|
|
def indexer(self, *args, **kwargs):
|
|
if self.indexerID:
|
|
if indexerConfig[self.indexerID]['active'] or ('no_dummy' in kwargs and True is kwargs['no_dummy']):
|
|
if 'no_dummy' in kwargs:
|
|
kwargs.pop('no_dummy')
|
|
return indexerConfig[self.indexerID]['module'](*args, **kwargs)
|
|
else:
|
|
return DummyIndexer(*args, **kwargs)
|
|
|
|
@property
|
|
def config(self):
|
|
if self.indexerID:
|
|
return indexerConfig[self.indexerID]
|
|
return initConfig
|
|
|
|
@property
|
|
def name(self):
|
|
if self.indexerID:
|
|
return indexerConfig[self.indexerID]['name']
|
|
|
|
@property
|
|
def api_params(self):
|
|
if self.indexerID:
|
|
if sickbeard.CACHE_DIR:
|
|
indexerConfig[self.indexerID]['api_params']['cache'] = os.path.join(
|
|
sickbeard.CACHE_DIR, 'indexers', self.name)
|
|
if sickbeard.PROXY_SETTING and sickbeard.PROXY_INDEXERS:
|
|
(proxy_address, pac_found) = proxy_setting(sickbeard.PROXY_SETTING,
|
|
indexerConfig[self.indexerID]['base_url'],
|
|
force=True)
|
|
if proxy_address:
|
|
indexerConfig[self.indexerID]['api_params']['proxy'] = proxy_address
|
|
|
|
return indexerConfig[self.indexerID]['api_params']
|
|
|
|
@property
|
|
def cache(self):
|
|
if sickbeard.CACHE_DIR:
|
|
return self.api_params['cache']
|
|
|
|
@property
|
|
def indexers(self):
|
|
return dict((int(x['id']), x['name']) for x in indexerConfig.values() if not x['mapped_only'])
|
|
|
|
@property
|
|
def all_indexers(self):
|
|
"""
|
|
return all indexers including mapped only indexers
|
|
"""
|
|
return dict((int(x['id']), x['name']) for x in indexerConfig.values())
|
|
|
|
@property
|
|
def xem_supported_indexers(self):
|
|
return dict((int(x['id']), x['name']) for x in indexerConfig.values() if x.get('xem_origin'))
|