panni
93 lines
2.7 KiB
Python
93 lines
2.7 KiB
Python
# coding=utf-8
|
|
|
|
import datetime
|
|
import pprint
|
|
|
|
|
|
def get_subtitle_info(rating_key):
|
|
return Dict["subs"].get(rating_key)
|
|
|
|
|
|
def whack_missing_parts(videos, existing_parts=None):
|
|
"""
|
|
cleans out our internal storage's video parts (parts may get updated/deleted/whatever)
|
|
:param existing_parts: optional list of part ids known
|
|
:param videos: videos to check for
|
|
:return:
|
|
"""
|
|
# shortcut
|
|
|
|
if not existing_parts:
|
|
existing_parts = []
|
|
for part in videos.viewvalues():
|
|
existing_parts.append(part.id)
|
|
|
|
whacked_parts = False
|
|
for video in videos.keys():
|
|
if video.id not in Dict["subs"]:
|
|
continue
|
|
|
|
for part_id in Dict["subs"][video.id].keys():
|
|
if part_id not in existing_parts:
|
|
del Dict["subs"][video.id][part_id]
|
|
Log.Info("Whacking part %s in internal storage of video %s", part_id, video.id)
|
|
whacked_parts = True
|
|
|
|
if whacked_parts:
|
|
Dict.Save()
|
|
|
|
|
|
def store_subtitle_info(videos, subtitles, storage_type):
|
|
"""
|
|
stores information about downloaded subtitles in plex's Dict()
|
|
"""
|
|
if "subs" not in Dict:
|
|
Dict["subs"] = {}
|
|
|
|
storage = Dict["subs"]
|
|
|
|
existing_parts = []
|
|
for video, video_subtitles in subtitles.items():
|
|
part = videos[video]
|
|
|
|
if video.id not in storage:
|
|
storage[video.id] = {}
|
|
|
|
video_dict = storage[video.id]
|
|
if part.id not in video_dict:
|
|
video_dict[part.id] = {}
|
|
|
|
existing_parts.append(part.id)
|
|
|
|
part_dict = video_dict[part.id]
|
|
for subtitle in video_subtitles:
|
|
lang = Locale.Language.Match(subtitle.language.alpha2)
|
|
if lang not in part_dict:
|
|
part_dict[lang] = {}
|
|
lang_dict = part_dict[lang]
|
|
sub_key = (subtitle.provider_name, subtitle.id)
|
|
lang_dict[sub_key] = dict(score=subtitle.score, link=subtitle.page_link, storage=storage_type, hash=Hash.MD5(subtitle.content),
|
|
date_added=datetime.datetime.now())
|
|
lang_dict["current"] = sub_key
|
|
|
|
if existing_parts:
|
|
whack_missing_parts(videos, existing_parts=existing_parts)
|
|
Dict.Save()
|
|
|
|
|
|
def reset_storage(key):
|
|
"""
|
|
resets the Dict[key] storage, thanks to https://docs.google.com/document/d/1hhLjV1pI-TA5y91TiJq64BdgKwdLnFt4hWgeOqpz1NA/edit#
|
|
We can't use the nice Plex interface for this, as it calls get multiple times before set
|
|
#Plex[":/plugins/*/prefs"].set("com.plexapp.agents.subzero", "reset_storage", False)
|
|
"""
|
|
|
|
Log.Debug("resetting storage")
|
|
Dict[key] = {}
|
|
Dict.Save()
|
|
|
|
|
|
def log_storage(key):
|
|
if key in Dict:
|
|
Log.Debug(pprint.pformat(Dict[key]))
|