407 lines
20 KiB
Python
407 lines
20 KiB
Python
# -*- coding: utf-8 -*-
|
|
import datetime
|
|
import math
|
|
import os
|
|
import sys
|
|
import threading
|
|
import traceback
|
|
import xbmc
|
|
import xbmcgui
|
|
from platformcode import config
|
|
librerias = xbmc.translatePath(os.path.join(config.get_runtime_path(), 'lib'))
|
|
sys.path.insert(0, librerias)
|
|
|
|
from core import videolibrarytools, filetools, channeltools
|
|
from lib import schedule
|
|
from platformcode import logger, platformtools, updater
|
|
from specials import videolibrary
|
|
from servers import torrent
|
|
|
|
|
|
def update(path, p_dialog, i, t, serie, overwrite):
|
|
logger.info("Updating " + path)
|
|
insertados_total = 0
|
|
|
|
head_nfo, it = videolibrarytools.read_nfo(path + '/tvshow.nfo')
|
|
# videolibrarytools.check_renumber_options(it)
|
|
videolibrarytools.update_renumber_options(it, head_nfo, path)
|
|
category = serie.category
|
|
|
|
# logger.debug("%s: %s" %(serie.contentSerieName,str(list_canales) ))
|
|
for channel, url in serie.library_urls.items():
|
|
serie.channel = channel
|
|
serie.url = url
|
|
|
|
###### Redirección al canal NewPct1.py si es un clone, o a otro canal y url si ha intervención judicial
|
|
try:
|
|
head_nfo, it = videolibrarytools.read_nfo(path + '/tvshow.nfo') #Refresca el .nfo para recoger actualizaciones
|
|
if it.emergency_urls:
|
|
serie.emergency_urls = it.emergency_urls
|
|
serie.category = category
|
|
except:
|
|
logger.error(traceback.format_exc())
|
|
|
|
channel_enabled = channeltools.is_enabled(serie.channel)
|
|
|
|
if channel_enabled:
|
|
|
|
heading = config.get_localized_string(20000)
|
|
p_dialog.update(int(math.ceil((i + 1) * t)), heading, config.get_localized_string(60389) % (serie.contentSerieName,
|
|
serie.channel.capitalize()))
|
|
try:
|
|
pathchannels = filetools.join(config.get_runtime_path(), "channels", serie.channel + '.py')
|
|
logger.info("loading channel: " + pathchannels + " " +
|
|
serie.channel)
|
|
|
|
if serie.library_filter_show:
|
|
serie.show = serie.library_filter_show.get(serie.channel, serie.contentSerieName)
|
|
|
|
obj = __import__('channels.%s' % serie.channel, fromlist=[pathchannels])
|
|
|
|
itemlist = obj.episodios(serie)
|
|
|
|
try:
|
|
if int(overwrite) == 3:
|
|
# Sobrescribir todos los archivos (tvshow.nfo, 1x01.nfo, 1x01 [canal].json, 1x01.strm, etc...)
|
|
insertados, sobreescritos, fallidos, notusedpath = videolibrarytools.save_tvshow(serie, itemlist)
|
|
#serie= videolibrary.check_season_playcount(serie, serie.contentSeason)
|
|
#if filetools.write(path + '/tvshow.nfo', head_nfo + it.tojson()):
|
|
# serie.infoLabels['playcount'] = serie.playcount
|
|
else:
|
|
insertados, sobreescritos, fallidos = videolibrarytools.save_episodes(path, itemlist, serie,
|
|
silent=True,
|
|
overwrite=overwrite)
|
|
#it = videolibrary.check_season_playcount(it, it.contentSeason)
|
|
#if filetools.write(path + '/tvshow.nfo', head_nfo + it.tojson()):
|
|
# serie.infoLabels['playcount'] = serie.playcount
|
|
insertados_total += insertados
|
|
|
|
except Exception as ex:
|
|
logger.error("Error when saving the chapters of the series")
|
|
template = "An exception of type %s occured. Arguments:\n%r"
|
|
message = template % (type(ex).__name__, ex.args)
|
|
logger.error(message)
|
|
|
|
except Exception as ex:
|
|
logger.error("Error in obtaining the episodes of: %s" % serie.show)
|
|
template = "An exception of type %s occured. Arguments:\n%r"
|
|
message = template % (type(ex).__name__, ex.args)
|
|
logger.error(message)
|
|
|
|
else:
|
|
logger.debug("Channel %s not active is not updated" % serie.channel)
|
|
|
|
#Sincronizamos los episodios vistos desde la videoteca de Kodi con la de Alfa
|
|
try:
|
|
if config.is_xbmc(): #Si es Kodi, lo hacemos
|
|
from platformcode import xbmc_videolibrary
|
|
xbmc_videolibrary.mark_content_as_watched_on_kod(filetools.join(path,'tvshow.nfo'))
|
|
except:
|
|
logger.error(traceback.format_exc())
|
|
|
|
return insertados_total > 0
|
|
|
|
|
|
def check_for_update(overwrite=True):
|
|
logger.info("Update Series...")
|
|
p_dialog = None
|
|
serie_actualizada = False
|
|
update_when_finished = False
|
|
hoy = datetime.date.today()
|
|
estado_verify_playcount_series = False
|
|
|
|
try:
|
|
if config.get_setting("update", "videolibrary") != 0 or overwrite:
|
|
config.set_setting("updatelibrary_last_check", hoy.strftime('%Y-%m-%d'), "videolibrary")
|
|
|
|
heading = config.get_localized_string(60389)
|
|
p_dialog = platformtools.dialog_progress_bg(config.get_localized_string(20000), heading)
|
|
p_dialog.update(0, '')
|
|
show_list = []
|
|
|
|
for path, folders, files in filetools.walk(videolibrarytools.TVSHOWS_PATH):
|
|
show_list.extend([filetools.join(path, f) for f in files if f == "tvshow.nfo"])
|
|
|
|
if show_list:
|
|
t = float(100) / len(show_list)
|
|
|
|
for i, tvshow_file in enumerate(show_list):
|
|
head_nfo, serie = videolibrarytools.read_nfo(tvshow_file)
|
|
path = filetools.dirname(tvshow_file)
|
|
|
|
logger.info("serie=" + serie.contentSerieName)
|
|
p_dialog.update(int(math.ceil((i + 1) * t)), heading, serie.contentSerieName)
|
|
|
|
#Verificamos el estado del serie.library_playcounts de la Serie por si está incompleto
|
|
try:
|
|
estado = False
|
|
#Si no hemos hecho la verificación o no tiene playcount, entramos
|
|
estado = config.get_setting("verify_playcount", "videolibrary")
|
|
if not estado or estado == False or not serie.library_playcounts: #Si no se ha pasado antes, lo hacemos ahora
|
|
serie, estado = videolibrary.verify_playcount_series(serie, path) #También se pasa si falta un PlayCount por completo
|
|
except:
|
|
logger.error(traceback.format_exc())
|
|
else:
|
|
if estado: #Si ha tenido éxito la actualización...
|
|
estado_verify_playcount_series = True #... se marca para cambiar la opción de la Videoteca
|
|
|
|
interval = int(serie.active) # Podria ser del tipo bool
|
|
|
|
if not serie.active:
|
|
# si la serie no esta activa descartar
|
|
if not overwrite:
|
|
#Sincronizamos los episodios vistos desde la videoteca de Kodi con la de Alfa, aunque la serie esté desactivada
|
|
try:
|
|
if config.is_xbmc(): #Si es Kodi, lo hacemos
|
|
from platformcode import xbmc_videolibrary
|
|
xbmc_videolibrary.mark_content_as_watched_on_kod(filetools.join(path,'tvshow.nfo'))
|
|
except:
|
|
logger.error(traceback.format_exc())
|
|
|
|
continue
|
|
|
|
# obtenemos las fecha de actualizacion y de la proxima programada para esta serie
|
|
update_next = serie.update_next
|
|
if update_next:
|
|
y, m, d = update_next.split('-')
|
|
update_next = datetime.date(int(y), int(m), int(d))
|
|
else:
|
|
update_next = hoy
|
|
|
|
update_last = serie.update_last
|
|
if update_last:
|
|
y, m, d = update_last.split('-')
|
|
update_last = datetime.date(int(y), int(m), int(d))
|
|
else:
|
|
update_last = hoy
|
|
|
|
# si la serie esta activa ...
|
|
if overwrite or config.get_setting("updatetvshows_interval", "videolibrary") == 0:
|
|
# ... forzar actualizacion independientemente del intervalo
|
|
serie_actualizada = update(path, p_dialog, i, t, serie, overwrite)
|
|
if not serie_actualizada:
|
|
update_next = hoy + datetime.timedelta(days=interval)
|
|
|
|
elif interval == 1 and update_next <= hoy:
|
|
# ...actualizacion diaria
|
|
serie_actualizada = update(path, p_dialog, i, t, serie, overwrite)
|
|
if not serie_actualizada and update_last <= hoy - datetime.timedelta(days=7):
|
|
# si hace una semana q no se actualiza, pasar el intervalo a semanal
|
|
interval = 7
|
|
update_next = hoy + datetime.timedelta(days=interval)
|
|
|
|
elif interval == 7 and update_next <= hoy:
|
|
# ...actualizacion semanal
|
|
serie_actualizada = update(path, p_dialog, i, t, serie, overwrite)
|
|
if not serie_actualizada:
|
|
if update_last <= hoy - datetime.timedelta(days=14):
|
|
# si hace 2 semanas q no se actualiza, pasar el intervalo a mensual
|
|
interval = 30
|
|
|
|
update_next += datetime.timedelta(days=interval)
|
|
|
|
elif interval == 30 and update_next <= hoy:
|
|
# ...actualizacion mensual
|
|
serie_actualizada = update(path, p_dialog, i, t, serie, overwrite)
|
|
if not serie_actualizada:
|
|
update_next += datetime.timedelta(days=interval)
|
|
|
|
if serie_actualizada:
|
|
update_last = hoy
|
|
update_next = hoy + datetime.timedelta(days=interval)
|
|
|
|
head_nfo, serie = videolibrarytools.read_nfo(tvshow_file) #Vuelve a leer el.nfo, que ha sido modificado
|
|
if interval != int(serie.active) or update_next.strftime('%Y-%m-%d') != serie.update_next or update_last.strftime('%Y-%m-%d') != serie.update_last:
|
|
serie.update_last = update_last.strftime('%Y-%m-%d')
|
|
if update_next > hoy:
|
|
serie.update_next = update_next.strftime('%Y-%m-%d')
|
|
serie.active = interval
|
|
serie.channel = "videolibrary"
|
|
serie.action = "get_seasons"
|
|
filetools.write(tvshow_file, head_nfo + serie.tojson())
|
|
|
|
if serie_actualizada:
|
|
if config.get_setting("search_new_content", "videolibrary") == 0:
|
|
# Actualizamos la videoteca de Kodi: Buscar contenido en la carpeta de la serie
|
|
if config.is_xbmc() and config.get_setting("videolibrary_kodi"):
|
|
from platformcode import xbmc_videolibrary
|
|
xbmc_videolibrary.update(folder=filetools.basename(path))
|
|
else:
|
|
update_when_finished = True
|
|
|
|
if estado_verify_playcount_series: #Si se ha cambiado algún playcount, ...
|
|
estado = config.set_setting("verify_playcount", True, "videolibrary") #... actualizamos la opción de Videolibrary
|
|
|
|
if config.get_setting("search_new_content", "videolibrary") == 1 and update_when_finished:
|
|
# Actualizamos la videoteca de Kodi: Buscar contenido en todas las series
|
|
if config.is_xbmc() and config.get_setting("videolibrary_kodi"):
|
|
from platformcode import xbmc_videolibrary
|
|
xbmc_videolibrary.update()
|
|
|
|
p_dialog.close()
|
|
|
|
else:
|
|
logger.info("Not update the video library, it is disabled")
|
|
|
|
except Exception as ex:
|
|
logger.error("An error occurred while updating the series")
|
|
template = "An exception of type %s occured. Arguments:\n%r"
|
|
message = template % (type(ex).__name__, ex.args)
|
|
logger.error(message)
|
|
|
|
if p_dialog:
|
|
p_dialog.close()
|
|
|
|
from core.item import Item
|
|
item_dummy = Item()
|
|
videolibrary.list_movies(item_dummy, silent=True)
|
|
|
|
if config.get_setting('trakt_sync'):
|
|
from core import trakt_tools
|
|
trakt_tools.update_all()
|
|
|
|
|
|
def viewmodeMonitor():
|
|
try:
|
|
currentModeName = xbmc.getInfoLabel('Container.Viewmode')
|
|
win = xbmcgui.Window(xbmcgui.getCurrentWindowId())
|
|
currentMode = int(win.getFocusId())
|
|
if currentModeName and 'plugin.video.kod' in xbmc.getInfoLabel(
|
|
'Container.FolderPath') and currentMode < 1000 and currentMode > 50: # inside addon and in itemlist view
|
|
content, Type = platformtools.getCurrentView()
|
|
if content:
|
|
defaultMode = int(config.get_setting('view_mode_%s' % content).split(',')[-1])
|
|
if currentMode != defaultMode:
|
|
logger.info('viewmode changed: ' + currentModeName + '-' + str(currentMode) + ' - content: ' + content)
|
|
config.set_setting('view_mode_%s' % content, currentModeName + ', ' + str(currentMode))
|
|
except:
|
|
logger.error(traceback.print_exc())
|
|
|
|
|
|
def updaterCheck():
|
|
# updater check
|
|
updated, needsReload = updater.check(background=True)
|
|
if needsReload:
|
|
xbmc.executescript(__file__)
|
|
exit(0)
|
|
|
|
|
|
def run_threaded(job_func, args):
|
|
job_thread = threading.Thread(target=job_func, args=args)
|
|
job_thread.start()
|
|
|
|
|
|
class AddonMonitor(xbmc.Monitor):
|
|
def __init__(self):
|
|
self.settings_pre = config.get_all_settings_addon()
|
|
|
|
self.updaterPeriod = None
|
|
self.update_setting = None
|
|
self.update_hour = None
|
|
self.scheduleScreenOnJobs()
|
|
self.scheduleUpdater()
|
|
|
|
# videolibrary wait
|
|
update_wait = [0, 10000, 20000, 30000, 60000]
|
|
wait = update_wait[int(config.get_setting("update_wait", "videolibrary"))]
|
|
if wait > 0:
|
|
xbmc.sleep(wait)
|
|
if not config.get_setting("update", "videolibrary") == 2:
|
|
run_threaded(check_for_update, (False,))
|
|
self.scheduleVideolibrary()
|
|
super(AddonMonitor, self).__init__()
|
|
|
|
def onSettingsChanged(self):
|
|
logger.info('settings changed')
|
|
settings_post = config.get_all_settings_addon()
|
|
from platformcode import xbmc_videolibrary
|
|
|
|
if self.settings_pre.get('downloadpath', None) != settings_post.get('downloadpath', None):
|
|
xbmc_videolibrary.update_sources(settings_post.get('downloadpath', None),
|
|
self.settings_pre.get('downloadpath', None))
|
|
|
|
# si se ha cambiado la ruta de la videoteca llamamos a comprobar directorios para que lo cree y pregunte
|
|
# automaticamente si configurar la videoteca
|
|
if self.settings_pre.get("videolibrarypath", None) != settings_post.get("videolibrarypath", None) or \
|
|
self.settings_pre.get("folder_movies", None) != settings_post.get("folder_movies", None) or \
|
|
self.settings_pre.get("folder_tvshows", None) != settings_post.get("folder_tvshows", None):
|
|
videolibrary.move_videolibrary(self.settings_pre.get("videolibrarypath", None),
|
|
settings_post.get("videolibrarypath", None),
|
|
self.settings_pre.get("folder_movies", None),
|
|
settings_post.get("folder_movies", None),
|
|
self.settings_pre.get("folder_tvshows", None),
|
|
settings_post.get("folder_tvshows", None))
|
|
|
|
# si se ha puesto que se quiere autoconfigurar y se había creado el directorio de la videoteca
|
|
if not self.settings_pre.get("videolibrary_kodi", None) and settings_post.get("videolibrary_kodi", None):
|
|
xbmc_videolibrary.ask_set_content(silent=True)
|
|
elif self.settings_pre.get("videolibrary_kodi", None) and not settings_post.get("videolibrary_kodi", None):
|
|
xbmc_videolibrary.clean()
|
|
|
|
if self.settings_pre.get('addon_update_timer') != settings_post.get('addon_update_timer'):
|
|
schedule.clear('updater')
|
|
self.scheduleUpdater()
|
|
|
|
if self.update_setting != config.get_setting("update", "videolibrary") or self.update_hour != config.get_setting("everyday_delay", "videolibrary") * 4:
|
|
schedule.clear('videolibrary')
|
|
self.scheduleVideolibrary()
|
|
|
|
if self.settings_pre.get('elementum_on_seed') != settings_post.get('elementum_on_seed') and settings_post.get('elementum_on_seed'):
|
|
if not platformtools.dialog_yesno(config.get_localized_string(70805), config.get_localized_string(70806)):
|
|
config.set_setting('elementum_on_seed', False)
|
|
|
|
self.settings_pre = settings_post
|
|
|
|
def onScreensaverActivated(self):
|
|
logger.info('screensaver activated, un-scheduling screen-on jobs')
|
|
schedule.clear('screenOn')
|
|
|
|
def onScreensaverDeactivated(self):
|
|
logger.info('screensaver deactivated, re-scheduling screen-on jobs')
|
|
self.scheduleScreenOnJobs()
|
|
|
|
def scheduleUpdater(self):
|
|
if not config.dev_mode():
|
|
updaterCheck()
|
|
self.updaterPeriod = config.get_setting('addon_update_timer')
|
|
schedule.every(self.updaterPeriod).hours.do(updaterCheck).tag('updater')
|
|
logger.info('scheduled updater every ' + str(self.updaterPeriod) + ' hours')
|
|
|
|
def scheduleVideolibrary(self):
|
|
self.update_setting = config.get_setting("update", "videolibrary")
|
|
# 2= daily 3=daily and when kodi starts
|
|
if self.update_setting == 2 or self.update_setting == 3:
|
|
self.update_hour = config.get_setting("everyday_delay", "videolibrary") * 4
|
|
schedule.every().day.at(str(self.update_hour).zfill(2) + ':00').do(run_threaded, check_for_update, (False,)).tag('videolibrary')
|
|
logger.info('scheduled videolibrary at ' + str(self.update_hour).zfill(2) + ':00')
|
|
|
|
def scheduleScreenOnJobs(self):
|
|
schedule.every().second.do(viewmodeMonitor).tag('screenOn')
|
|
schedule.every().second.do(torrent.elementum_monitor).tag('screenOn')
|
|
|
|
def onDPMSActivated(self):
|
|
logger.info('DPMS activated, un-scheduling screen-on jobs')
|
|
schedule.clear('screenOn')
|
|
|
|
def onDPMSDeactivated(self):
|
|
logger.info('DPMS deactivated, re-scheduling screen-on jobs')
|
|
self.scheduleScreenOnJobs()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logger.info('Starting KoD service')
|
|
monitor = AddonMonitor()
|
|
|
|
# mark as stopped all downloads (if we are here, probably kodi just started)
|
|
from specials.downloads import stop_all
|
|
try:
|
|
stop_all()
|
|
except:
|
|
logger.error(traceback.format_exc())
|
|
|
|
while True:
|
|
schedule.run_pending()
|
|
|
|
if monitor.waitForAbort(1): # every second
|
|
break
|