mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-14 05:15:12 -05:00
117 lines
5.0 KiB
Python
117 lines
5.0 KiB
Python
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# This file is part of SickRage.
|
|
#
|
|
# SickRage 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.
|
|
#
|
|
# SickRage 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 SickRage. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import datetime
|
|
import os
|
|
|
|
import sickbeard
|
|
|
|
from sickbeard import logger
|
|
from sickbeard import exceptions
|
|
from sickbeard import ui
|
|
from sickbeard.exceptions import ex
|
|
from sickbeard import encodingKludge as ek
|
|
from sickbeard import db
|
|
from sickbeard import network_timezones
|
|
from sickbeard import failed_history
|
|
|
|
class ShowUpdater():
|
|
def __init__(self):
|
|
self.updateInterval = datetime.timedelta(hours=1)
|
|
|
|
def run(self, force=False):
|
|
|
|
# update at 3 AM
|
|
run_updater_time = datetime.time(hour=3)
|
|
|
|
update_datetime = datetime.datetime.now()
|
|
update_date = update_datetime.date()
|
|
|
|
logger.log(u"Checking update interval", logger.DEBUG)
|
|
|
|
hour_diff = update_datetime.time().hour - run_updater_time.hour
|
|
|
|
# if it's less than an interval after the update time then do an update (or if we're forcing it)
|
|
if hour_diff >= 0 and hour_diff < self.updateInterval.seconds / 3600 or force:
|
|
logger.log(u"Doing full update on all shows")
|
|
else:
|
|
return
|
|
|
|
# clean out cache directory, remove everything > 12 hours old
|
|
if sickbeard.CACHE_DIR:
|
|
for indexer in sickbeard.indexerApi().indexers:
|
|
cache_dir = sickbeard.indexerApi(indexer).cache
|
|
logger.log(u"Trying to clean cache folder " + cache_dir)
|
|
|
|
# Does our cache_dir exists
|
|
if not ek.ek(os.path.isdir, cache_dir):
|
|
logger.log(u"Can't clean " + cache_dir + " if it doesn't exist", logger.WARNING)
|
|
else:
|
|
max_age = datetime.timedelta(hours=12)
|
|
# Get all our cache files
|
|
cache_files = ek.ek(os.listdir, cache_dir)
|
|
|
|
for cache_file in cache_files:
|
|
cache_file_path = ek.ek(os.path.join, cache_dir, cache_file)
|
|
|
|
if ek.ek(os.path.isfile, cache_file_path):
|
|
cache_file_modified = datetime.datetime.fromtimestamp(
|
|
ek.ek(os.path.getmtime, cache_file_path))
|
|
|
|
if update_datetime - cache_file_modified > max_age:
|
|
try:
|
|
ek.ek(os.remove, cache_file_path)
|
|
except OSError, e:
|
|
logger.log(u"Unable to clean " + cache_dir + ": " + repr(e) + " / " + str(e),
|
|
logger.WARNING)
|
|
break
|
|
|
|
# select 10 'Ended' tv_shows updated more than 90 days ago to include in this update
|
|
stale_should_update = []
|
|
stale_update_date = (update_date - datetime.timedelta(days=90)).toordinal()
|
|
|
|
# last_update_date <= 90 days, sorted ASC because dates are ordinal
|
|
with db.DBConnection() as myDB:
|
|
sql_result = myDB.select(
|
|
"SELECT indexer_id FROM tv_shows WHERE status = 'Ended' AND last_update_indexer <= ? ORDER BY last_update_indexer ASC LIMIT 10;",
|
|
[stale_update_date])
|
|
|
|
for cur_result in sql_result:
|
|
stale_should_update.append(int(cur_result['indexer_id']))
|
|
|
|
# start update process
|
|
piList = []
|
|
for curShow in sickbeard.showList:
|
|
|
|
try:
|
|
# if should_update returns True (not 'Ended') or show is selected stale 'Ended' then update, otherwise just refresh
|
|
if curShow.should_update(update_date=update_date) or curShow.indexerid in stale_should_update:
|
|
curQueueItem = sickbeard.showQueueScheduler.action.updateShow(curShow, True) # @UndefinedVariable
|
|
else:
|
|
logger.log(
|
|
u"Not updating episodes for show " + curShow.name + " because it's marked as ended and last/next episode is not within the grace period.",
|
|
logger.DEBUG)
|
|
curQueueItem = sickbeard.showQueueScheduler.action.refreshShow(curShow, True) # @UndefinedVariable
|
|
|
|
piList.append(curQueueItem)
|
|
|
|
except (exceptions.CantUpdateException, exceptions.CantRefreshException), e:
|
|
logger.log(u"Automatic update failed: " + ex(e), logger.ERROR)
|
|
|
|
ui.ProgressIndicators.setIndicator('dailyUpdate', ui.QueueProgressIndicator("Daily Update", piList))
|