544 lines
28 KiB
Python
544 lines
28 KiB
Python
import os.path
|
||
import time
|
||
from xml.dom import minidom
|
||
|
||
import log
|
||
from app.helper import FfmpegHelper
|
||
from app.media.douban import DouBan
|
||
from config import TMDB_IMAGE_W500_URL
|
||
from app.utils import DomUtils, RequestUtils, ExceptionUtils
|
||
from app.utils.types import MediaType
|
||
from app.media import Media
|
||
|
||
|
||
class Scraper:
|
||
media = None
|
||
|
||
def __init__(self):
|
||
self.media = Media()
|
||
self.douban = DouBan()
|
||
|
||
def __gen_common_nfo(self,
|
||
tmdbinfo: dict,
|
||
doubaninfo: dict,
|
||
scraper_nfo: dict,
|
||
doc,
|
||
root,
|
||
chinese=False):
|
||
if scraper_nfo.get("basic"):
|
||
# 添加时间
|
||
DomUtils.add_node(doc, root, "dateadded", time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())))
|
||
# TMDB
|
||
DomUtils.add_node(doc, root, "tmdbid", tmdbinfo.get("id") or "")
|
||
uniqueid_tmdb = DomUtils.add_node(doc, root, "uniqueid", tmdbinfo.get("id") or "")
|
||
uniqueid_tmdb.setAttribute("type", "tmdb")
|
||
uniqueid_tmdb.setAttribute("default", "true")
|
||
# TVDB IMDB
|
||
if tmdbinfo.get("external_ids"):
|
||
tvdbid = tmdbinfo.get("external_ids", {}).get("tvdb_id", 0)
|
||
if tvdbid:
|
||
DomUtils.add_node(doc, root, "tvdbid", tvdbid)
|
||
uniqueid_tvdb = DomUtils.add_node(doc, root, "uniqueid", tvdbid)
|
||
uniqueid_tvdb.setAttribute("type", "tvdb")
|
||
imdbid = tmdbinfo.get("external_ids", {}).get("imdb_id", "")
|
||
if imdbid:
|
||
DomUtils.add_node(doc, root, "imdbid", imdbid)
|
||
uniqueid_imdb = DomUtils.add_node(doc, root, "uniqueid", imdbid)
|
||
uniqueid_imdb.setAttribute("type", "imdb")
|
||
uniqueid_imdb.setAttribute("default", "true")
|
||
uniqueid_tmdb.setAttribute("default", "false")
|
||
|
||
# 简介
|
||
xplot = DomUtils.add_node(doc, root, "plot")
|
||
xplot.appendChild(doc.createCDATASection(tmdbinfo.get("overview") or ""))
|
||
xoutline = DomUtils.add_node(doc, root, "outline")
|
||
xoutline.appendChild(doc.createCDATASection(tmdbinfo.get("overview") or ""))
|
||
if scraper_nfo.get("credits"):
|
||
# 导演
|
||
directors, actors = self.media.get_tmdb_directors_actors(tmdbinfo=tmdbinfo)
|
||
if chinese:
|
||
directors, actors = self.__gen_people_chinese_info(directors, actors, doubaninfo)
|
||
for director in directors:
|
||
xdirector = DomUtils.add_node(doc, root, "director", director.get("name") or "")
|
||
xdirector.setAttribute("tmdbid", str(director.get("id") or ""))
|
||
# 演员
|
||
for actor in actors:
|
||
xactor = DomUtils.add_node(doc, root, "actor")
|
||
DomUtils.add_node(doc, xactor, "name", actor.get("name") or "")
|
||
DomUtils.add_node(doc, xactor, "type", "Actor")
|
||
DomUtils.add_node(doc, xactor, "role", actor.get("role") or "")
|
||
DomUtils.add_node(doc, xactor, "order", actor.get("order") if actor.get("order") is not None else "")
|
||
DomUtils.add_node(doc, xactor, "tmdbid", actor.get("id") or "")
|
||
DomUtils.add_node(doc, xactor, "thumb", actor.get('image'))
|
||
DomUtils.add_node(doc, xactor, "profile", actor.get('profile'))
|
||
if scraper_nfo.get("basic"):
|
||
# 风格
|
||
genres = tmdbinfo.get("genres") or []
|
||
for genre in genres:
|
||
DomUtils.add_node(doc, root, "genre", genre.get("name") or "")
|
||
# 评分
|
||
DomUtils.add_node(doc, root, "rating", tmdbinfo.get("vote_average") or "0")
|
||
return doc
|
||
|
||
def gen_movie_nfo_file(self,
|
||
tmdbinfo: dict,
|
||
doubaninfo: dict,
|
||
scraper_movie_nfo: dict,
|
||
out_path,
|
||
file_name):
|
||
"""
|
||
生成电影的NFO描述文件
|
||
:param tmdbinfo: TMDB元数据
|
||
:param doubaninfo: 豆瓣元数据
|
||
:param scraper_movie_nfo: 刮削配置
|
||
:param out_path: 电影根目录
|
||
:param file_name: 电影文件名，不含后缀
|
||
"""
|
||
# 开始生成XML
|
||
log.info("【Scraper】正在生成电影NFO文件：%s" % file_name)
|
||
doc = minidom.Document()
|
||
root = DomUtils.add_node(doc, doc, "movie")
|
||
# 公共部分
|
||
doc = self.__gen_common_nfo(tmdbinfo=tmdbinfo,
|
||
doubaninfo=doubaninfo,
|
||
scraper_nfo=scraper_movie_nfo,
|
||
doc=doc,
|
||
root=root,
|
||
chinese=scraper_movie_nfo.get("credits_chinese"))
|
||
# 基础部分
|
||
if scraper_movie_nfo.get("basic"):
|
||
# 标题
|
||
DomUtils.add_node(doc, root, "title", tmdbinfo.get("title") or "")
|
||
DomUtils.add_node(doc, root, "originaltitle", tmdbinfo.get("original_title") or "")
|
||
# 发布日期
|
||
DomUtils.add_node(doc, root, "premiered", tmdbinfo.get("release_date") or "")
|
||
# 年份
|
||
DomUtils.add_node(doc, root, "year",
|
||
tmdbinfo.get("release_date")[:4] if tmdbinfo.get("release_date") else "")
|
||
# 保存
|
||
self.__save_nfo(doc, os.path.join(out_path, "%s.nfo" % file_name))
|
||
|
||
def gen_tv_nfo_file(self,
|
||
tmdbinfo: dict,
|
||
doubaninfo: dict,
|
||
scraper_tv_nfo: dict,
|
||
out_path):
|
||
"""
|
||
生成电视剧的NFO描述文件
|
||
:param tmdbinfo: TMDB元数据
|
||
:param doubaninfo: 豆瓣元数据
|
||
:param scraper_tv_nfo: 刮削配置
|
||
:param out_path: 电视剧根目录
|
||
"""
|
||
# 开始生成XML
|
||
log.info("【Scraper】正在生成电视剧NFO文件：%s" % out_path)
|
||
doc = minidom.Document()
|
||
root = DomUtils.add_node(doc, doc, "tvshow")
|
||
# 公共部分
|
||
doc = self.__gen_common_nfo(tmdbinfo=tmdbinfo,
|
||
doubaninfo=doubaninfo,
|
||
scraper_nfo=scraper_tv_nfo,
|
||
doc=doc,
|
||
root=root,
|
||
chinese=scraper_tv_nfo.get("credits_chinese"))
|
||
if scraper_tv_nfo.get("basic"):
|
||
# 标题
|
||
DomUtils.add_node(doc, root, "title", tmdbinfo.get("name") or "")
|
||
DomUtils.add_node(doc, root, "originaltitle", tmdbinfo.get("original_name") or "")
|
||
# 发布日期
|
||
DomUtils.add_node(doc, root, "premiered", tmdbinfo.get("first_air_date") or "")
|
||
# 年份
|
||
DomUtils.add_node(doc, root, "year",
|
||
tmdbinfo.get("first_air_date")[:4] if tmdbinfo.get("first_air_date") else "")
|
||
DomUtils.add_node(doc, root, "season", "-1")
|
||
DomUtils.add_node(doc, root, "episode", "-1")
|
||
# 保存
|
||
self.__save_nfo(doc, os.path.join(out_path, "tvshow.nfo"))
|
||
|
||
def gen_tv_season_nfo_file(self, tmdbinfo: dict, season, out_path):
|
||
"""
|
||
生成电视剧季的NFO描述文件
|
||
:param tmdbinfo: TMDB季媒体信息
|
||
:param season: 季号
|
||
:param out_path: 电视剧季的目录
|
||
"""
|
||
log.info("【Scraper】正在生成季NFO文件：%s" % out_path)
|
||
doc = minidom.Document()
|
||
root = DomUtils.add_node(doc, doc, "season")
|
||
# 添加时间
|
||
DomUtils.add_node(doc, root, "dateadded", time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())))
|
||
# 简介
|
||
xplot = DomUtils.add_node(doc, root, "plot")
|
||
xplot.appendChild(doc.createCDATASection(tmdbinfo.get("overview") or ""))
|
||
xoutline = DomUtils.add_node(doc, root, "outline")
|
||
xoutline.appendChild(doc.createCDATASection(tmdbinfo.get("overview") or ""))
|
||
# 标题
|
||
DomUtils.add_node(doc, root, "title", "季 %s" % season)
|
||
# 发行日期
|
||
DomUtils.add_node(doc, root, "premiered", tmdbinfo.get("air_date") or "")
|
||
DomUtils.add_node(doc, root, "releasedate", tmdbinfo.get("air_date") or "")
|
||
# 发行年份
|
||
DomUtils.add_node(doc, root, "year", tmdbinfo.get("air_date")[:4] if tmdbinfo.get("air_date") else "")
|
||
# seasonnumber
|
||
DomUtils.add_node(doc, root, "seasonnumber", season)
|
||
# 保存
|
||
self.__save_nfo(doc, os.path.join(out_path, "season.nfo"))
|
||
|
||
def gen_tv_episode_nfo_file(self,
|
||
tmdbinfo: dict,
|
||
scraper_tv_nfo,
|
||
season: int,
|
||
episode: int,
|
||
out_path,
|
||
file_name):
|
||
"""
|
||
生成电视剧集的NFO描述文件
|
||
:param tmdbinfo: TMDB元数据
|
||
:param scraper_tv_nfo: 刮削配置
|
||
:param season: 季号
|
||
:param episode: 集号
|
||
:param out_path: 电视剧季的目录
|
||
:param file_name: 电视剧文件名，不含后缀
|
||
"""
|
||
# 开始生成集的信息
|
||
log.info("【Scraper】正在生成剧集NFO文件：%s" % file_name)
|
||
# 集的信息
|
||
episode_detail = {}
|
||
for episode_info in tmdbinfo.get("episodes") or []:
|
||
if int(episode_info.get("episode_number")) == int(episode):
|
||
episode_detail = episode_info
|
||
if not episode_detail:
|
||
return
|
||
doc = minidom.Document()
|
||
root = DomUtils.add_node(doc, doc, "episodedetails")
|
||
if scraper_tv_nfo.get("episode_basic"):
|
||
# 添加时间
|
||
DomUtils.add_node(doc, root, "dateadded", time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())))
|
||
# TMDBID
|
||
uniqueid = DomUtils.add_node(doc, root, "uniqueid", tmdbinfo.get("id") or "")
|
||
uniqueid.setAttribute("type", "tmdb")
|
||
uniqueid.setAttribute("default", "true")
|
||
# tmdbid
|
||
DomUtils.add_node(doc, root, "tmdbid", tmdbinfo.get("id") or "")
|
||
# 标题
|
||
DomUtils.add_node(doc, root, "title", episode_detail.get("name") or "第 %s 集" % episode)
|
||
# 简介
|
||
xplot = DomUtils.add_node(doc, root, "plot")
|
||
xplot.appendChild(doc.createCDATASection(episode_detail.get("overview") or ""))
|
||
xoutline = DomUtils.add_node(doc, root, "outline")
|
||
xoutline.appendChild(doc.createCDATASection(episode_detail.get("overview") or ""))
|
||
# 发布日期
|
||
DomUtils.add_node(doc, root, "aired", episode_detail.get("air_date") or "")
|
||
# 年份
|
||
DomUtils.add_node(doc, root, "year",
|
||
episode_detail.get("air_date")[:4] if episode_detail.get("air_date") else "")
|
||
# 季
|
||
DomUtils.add_node(doc, root, "season", season)
|
||
# 集
|
||
DomUtils.add_node(doc, root, "episode", episode)
|
||
# 评分
|
||
DomUtils.add_node(doc, root, "rating", episode_detail.get("vote_average") or "0")
|
||
if scraper_tv_nfo.get("episode_credits"):
|
||
# 导演
|
||
directors = episode_detail.get("crew") or []
|
||
for director in directors:
|
||
if director.get("known_for_department") == "Directing":
|
||
xdirector = DomUtils.add_node(doc, root, "director", director.get("name") or "")
|
||
xdirector.setAttribute("tmdbid", str(director.get("id") or ""))
|
||
# 演员
|
||
actors = episode_detail.get("guest_stars") or []
|
||
for actor in actors:
|
||
if actor.get("known_for_department") == "Acting":
|
||
xactor = DomUtils.add_node(doc, root, "actor")
|
||
DomUtils.add_node(doc, xactor, "name", actor.get("name") or "")
|
||
DomUtils.add_node(doc, xactor, "type", "Actor")
|
||
DomUtils.add_node(doc, xactor, "tmdbid", actor.get("id") or "")
|
||
# 保存文件
|
||
self.__save_nfo(doc, os.path.join(out_path, os.path.join(out_path, "%s.nfo" % file_name)))
|
||
|
||
@staticmethod
|
||
def __save_image(url, out_path, itype=''):
|
||
"""
|
||
下载poster.jpg并保存
|
||
"""
|
||
if not url or not out_path:
|
||
return
|
||
if itype:
|
||
image_path = os.path.join(out_path, "%s.%s" % (itype, str(url).split('.')[-1]))
|
||
else:
|
||
image_path = out_path
|
||
if os.path.exists(image_path):
|
||
return
|
||
try:
|
||
log.info(f"【Scraper】正在下载{itype}图片：{url} ...")
|
||
r = RequestUtils().get_res(url)
|
||
if r:
|
||
with open(file=image_path,
|
||
mode="wb") as img:
|
||
img.write(r.content)
|
||
log.info(f"【Scraper】{itype}图片已保存：{out_path}")
|
||
else:
|
||
log.info(f"【Scraper】{itype}图片下载失败，请检查网络连通性")
|
||
except Exception as err:
|
||
ExceptionUtils.exception_traceback(err)
|
||
|
||
@staticmethod
|
||
def __save_nfo(doc, out_file):
|
||
xml_str = doc.toprettyxml(indent=" ", encoding="utf-8")
|
||
with open(out_file, "wb") as xml_file:
|
||
xml_file.write(xml_str)
|
||
|
||
def gen_scraper_files(self, media, scraper_nfo, scraper_pic, dir_path, file_name, file_ext):
|
||
"""
|
||
刮削元数据
|
||
:param media: 已识别的媒体信息
|
||
:param scraper_nfo: NFO刮削配置
|
||
:param scraper_pic: 图片刮削配置
|
||
:param dir_path: 文件路径
|
||
:param file_name: 文件名，不含后缀
|
||
:param file_ext: 文件后缀
|
||
"""
|
||
if not scraper_nfo:
|
||
scraper_nfo = {}
|
||
if not scraper_pic:
|
||
scraper_pic = {}
|
||
try:
|
||
# 电影
|
||
if media.type == MediaType.MOVIE:
|
||
scraper_movie_nfo = scraper_nfo.get("movie")
|
||
scraper_movie_pic = scraper_pic.get("movie")
|
||
# movie nfo
|
||
if scraper_movie_nfo.get("basic") or scraper_movie_nfo.get("credits"):
|
||
# 已存在时不处理
|
||
if not os.path.exists(os.path.join(dir_path, "movie.nfo")) \
|
||
and not os.path.exists(os.path.join(dir_path, "%s.nfo" % file_name)):
|
||
# 查询Douban信息
|
||
if scraper_movie_nfo.get("credits") and scraper_movie_nfo.get("credits_chinese"):
|
||
doubaninfo = self.douban.get_douban_info(media)
|
||
else:
|
||
doubaninfo = None
|
||
# 生成电影描述文件
|
||
self.gen_movie_nfo_file(tmdbinfo=media.tmdb_info,
|
||
doubaninfo=doubaninfo,
|
||
scraper_movie_nfo=scraper_movie_nfo,
|
||
out_path=dir_path,
|
||
file_name=file_name)
|
||
# poster
|
||
if scraper_movie_pic.get("poster"):
|
||
poster_image = media.get_poster_image(original=True)
|
||
if poster_image:
|
||
self.__save_image(poster_image, dir_path, "poster")
|
||
# backdrop
|
||
if scraper_movie_pic.get("backdrop"):
|
||
backdrop_image = media.get_backdrop_image(default=False, original=True)
|
||
if backdrop_image:
|
||
self.__save_image(backdrop_image, dir_path, "fanart")
|
||
# background
|
||
if scraper_movie_pic.get("background"):
|
||
background_image = media.fanart.get_background(media_type=media.type, queryid=media.tmdb_id)
|
||
if background_image:
|
||
self.__save_image(background_image, dir_path, "background")
|
||
# logo
|
||
if scraper_movie_pic.get("logo"):
|
||
logo_image = media.fanart.get_logo(media_type=media.type, queryid=media.tmdb_id)
|
||
if logo_image:
|
||
self.__save_image(logo_image, dir_path, "logo")
|
||
# disc
|
||
if scraper_movie_pic.get("disc"):
|
||
disc_image = media.fanart.get_disc(media_type=media.type, queryid=media.tmdb_id)
|
||
if disc_image:
|
||
self.__save_image(disc_image, dir_path, "disc")
|
||
# banner
|
||
if scraper_movie_pic.get("banner"):
|
||
banner_image = media.fanart.get_banner(media_type=media.type, queryid=media.tmdb_id)
|
||
if banner_image:
|
||
self.__save_image(banner_image, dir_path, "banner")
|
||
# thumb
|
||
if scraper_movie_pic.get("thumb"):
|
||
thumb_image = media.fanart.get_thumb(media_type=media.type, queryid=media.tmdb_id)
|
||
if thumb_image:
|
||
self.__save_image(thumb_image, dir_path, "thumb")
|
||
# 电视剧
|
||
else:
|
||
scraper_tv_nfo = scraper_nfo.get("tv")
|
||
scraper_tv_pic = scraper_pic.get("tv")
|
||
# tv nfo
|
||
if not os.path.exists(os.path.join(os.path.dirname(dir_path), "tvshow.nfo")):
|
||
if scraper_tv_nfo.get("basic") or scraper_tv_nfo.get("credits"):
|
||
# 查询Douban信息
|
||
if scraper_tv_nfo.get("credits") and scraper_tv_nfo.get("credits_chinese"):
|
||
doubaninfo = self.douban.get_douban_info(media)
|
||
else:
|
||
doubaninfo = None
|
||
# 根目录描述文件
|
||
self.gen_tv_nfo_file(media.tmdb_info, doubaninfo, scraper_tv_nfo, os.path.dirname(dir_path))
|
||
# poster
|
||
if scraper_tv_pic.get("poster"):
|
||
poster_image = media.get_poster_image(original=True)
|
||
if poster_image:
|
||
self.__save_image(poster_image, os.path.dirname(dir_path), "poster")
|
||
# backdrop
|
||
if scraper_tv_pic.get("backdrop"):
|
||
backdrop_image = media.get_backdrop_image(default=False, original=True)
|
||
if backdrop_image:
|
||
self.__save_image(backdrop_image, os.path.dirname(dir_path), "fanart")
|
||
# background
|
||
if scraper_tv_pic.get("background"):
|
||
background_image = media.fanart.get_background(media_type=media.type, queryid=media.tvdb_id)
|
||
if background_image:
|
||
self.__save_image(background_image, os.path.dirname(dir_path), "show")
|
||
# logo
|
||
if scraper_tv_pic.get("logo"):
|
||
logo_image = media.fanart.get_logo(media_type=media.type, queryid=media.tvdb_id)
|
||
if logo_image:
|
||
self.__save_image(logo_image, os.path.dirname(dir_path), "logo")
|
||
# clearart
|
||
if scraper_tv_pic.get("clearart"):
|
||
clearart_image = media.fanart.get_disc(media_type=media.type, queryid=media.tvdb_id)
|
||
if clearart_image:
|
||
self.__save_image(clearart_image, os.path.dirname(dir_path), "clearart")
|
||
# banner
|
||
if scraper_tv_pic.get("banner"):
|
||
banner_image = media.fanart.get_banner(media_type=media.type, queryid=media.tvdb_id)
|
||
if banner_image:
|
||
self.__save_image(banner_image, os.path.dirname(dir_path), "banner")
|
||
# thumb
|
||
if scraper_tv_pic.get("thumb"):
|
||
thumb_image = media.fanart.get_thumb(media_type=media.type, queryid=media.tvdb_id)
|
||
if thumb_image:
|
||
self.__save_image(thumb_image, os.path.dirname(dir_path), "thumb")
|
||
# season nfo
|
||
if scraper_tv_nfo.get("season_basic"):
|
||
if not os.path.exists(os.path.join(dir_path, "season.nfo")):
|
||
# season nfo
|
||
seasoninfo = self.media.get_tmdb_tv_season_detail(tmdbid=media.tmdb_id,
|
||
season=int(media.get_season_seq()))
|
||
if seasoninfo:
|
||
self.gen_tv_season_nfo_file(seasoninfo, int(media.get_season_seq()), dir_path)
|
||
# episode nfo
|
||
if scraper_tv_nfo.get("episode_basic") \
|
||
or scraper_tv_nfo.get("episode_credits"):
|
||
if not os.path.exists(os.path.join(dir_path, "%s.nfo" % file_name)):
|
||
seasoninfo = self.media.get_tmdb_tv_season_detail(tmdbid=media.tmdb_id,
|
||
season=int(media.get_season_seq()))
|
||
if seasoninfo:
|
||
self.gen_tv_episode_nfo_file(tmdbinfo=seasoninfo,
|
||
scraper_tv_nfo=scraper_tv_nfo,
|
||
season=int(media.get_season_seq()),
|
||
episode=int(media.get_episode_seq()),
|
||
out_path=dir_path,
|
||
file_name=file_name)
|
||
# season poster
|
||
if scraper_tv_pic.get("season_poster"):
|
||
season_poster = "season%s-poster" % media.get_season_seq().rjust(2, '0')
|
||
seasonposter = media.fanart.get_seasonposter(media_type=media.type,
|
||
queryid=media.tvdb_id,
|
||
season=media.get_season_seq())
|
||
if seasonposter:
|
||
self.__save_image(seasonposter,
|
||
os.path.dirname(dir_path),
|
||
season_poster)
|
||
else:
|
||
seasoninfo = self.media.get_tmdb_tv_season_detail(tmdbid=media.tmdb_id,
|
||
season=int(media.get_season_seq()))
|
||
if seasoninfo:
|
||
self.__save_image(TMDB_IMAGE_W500_URL % seasoninfo.get("poster_path"),
|
||
os.path.dirname(dir_path),
|
||
season_poster)
|
||
# season banner
|
||
if scraper_tv_pic.get("season_banner"):
|
||
seasonbanner = media.fanart.get_seasonbanner(media_type=media.type,
|
||
queryid=media.tvdb_id,
|
||
season=media.get_season_seq())
|
||
if seasonbanner:
|
||
self.__save_image(seasonbanner,
|
||
os.path.dirname(dir_path),
|
||
"season%s-banner" % media.get_season_seq().rjust(2, '0'))
|
||
# season thumb
|
||
if scraper_tv_pic.get("season_thumb"):
|
||
seasonthumb = media.fanart.get_seasonthumb(media_type=media.type,
|
||
queryid=media.tvdb_id,
|
||
season=media.get_season_seq())
|
||
if seasonthumb:
|
||
self.__save_image(seasonthumb,
|
||
os.path.dirname(dir_path),
|
||
"season%s-landscape" % media.get_season_seq().rjust(2, '0'))
|
||
# episode thumb
|
||
if scraper_tv_pic.get("episode_thumb"):
|
||
episode_thumb = os.path.join(dir_path, file_name + "-thumb.jpg")
|
||
if not os.path.exists(episode_thumb):
|
||
# 优先从TMDB查询
|
||
episode_image = self.media.get_episode_images(tv_id=media.tmdb_id,
|
||
season_id=media.get_season_seq(),
|
||
episode_id=media.get_episode_seq(),
|
||
orginal=True)
|
||
if episode_image:
|
||
self.__save_image(episode_image, episode_thumb)
|
||
else:
|
||
# 开启ffmpeg，则从视频文件生成缩略图
|
||
if scraper_tv_pic.get("episode_thumb_ffmpeg"):
|
||
video_path = os.path.join(dir_path, file_name + file_ext)
|
||
log.info(f"【Scraper】正在生成缩略图：{video_path} ...")
|
||
FfmpegHelper().get_thumb_image_from_video(video_path=video_path,
|
||
image_path=episode_thumb)
|
||
log.info(f"【Scraper】缩略图生成完成：{episode_thumb}")
|
||
|
||
except Exception as e:
|
||
ExceptionUtils.exception_traceback(e)
|
||
|
||
def __gen_people_chinese_info(self, directors, actors, doubaninfo):
|
||
"""
|
||
匹配豆瓣演职人员中文名
|
||
"""
|
||
if doubaninfo:
|
||
directors_douban = doubaninfo.get("directors") or []
|
||
actors_douban = doubaninfo.get("actors") or []
|
||
# douban英文名姓和名分开匹配，（豆瓣中名前姓后，TMDB中不确定）
|
||
for director_douban in directors_douban:
|
||
if director_douban["latin_name"]:
|
||
director_douban["latin_name"] = director_douban.get("latin_name", "").lower().split(" ")
|
||
else:
|
||
director_douban["latin_name"] = director_douban.get("name", "").lower().split(" ")
|
||
for actor_douban in actors_douban:
|
||
if actor_douban["latin_name"]:
|
||
actor_douban["latin_name"] = actor_douban.get("latin_name", "").lower().split(" ")
|
||
else:
|
||
actor_douban["latin_name"] = actor_douban.get("name", "").lower().split(" ")
|
||
# 导演
|
||
if directors:
|
||
for director in directors:
|
||
director_douban = self.__match_people_in_douban(director, directors_douban)
|
||
if director_douban:
|
||
director["name"] = director_douban.get("name")
|
||
else:
|
||
log.info("【Scraper】豆瓣该影片或剧集无导演 %s 信息" % director.get("name"))
|
||
# 演员
|
||
if actors:
|
||
for actor in actors:
|
||
actor_douban = self.__match_people_in_douban(actor, actors_douban)
|
||
if actor_douban:
|
||
actor["name"] = actor_douban.get("name")
|
||
if actor_douban.get("character") != "演员":
|
||
actor["character"] = actor_douban.get("character")[2:]
|
||
else:
|
||
log.info("【Scraper】豆瓣该影片或剧集无演员 %s 信息" % actor.get("name"))
|
||
else:
|
||
log.info("【Scraper】豆瓣无该影片或剧集信息")
|
||
return directors, actors
|
||
|
||
def __match_people_in_douban(self, people, peoples_douban):
|
||
"""
|
||
名字加又名构成匹配列表
|
||
"""
|
||
people_aka_names = self.media.get_tmdbperson_aka_names(people.get("id")) or []
|
||
people_aka_names.append(people.get("name"))
|
||
for people_aka_name in people_aka_names:
|
||
for people_douban in peoples_douban:
|
||
latin_match_res = True
|
||
# 姓和名分开匹配
|
||
for latin_name in people_douban.get("latin_name"):
|
||
latin_match_res = latin_match_res and (latin_name in people_aka_name.lower())
|
||
if latin_match_res or (people_douban.get("name") == people_aka_name):
|
||
return people_douban
|
||
return None
|