289 lines
11 KiB
Python
289 lines
11 KiB
Python
# -*- coding: utf-8 -*-
|
|
# -*- Tools for trakt sync -*-
|
|
# -*- Created for Alfa-addon -*-
|
|
# -*- By the Alfa Develop Group -*
|
|
|
|
#from builtins import str
|
|
|
|
import os, xbmc
|
|
from core import httptools, jsontools
|
|
from core.item import Item
|
|
from platformcode import config, logger
|
|
from threading import Thread
|
|
|
|
import sys
|
|
if sys.version_info[0] >= 3: from concurrent import futures
|
|
else: from concurrent_py2 import futures
|
|
|
|
client_id = "502bd1660b833c1ae69828163c0848e84e9850061e5529f30930e7356cae73b1"
|
|
client_secret = "1d30d5b24acf223a5e1ab6c61d08b69992d98ed5b0c7e26b052b5e6a592035a4"
|
|
|
|
|
|
def auth_trakt():
|
|
item = Item()
|
|
folder = (config.get_platform() == "plex")
|
|
item.folder = folder
|
|
# Autentificación de cuenta Trakt
|
|
headers = {'Content-Type': 'application/json', 'trakt-api-key': client_id, 'trakt-api-version': '2'}
|
|
try:
|
|
post = {'client_id': client_id}
|
|
post = jsontools.dump(post)
|
|
# Se solicita url y código de verificación para conceder permiso a la app
|
|
url = "http://api-v2launch.trakt.tv/oauth/device/code"
|
|
data = httptools.downloadpage(url, post=post, headers=headers).data
|
|
data = jsontools.load(data)
|
|
item.verify_url = data["verification_url"]
|
|
item.user_code = data["user_code"]
|
|
item.device_code = data["device_code"]
|
|
item.intervalo = data["interval"]
|
|
if not item.folder:
|
|
token_trakt(item)
|
|
|
|
else:
|
|
itemlist = []
|
|
title = config.get_localized_string(60248) % item.verify_url
|
|
itemlist.append(item.clone(title=title, action=""))
|
|
title = config.get_localized_string(60249) % item.user_code
|
|
itemlist.append(item.clone(title=title, action=""))
|
|
title = config.get_localized_string(60250)
|
|
itemlist.append(item.clone(title=title, action="token_trakt"))
|
|
return itemlist
|
|
except:
|
|
import traceback
|
|
logger.error(traceback.format_exc())
|
|
|
|
|
|
def token_trakt(item):
|
|
from platformcode import platformtools
|
|
|
|
headers = {'Content-Type': 'application/json', 'trakt-api-key': client_id, 'trakt-api-version': '2'}
|
|
try:
|
|
if item.extra == "renew":
|
|
refresh = config.get_setting("refresh_token_trakt", "trakt")
|
|
url = "https://api.trakt.tv/oauth/token"
|
|
post = {'refresh_token': refresh, 'client_id': client_id, 'client_secret': client_secret,
|
|
'redirect_uri': 'urn:ietf:wg:oauth:2.0:oob', 'grant_type': 'refresh_token'}
|
|
data = httptools.downloadpage(url, post=post).data
|
|
data = jsontools.load(data)
|
|
elif item.action == "token_trakt":
|
|
url = "https://api-v2launch.trakt.tv/oauth/device/token"
|
|
post = "code=%s&client_id=%s&client_secret=%s" % (item.device_code, client_id, client_secret)
|
|
data = httptools.downloadpage(url, post=post, headers=headers).data
|
|
data = jsontools.load(data)
|
|
else:
|
|
import time
|
|
dialog_auth = platformtools.dialog_progress(config.get_localized_string(60251),
|
|
config.get_localized_string(60252) % item.verify_url + '\n' +
|
|
config.get_localized_string(60253) % item.user_code + '\n' +
|
|
config.get_localized_string(60254))
|
|
|
|
# Generalmente cada 5 segundos se intenta comprobar si el usuario ha introducido el código
|
|
while True:
|
|
time.sleep(item.intervalo)
|
|
try:
|
|
if dialog_auth.iscanceled():
|
|
config.set_setting("trakt_sync", False)
|
|
return
|
|
|
|
url = "http://api-v2launch.trakt.tv/oauth/device/token"
|
|
post = {'code': item.device_code, 'client_id': client_id, 'client_secret': client_secret}
|
|
post = jsontools.dump(post)
|
|
data = httptools.downloadpage(url, post=post, headers=headers).data
|
|
data = jsontools.load(data)
|
|
if "access_token" in data:
|
|
# Código introducido, salimos del bucle
|
|
break
|
|
except:
|
|
pass
|
|
|
|
try:
|
|
dialog_auth.close()
|
|
except:
|
|
pass
|
|
|
|
token = data["access_token"]
|
|
refresh = data["refresh_token"]
|
|
|
|
config.set_setting("token_trakt", token, "trakt")
|
|
config.set_setting("refresh_token_trakt", refresh, "trakt")
|
|
if not item.folder:
|
|
platformtools.dialog_notification(config.get_localized_string(60255), config.get_localized_string(60256))
|
|
xbmc.executebuiltin("Container.Refresh")
|
|
return
|
|
|
|
except:
|
|
import traceback
|
|
logger.error(traceback.format_exc())
|
|
if not item.folder:
|
|
return platformtools.dialog_notification(config.get_localized_string(60527), config.get_localized_string(60258))
|
|
token = ""
|
|
|
|
itemlist = []
|
|
if token:
|
|
itemlist.append(item.clone(config.get_localized_string(60256), action=""))
|
|
else:
|
|
itemlist.append(item.clone(config.get_localized_string(60260), action=""))
|
|
|
|
return itemlist
|
|
|
|
|
|
def set_trakt_info(item):
|
|
logger.debug()
|
|
import xbmcgui
|
|
from core import scrapertools
|
|
# Envia los datos a trakt
|
|
try:
|
|
info = item.infoLabels
|
|
ids = jsontools.dump({'tmdb': info['tmdb_id'], 'tvdb': info.get('tvdb_id'), 'imdb': info['imdb_id'],
|
|
'slug': scrapertools.slugify(info['title'])})
|
|
xbmcgui.Window(10000).setProperty('script.trakt.ids', ids)
|
|
except:
|
|
pass
|
|
|
|
def get_trakt_watched(id_type, mediatype, update=False):
|
|
logger.debug()
|
|
|
|
id_list = []
|
|
id_dict = dict()
|
|
|
|
token_auth = config.get_setting("token_trakt", "trakt")
|
|
|
|
if token_auth:
|
|
sync_path = os.path.join(config.get_data_path(), 'settings_channels', 'trakt')
|
|
|
|
if os.path.exists(sync_path) and not update:
|
|
trakt_node = jsontools.get_node_from_file('trakt', "TRAKT")
|
|
if mediatype == 'shows':
|
|
return trakt_node['shows']
|
|
if mediatype == 'movies':
|
|
return trakt_node['movies']
|
|
|
|
else:
|
|
token_auth = config.get_setting("token_trakt", "trakt")
|
|
if token_auth:
|
|
try:
|
|
token_auth = config.get_setting("token_trakt", "trakt")
|
|
headers = [['Content-Type', 'application/json'], ['trakt-api-key', client_id],
|
|
['trakt-api-version', '2']]
|
|
if token_auth:
|
|
headers.append(['Authorization', "Bearer %s" % token_auth])
|
|
url = "https://api.trakt.tv/sync/watched/%s" % mediatype
|
|
data = httptools.downloadpage(url, headers=headers)
|
|
if data.code == 401:
|
|
token_trakt(Item(extra="renew"))
|
|
return get_trakt_watched(id_type, mediatype, update)
|
|
|
|
watched_dict = jsontools.load(data.data)
|
|
|
|
if mediatype == 'shows':
|
|
|
|
dict_show = dict()
|
|
for item in watched_dict:
|
|
temp = []
|
|
id_ = str(item['show']['ids']['tmdb'])
|
|
season_dict = dict()
|
|
for season in item['seasons']:
|
|
ep = []
|
|
number = str(season['number'])
|
|
# season_dict = dict()
|
|
for episode in season['episodes']:
|
|
ep.append(str(episode['number']))
|
|
season_dict[number] = ep
|
|
temp.append(season_dict)
|
|
dict_show[id_] = season_dict
|
|
id_dict = dict_show
|
|
return id_dict
|
|
|
|
elif mediatype == 'movies':
|
|
for item in watched_dict:
|
|
id_list.append(str(item['movie']['ids'][id_type]))
|
|
except:
|
|
pass
|
|
|
|
return id_list
|
|
|
|
|
|
def trakt_check(itemlist):
|
|
if type(itemlist) != list:
|
|
return
|
|
def sync(item, id_result):
|
|
info = item.infoLabels
|
|
try:
|
|
if info['mediatype'] == 'movie' and info['tmdb_id'] in id_result[info['mediatype']]:
|
|
item.infoLabels['playcount'] = 1
|
|
elif info['mediatype'] == 'episode' and info['tmdb_id'] in id_result[info['mediatype']]:
|
|
id = info['tmdb_id']
|
|
if info['season'] and info['episode'] and \
|
|
str(info['season']) in id_result[info['mediatype']][id] and \
|
|
str(info['episode']) in id_result[info['mediatype']][id][str(info['season'])]:
|
|
item.infoLabels['playcount'] = 1
|
|
except:
|
|
pass
|
|
|
|
if itemlist and itemlist[0].channel != 'videolibrary' \
|
|
and 'mediatype' in itemlist[0].infoLabels \
|
|
and itemlist[0].infoLabels['mediatype'] in ['movie', 'episode']:
|
|
|
|
id_result = {}
|
|
id_result['movie'] = get_trakt_watched('tmdb', 'movies', True)
|
|
id_result['episode'] = get_trakt_watched('tmdb', 'shows', True)
|
|
|
|
with futures.ThreadPoolExecutor() as executor:
|
|
[executor.submit(sync, it, id_result) for it in itemlist]
|
|
|
|
return itemlist
|
|
|
|
|
|
def get_sync_from_file():
|
|
logger.debug()
|
|
sync_path = os.path.join(config.get_data_path(), 'settings_channels', 'trakt_data.json')
|
|
trakt_node = {}
|
|
if os.path.exists(sync_path):
|
|
trakt_node = jsontools.get_node_from_file('trakt', "TRAKT")
|
|
|
|
trakt_node['movies'] = get_trakt_watched('tmdb', 'movies')
|
|
trakt_node['shows'] = get_trakt_watched('tmdb', 'shows')
|
|
jsontools.update_node(trakt_node, 'trakt', 'TRAKT')
|
|
|
|
|
|
def update_trakt_data(mediatype, trakt_data):
|
|
logger.debug()
|
|
|
|
sync_path = os.path.join(config.get_data_path(), 'settings_channels', 'trakt_data.json')
|
|
if os.path.exists(sync_path):
|
|
trakt_node = jsontools.get_node_from_file('trakt', "TRAKT")
|
|
trakt_node[mediatype] = trakt_data
|
|
jsontools.update_node(trakt_node, 'trakt', 'TRAKT')
|
|
|
|
|
|
def ask_install_script():
|
|
logger.debug()
|
|
|
|
from platformcode import platformtools
|
|
|
|
respuesta = platformtools.dialog_yesno(config.get_localized_string(20000), config.get_localized_string(70521))
|
|
if respuesta:
|
|
xbmc.executebuiltin("InstallAddon(script.trakt)")
|
|
return
|
|
else:
|
|
config.set_setting('install_trakt', False)
|
|
return
|
|
|
|
|
|
def wait_for_update_trakt():
|
|
logger.debug()
|
|
t = Thread(update_all)
|
|
t.setDaemon(True)
|
|
t.start()
|
|
t.is_alive()
|
|
|
|
def update_all():
|
|
from time import sleep
|
|
logger.debug()
|
|
sleep(20)
|
|
while xbmc.Player().isPlaying():
|
|
sleep(20)
|
|
for mediatype in ['movies', 'shows']:
|
|
trakt_data = get_trakt_watched('tmdb', mediatype, True)
|
|
update_trakt_data(mediatype, trakt_data)
|