stevazi/addon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

Trakt, segna come letto in Thread

Browse Source
This commit is contained in:
Alhaziel01
2020-05-22 12:22:28 +02:00
parent 15db87fa3e
commit 3d43ada7f3
1 changed files with 26 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
core/trakt_tools.py
View File

@@ -8,6 +8,9 @@ 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"
@@ -196,48 +199,30 @@ def get_trakt_watched(id_type, mediatype, update=False):
def trakt_check(itemlist):
id_result = ''
synced = False
try:
for item in itemlist:
info = item.infoLabels
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 info != '' and info['mediatype'] in ['movie', 'episode'] and item.channel != 'videolibrary':
if itemlist and itemlist[0].channel != 'videolibrary' \
and 'mediatype' in itemlist[0].infoLabels \
and itemlist[0].infoLabels['mediatype'] in ['movie', 'episode']:
if not synced:
get_sync_from_file()
synced = True
id_result = {}
id_result['movie'] = get_trakt_watched('tmdb', 'movies')
id_result['episode'] = get_trakt_watched('tmdb', 'shows')
mediatype = 'movies'
id_type = 'tmdb'
if info['mediatype'] == 'episode':
mediatype = 'shows'
if id_result == '':
id_result = get_trakt_watched(id_type, mediatype)
if info['mediatype'] == 'movie':
if info[id_type + '_id'] in id_result:
item.infoLabels['playcount'] = 1
elif info['mediatype'] == 'episode':
if info[id_type + '_id'] in id_result:
id = info[id_type + '_id']
if info['season'] != '' and info['episode'] != '':
season = str(info['season'])
if season in id_result[id]:
episode = str(info['episode'])
if episode in id_result[id][season]:
season_watched = id_result[id][season]
if episode in season_watched:
item.infoLabels['playcount'] = 1
else:
break
except:
pass
with futures.ThreadPoolExecutor() as executor:
[executor.submit(sync, it, id_result) for it in itemlist]
return itemlist
@@ -249,8 +234,8 @@ def get_sync_from_file():
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')
trakt_node['movies'] = get_trakt_watched('tmdb', 'movies', True)
trakt_node['shows'] = get_trakt_watched('tmdb', 'shows', True)
jsontools.update_node(trakt_node, 'trakt', 'TRAKT')