# Author: Nic Wolfe # 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 . 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'))