panni
139 lines
4.8 KiB
Python
139 lines
4.8 KiB
Python
# coding=utf-8
|
|
|
|
import os
|
|
import subliminal
|
|
import helpers
|
|
|
|
from items import get_item
|
|
from subzero import intent
|
|
|
|
|
|
def flatten_media(media, kind="series"):
|
|
"""
|
|
iterates through media and returns the associated parts (videos)
|
|
:param media:
|
|
:param kind:
|
|
:return:
|
|
"""
|
|
parts = []
|
|
|
|
def get_metadata_dict(item, part, add):
|
|
data = {
|
|
"section": item.section.title,
|
|
"path": part.file,
|
|
"folder": os.path.dirname(part.file),
|
|
"filename": os.path.basename(part.file)
|
|
}
|
|
data.update(add)
|
|
return data
|
|
|
|
if kind == "series":
|
|
for season in media.seasons:
|
|
season_object = media.seasons[season]
|
|
for episode in media.seasons[season].episodes:
|
|
ep = media.seasons[season].episodes[episode]
|
|
|
|
# get plex item via API for additional metadata
|
|
plex_episode = get_item(ep.id)
|
|
|
|
for item in media.seasons[season].episodes[episode].items:
|
|
for part in item.parts:
|
|
parts.append(
|
|
get_metadata_dict(plex_episode, part,
|
|
{"video": part, "type": "episode", "title": ep.title,
|
|
"series": media.title, "id": ep.id,
|
|
"series_id": media.id, "season_id": season_object.id,
|
|
"season": plex_episode.season.index,
|
|
})
|
|
)
|
|
else:
|
|
plex_item = get_item(media.id)
|
|
for item in media.items:
|
|
for part in item.parts:
|
|
parts.append(
|
|
get_metadata_dict(plex_item, part, {"video": part, "type": "movie",
|
|
"title": media.title, "id": media.id,
|
|
"series_id": None,
|
|
"season_id": None,
|
|
"section": plex_item.section.title})
|
|
)
|
|
return parts
|
|
|
|
|
|
IGNORE_FN = ("subzero.ignore", ".subzero.ignore", ".nosz")
|
|
|
|
|
|
def convert_media_to_parts(media, kind="series"):
|
|
"""
|
|
returns a list of parts to be used later on; ignores folders with an existing "subzero.ignore" file
|
|
:param media:
|
|
:param kind:
|
|
:return:
|
|
"""
|
|
return flatten_media(media, kind=kind)
|
|
|
|
|
|
def get_stream_fps(streams):
|
|
"""
|
|
accepts a list of plex streams or a list of the plex api streams
|
|
"""
|
|
for stream in streams:
|
|
# video
|
|
stream_type = getattr(stream, "type", getattr(stream, "stream_type", None))
|
|
if stream_type == 1:
|
|
return getattr(stream, "frameRate", getattr(stream, "frame_rate", "25.000"))
|
|
return "25.000"
|
|
|
|
|
|
def get_media_item_ids(media, kind="series"):
|
|
ids = []
|
|
if kind == "movies":
|
|
ids.append(media.id)
|
|
else:
|
|
for season in media.seasons:
|
|
for episode in media.seasons[season].episodes:
|
|
ids.append(media.seasons[season].episodes[episode].id)
|
|
|
|
return ids
|
|
|
|
|
|
def scan_video(plex_video, ignore_all=False, hints=None):
|
|
embedded_subtitles = not ignore_all and Prefs['subtitles.scan.embedded']
|
|
external_subtitles = not ignore_all and Prefs['subtitles.scan.external']
|
|
|
|
if ignore_all:
|
|
Log.Debug("Force refresh intended.")
|
|
|
|
Log.Debug("Scanning video: %s, subtitles=%s, embedded_subtitles=%s" % (plex_video.file, external_subtitles, embedded_subtitles))
|
|
|
|
try:
|
|
return subliminal.video.scan_video(plex_video.file, subtitles=external_subtitles, embedded_subtitles=embedded_subtitles,
|
|
hints=hints or {}, video_fps=plex_video.fps)
|
|
|
|
except ValueError:
|
|
Log.Warn("File could not be guessed by subliminal")
|
|
|
|
|
|
def scan_parts(parts, kind="series"):
|
|
"""
|
|
receives a list of parts containing dictionaries returned by flattenToParts
|
|
:param parts:
|
|
:param kind: series or movies
|
|
:return: dictionary of subliminal.video.scan_video, key=subliminal scanned video, value=plex file part
|
|
"""
|
|
ret = {}
|
|
for part in parts:
|
|
force_refresh = intent.get("force", part["id"], part["series_id"], part["season_id"])
|
|
|
|
hints = helpers.get_item_hints(part["title"], kind, series=part["series"] if kind == "series" else None)
|
|
part["video"].fps = get_stream_fps(part["video"].streams)
|
|
scanned_video = scan_video(part["video"], ignore_all=force_refresh, hints=hints)
|
|
if not scanned_video:
|
|
continue
|
|
|
|
scanned_video.id = part["id"]
|
|
part_metadata = part.copy()
|
|
del part_metadata["video"]
|
|
scanned_video.plexapi_metadata = part_metadata
|
|
ret[scanned_video] = part["video"]
|
|
return ret |