GenericProvider.TORRENT: {p.id: p for p in TorrentProvider.getProviderList()},
}
sickrage.srCore.INDEXER_API = srIndexerApi
sickrage.srCore.SCHEDULER = srScheduler()
sickrage.srCore.NAMECACHE = srNameCache()
sickrage.srCore.NOTIFIERS = AttrDict(
libnotify=LibnotifyNotifier(),
kodi_notifier=KODINotifier(),
plex_notifier=PLEXNotifier(),
emby_notifier=EMBYNotifier(),
nmj_notifier=NMJNotifier(),
nmjv2_notifier=NMJv2Notifier(),
synoindex_notifier=synoIndexNotifier(),
synology_notifier=synologyNotifier(),
pytivo_notifier=pyTivoNotifier(),
growl_notifier=GrowlNotifier(),
prowl_notifier=ProwlNotifier(),
libnotify_notifier=LibnotifyNotifier(),
pushover_notifier=PushoverNotifier(),
boxcar_notifier=BoxcarNotifier(),
boxcar2_notifier=Boxcar2Notifier(),
nma_notifier=NMA_Notifier(),
pushalot_notifier=PushalotNotifier(),
pushbullet_notifier=PushbulletNotifier(),
freemobile_notifier=FreeMobileNotifier(),
twitter_notifier=TwitterNotifier(),
trakt_notifier=TraktNotifier(),
email_notifier=EmailNotifier()
)
def __init__(self):
self.STARTED = False
self.PID = None
# init system encoding
self.SYS_ENCODING = encodingInit()
# Check if we need to perform a restore first
os.chdir(sickrage.DATA_DIR)
restore_dir = os.path.join(sickrage.DATA_DIR, 'restore')
if os.path.exists(restore_dir):
success = restoreDB(restore_dir, sickrage.DATA_DIR)
print("Restore: restoring DB and config.ini %s!\n" % ("FAILED", "SUCCESSFUL")[success])
if success:
os.execl(sys.executable, sys.executable, *sys.argv)
# sickrage version
self.VERSION = None
self.NEWEST_VERSION = None
self.NEWEST_VERSION_STRING = None
# show list
self.SHOWLIST = []
# notifiers
self.NOTIFIERS = AttrDict(
libnotify=LibnotifyNotifier(),
kodi_notifier=KODINotifier(),
plex_notifier=PLEXNotifier(),
emby_notifier=EMBYNotifier(),
nmj_notifier=NMJNotifier(),
nmjv2_notifier=NMJv2Notifier(),
synoindex_notifier=synoIndexNotifier(),
synology_notifier=synologyNotifier(),
pytivo_notifier=pyTivoNotifier(),
growl_notifier=GrowlNotifier(),
prowl_notifier=ProwlNotifier(),
libnotify_notifier=LibnotifyNotifier(),
pushover_notifier=PushoverNotifier(),
boxcar_notifier=BoxcarNotifier(),
boxcar2_notifier=Boxcar2Notifier(),
nma_notifier=NMA_Notifier(),
pushalot_notifier=PushalotNotifier(),
pushbullet_notifier=PushbulletNotifier(),
freemobile_notifier=FreeMobileNotifier(),
twitter_notifier=TwitterNotifier(),
trakt_notifier=TraktNotifier(),
email_notifier=EmailNotifier()
)
# services
self.SCHEDULER = None
self.WEBSERVER = None
self.INDEXER_API = None
self.VERSIONUPDATER = None
# init caches
self.NAMECACHE = None
# init queues
self.SHOWUPDATER = None
self.SHOWQUEUE = None
self.SEARCHQUEUE = None
# init searchers
self.DAILYSEARCHER = None
self.BACKLOGSEARCHER = None
self.PROPERSEARCHER = None
self.TRAKTSEARCHER = None
self.SUBTITLESEARCHER = None
# init postprocessor
self.AUTOPOSTPROCESSOR = None
self.metadataProviderDict = None
self.providersDict = {
GenericProvider.NZB: {p.id: p for p in NZBProvider.getProviderList()},
GenericProvider.TORRENT: {p.id: p for p in TorrentProvider.getProviderList()},
}
self.newznabProviderList = None
self.torrentRssProviderList = None
def start(self):
self.PID = os.getpid()
# notifiers
self.NOTIFIERS = AttrDict(
libnotify=LibnotifyNotifier(),
kodi_notifier=KODINotifier(),
plex_notifier=PLEXNotifier(),
emby_notifier=EMBYNotifier(),
nmj_notifier=NMJNotifier(),
nmjv2_notifier=NMJv2Notifier(),
synoindex_notifier=synoIndexNotifier(),
synology_notifier=synologyNotifier(),
pytivo_notifier=pyTivoNotifier(),
growl_notifier=GrowlNotifier(),
prowl_notifier=ProwlNotifier(),
libnotify_notifier=LibnotifyNotifier(),
pushover_notifier=PushoverNotifier(),
boxcar_notifier=BoxcarNotifier(),
boxcar2_notifier=Boxcar2Notifier(),
nma_notifier=NMA_Notifier(),
pushalot_notifier=PushalotNotifier(),
pushbullet_notifier=PushbulletNotifier(),
freemobile_notifier=FreeMobileNotifier(),
twitter_notifier=TwitterNotifier(),
trakt_notifier=TraktNotifier(),
email_notifier=EmailNotifier()
)
# set socket timeout
socket.setdefaulttimeout(sickrage.srConfig.SOCKET_TIMEOUT)
# init version updater
self.VERSIONUPDATER = srVersionUpdater()
# init updater and get current version
self.VERSION = self.VERSIONUPDATER.updater.version
# init services
self.SCHEDULER = srScheduler()
self.WEBSERVER = srWebServer()
self.INDEXER_API = srIndexerApi
# init caches
self.NAMECACHE = srNameCache()
# init queues
self.SHOWUPDATER = srShowUpdater()
self.SHOWQUEUE = srShowQueue()
self.SEARCHQUEUE = srSearchQueue()
# init searchers
self.DAILYSEARCHER = srDailySearcher()
self.BACKLOGSEARCHER = srBacklogSearcher()
self.PROPERSEARCHER = srProperSearcher()
self.TRAKTSEARCHER = srTraktSearcher()
self.SUBTITLESEARCHER = srSubtitleSearcher()
# init postprocessor
self.AUTOPOSTPROCESSOR = srPostProcessor()
# migrate old database file names to new ones
if not os.path.exists(main_db.MainDB().filename) and os.path.exists("sickbeard.db"):
helpers.moveFile("sickbeard.db", main_db.MainDB().filename)
# initialize the main SB database
main_db.MainDB().InitialSchema().upgrade()
# initialize the cache database
cache_db.CacheDB().InitialSchema().upgrade()
# initialize the failed downloads database
failed_db.FailedDB().InitialSchema().upgrade()
# fix up any db problems
main_db.MainDB().SanityCheck()
# load data for shows from database
self.load_shows()
if sickrage.srConfig.DEFAULT_PAGE not in ('home', 'schedule', 'history', 'news', 'IRC'):
sickrage.srConfig.DEFAULT_PAGE = 'home'
if not makeDir(sickrage.srConfig.CACHE_DIR):
sickrage.srLogger.error("!!! Creating local cache dir failed")
sickrage.srConfig.CACHE_DIR = get_temp_dir()
# Check if we need to perform a restore of the cache folder
try:
restore_dir = os.path.join(self.DATA_DIR, 'restore')
if os.path.exists(restore_dir) and os.path.exists(os.path.join(restore_dir, 'cache')):
def restore_cache(src_dir, dst_dir):
def path_leaf(path):
head, tail = os.path.split(path)
return tail or os.path.basename(head)
try:
if os.path.isdir(dst_dir):
bak_filename = '{}-{}'.format(path_leaf(dst_dir), datetime.now().strftime('%Y%m%d_%H%M%S'))
shutil.move(dst_dir, os.path.join(os.path.dirname(dst_dir), bak_filename))
shutil.move(src_dir, dst_dir)
sickrage.srLogger.info("Restore: restoring cache successful")
except Exception as E:
sickrage.srLogger.error("Restore: restoring cache failed: {}".format(E.message))
restore_cache(os.path.join(restore_dir, 'cache'), sickrage.srConfig.CACHE_DIR)
except Exception as e:
sickrage.srLogger.error("Restore: restoring cache failed: {}".format(e.message))
finally:
if os.path.exists(os.path.join(self.DATA_DIR, 'restore')):
try:
removetree(os.path.join(self.DATA_DIR, 'restore'))
except Exception as e:
sickrage.srLogger.error("Restore: Unable to remove the restore directory: {}".format(e.message))
for cleanupDir in ['mako', 'sessions', 'indexers']:
try:
removetree(os.path.join(sickrage.srConfig.CACHE_DIR, cleanupDir))
except Exception as e:
sickrage.srLogger.warning(
"Restore: Unable to remove the cache/{} directory: {1}".format(cleanupDir, e))
if sickrage.srConfig.WEB_PORT < 21 or sickrage.srConfig.WEB_PORT > 65535:
sickrage.srConfig.WEB_PORT = 8081
if not sickrage.srConfig.WEB_COOKIE_SECRET:
sickrage.srConfig.WEB_COOKIE_SECRET = generateCookieSecret()
# attempt to help prevent users from breaking links by using a bad url
if not sickrage.srConfig.ANON_REDIRECT.endswith('?'):
sickrage.srConfig.ANON_REDIRECT = ''
if not re.match(r'\d+\|[^|]+(?:\|[^|]+)*', sickrage.srConfig.ROOT_DIRS):
sickrage.srConfig.ROOT_DIRS = ''
sickrage.srConfig.NAMING_FORCE_FOLDERS = check_force_season_folders()
if sickrage.srConfig.NZB_METHOD not in ('blackhole', 'sabnzbd', 'nzbget'):
sickrage.srConfig.NZB_METHOD = 'blackhole'
if not sickrage.srConfig.PROVIDER_ORDER:
sickrage.srConfig.PROVIDER_ORDER = self.providersDict[GenericProvider.NZB].keys() + \
self.providersDict[GenericProvider.TORRENT].keys()
if sickrage.srConfig.TORRENT_METHOD not in (
'blackhole', 'utorrent', 'transmission', 'deluge', 'deluged', 'download_station', 'rtorrent',
'qbittorrent', 'mlnet'):
sickrage.srConfig.TORRENT_METHOD = 'blackhole'
if sickrage.srConfig.PROPER_SEARCHER_INTERVAL not in ('15m', '45m', '90m', '4h', 'daily'):
sickrage.srConfig.PROPER_SEARCHER_INTERVAL = 'daily'
if sickrage.srConfig.AUTOPOSTPROCESSOR_FREQ < sickrage.srConfig.MIN_AUTOPOSTPROCESSOR_FREQ:
sickrage.srConfig.AUTOPOSTPROCESSOR_FREQ = sickrage.srConfig.MIN_AUTOPOSTPROCESSOR_FREQ
if sickrage.srConfig.NAMECACHE_FREQ < sickrage.srConfig.MIN_NAMECACHE_FREQ:
sickrage.srConfig.NAMECACHE_FREQ = sickrage.srConfig.MIN_NAMECACHE_FREQ
if sickrage.srConfig.DAILY_SEARCHER_FREQ < sickrage.srConfig.MIN_DAILY_SEARCHER_FREQ:
sickrage.srConfig.DAILY_SEARCHER_FREQ = sickrage.srConfig.MIN_DAILY_SEARCHER_FREQ
sickrage.srConfig.MIN_BACKLOG_SEARCHER_FREQ = get_backlog_cycle_time()
if sickrage.srConfig.BACKLOG_SEARCHER_FREQ < sickrage.srConfig.MIN_BACKLOG_SEARCHER_FREQ:
sickrage.srConfig.BACKLOG_SEARCHER_FREQ = sickrage.srConfig.MIN_BACKLOG_SEARCHER_FREQ
if sickrage.srConfig.VERSION_UPDATER_FREQ < sickrage.srConfig.MIN_VERSION_UPDATER_FREQ:
sickrage.srConfig.VERSION_UPDATER_FREQ = sickrage.srConfig.MIN_VERSION_UPDATER_FREQ
if sickrage.srConfig.SHOWUPDATE_HOUR > 23:
sickrage.srConfig.SHOWUPDATE_HOUR = 0
elif sickrage.srConfig.SHOWUPDATE_HOUR < 0:
sickrage.srConfig.SHOWUPDATE_HOUR = 0
if sickrage.srConfig.SUBTITLE_SEARCHER_FREQ < sickrage.srConfig.MIN_SUBTITLE_SEARCHER_FREQ:
sickrage.srConfig.SUBTITLE_SEARCHER_FREQ = sickrage.srConfig.MIN_SUBTITLE_SEARCHER_FREQ
sickrage.srConfig.NEWS_LATEST = sickrage.srConfig.NEWS_LAST_READ
if sickrage.srConfig.SUBTITLES_LANGUAGES[0] == '':
sickrage.srConfig.SUBTITLES_LANGUAGES = []
sickrage.srConfig.TIME_PRESET = sickrage.srConfig.TIME_PRESET_W_SECONDS.replace(":%S", "")
# initialize metadata_providers
self.metadataProviderDict = get_metadata_generator_dict()
for cur_metadata_tuple in [(sickrage.srConfig.METADATA_KODI, kodi),
(sickrage.srConfig.METADATA_KODI_12PLUS, kodi_12plus),
(sickrage.srConfig.METADATA_MEDIABROWSER, mediabrowser),
(sickrage.srConfig.METADATA_PS3, ps3),
(sickrage.srConfig.METADATA_WDTV, wdtv),
(sickrage.srConfig.METADATA_TIVO, tivo),
(sickrage.srConfig.METADATA_MEDE8ER, mede8er)]:
(cur_metadata_config, cur_metadata_class) = cur_metadata_tuple
tmp_provider = cur_metadata_class.metadata_class()
tmp_provider.set_config(cur_metadata_config)
self.metadataProviderDict[tmp_provider.name] = tmp_provider
# add version checker job to scheduler
self.SCHEDULER.add_job(
self.VERSIONUPDATER.run,
srIntervalTrigger(
**{'hours': sickrage.srConfig.VERSION_UPDATER_FREQ, 'min': sickrage.srConfig.MIN_VERSION_UPDATER_FREQ}),
name="VERSIONUPDATER",
id="VERSIONUPDATER",
replace_existing=True
)
# add network timezones updater job to scheduler
self.SCHEDULER.add_job(
update_network_dict,
srIntervalTrigger(**{'days': 1}),
name="TZUPDATER",
id="TZUPDATER",
replace_existing=True
)
# add namecache updater job to scheduler
self.SCHEDULER.add_job(
self.NAMECACHE.run,
srIntervalTrigger(**{'minutes': sickrage.srConfig.NAMECACHE_FREQ, 'min': sickrage.srConfig.MIN_NAMECACHE_FREQ}),
name="NAMECACHE",
id="NAMECACHE",
replace_existing=True
)
# add show queue job to scheduler
self.SCHEDULER.add_job(
self.SHOWQUEUE.run,
srIntervalTrigger(**{'seconds': 3}),
name="SHOWQUEUE",
id="SHOWQUEUE",
replace_existing=True
)
# add search queue job to scheduler
self.SCHEDULER.add_job(
self.SEARCHQUEUE.run,
srIntervalTrigger(**{'seconds': 1}),
name="SEARCHQUEUE",
id="SEARCHQUEUE",
replace_existing=True
)
# add show updater job to scheduler
self.SCHEDULER.add_job(
self.SHOWUPDATER.run,
srIntervalTrigger(
**{'hours': 1,
'start_date': datetime.now().replace(hour=sickrage.srConfig.SHOWUPDATE_HOUR)}),
name="SHOWUPDATER",
id="SHOWUPDATER",
replace_existing=True
)
# add daily search job to scheduler
self.SCHEDULER.add_job(
self.DAILYSEARCHER.run,
srIntervalTrigger(
**{'minutes': sickrage.srConfig.DAILY_SEARCHER_FREQ, 'min': sickrage.srConfig.MIN_DAILY_SEARCHER_FREQ}),
name="DAILYSEARCHER",
id="DAILYSEARCHER",
replace_existing=True
)
# add backlog search job to scheduler
self.SCHEDULER.add_job(
self.BACKLOGSEARCHER.run,
srIntervalTrigger(
**{'minutes': sickrage.srConfig.BACKLOG_SEARCHER_FREQ,
'min': sickrage.srConfig.MIN_BACKLOG_SEARCHER_FREQ}),
name="BACKLOG",
id="BACKLOG",
replace_existing=True
)
# add auto-postprocessing job to scheduler
job = self.SCHEDULER.add_job(
self.AUTOPOSTPROCESSOR.run,
srIntervalTrigger(**{'minutes': sickrage.srConfig.AUTOPOSTPROCESSOR_FREQ,
'min': sickrage.srConfig.MIN_AUTOPOSTPROCESSOR_FREQ}),
name="POSTPROCESSOR",
id="POSTPROCESSOR",
replace_existing=True
)
(job.pause, job.resume)[sickrage.srConfig.PROCESS_AUTOMATICALLY]()
# add find proper job to scheduler
job = self.SCHEDULER.add_job(
self.PROPERSEARCHER.run,
srIntervalTrigger(**{
'minutes': {'15m': 15, '45m': 45, '90m': 90, '4h': 4 * 60, 'daily': 24 * 60}[
sickrage.srConfig.PROPER_SEARCHER_INTERVAL]}),
name="PROPERSEARCHER",
id="PROPERSEARCHER",
replace_existing=True
)
(job.pause, job.resume)[sickrage.srConfig.DOWNLOAD_PROPERS]()
# add trakt.tv checker job to scheduler
job = self.SCHEDULER.add_job(
self.TRAKTSEARCHER.run,
srIntervalTrigger(**{'hours': 1}),
name="TRAKTSEARCHER",
id="TRAKTSEARCHER",
replace_existing=True,
)
(job.pause, job.resume)[sickrage.srConfig.USE_TRAKT]()
# add subtitles finder job to scheduler
job = self.SCHEDULER.add_job(
self.SUBTITLESEARCHER.run,
srIntervalTrigger(**{'hours': sickrage.srConfig.SUBTITLE_SEARCHER_FREQ}),
name="SUBTITLESEARCHER",
id="SUBTITLESEARCHER",
replace_existing=True
)
(job.pause, job.resume)[sickrage.srConfig.USE_SUBTITLES]()
# start scheduler
threading.Thread(None, self.SCHEDULER.start, 'SCHEDULER').start()
