codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix personmeta

Browse Source
This commit is contained in:
jxxghp 2023-10-01 18:11:01 +08:00
parent 1293fafd34
commit e21ba5ad51
1 changed files with 87 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
app/plugins/personmeta/__init__.py
View File

@ -63,6 +63,7 @@ class PersonMeta(_PluginBase):
_onlyonce = False _onlyonce = False
_cron = None _cron = None
_delay = 0 _delay = 0
_remove_nozh = False
def init_plugin(self, config: dict = None): def init_plugin(self, config: dict = None):
self.tmdbchain = TmdbChain(self.db) self.tmdbchain = TmdbChain(self.db)
@ -72,6 +73,7 @@ class PersonMeta(_PluginBase):
self._onlyonce = config.get("onlyonce") self._onlyonce = config.get("onlyonce")
self._cron = config.get("cron") self._cron = config.get("cron")
self._delay = config.get("delay") or 0 self._delay = config.get("delay") or 0
self._remove_nozh = config.get("remove_nozh") or False
# 停止现有任务 # 停止现有任务
self.stop_service() self.stop_service()
@ -113,7 +115,8 @@ class PersonMeta(_PluginBase):
"enabled": self._enabled, "enabled": self._enabled,
"onlyonce": self._onlyonce, "onlyonce": self._onlyonce,
"cron": self._cron, "cron": self._cron,
"delay": self._delay "delay": self._delay,
"remove_nozh": self._remove_nozh
}) })
def get_state(self) -> bool: def get_state(self) -> bool:
@ -209,6 +212,27 @@ class PersonMeta(_PluginBase):
] ]
} }
] ]
},
{
'component': 'VRow',
'content': [
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 6
},
'content': [
{
'component': 'VSwitch',
'props': {
'model': 'remove_nozh',
'label': '删除非中文演员',
}
}
]
}
]
} }
] ]
} }
@ -216,7 +240,8 @@ class PersonMeta(_PluginBase):
"enabled": False, "enabled": False,
"onlyonce": False, "onlyonce": False,
"cron": "", "cron": "",
"delay": 30 "delay": 30,
"remove_nozh": False
} }
def get_page(self) -> List[dict]: def get_page(self) -> List[dict]:
@ -325,12 +350,16 @@ class PersonMeta(_PluginBase):
if not people.get("Name"): if not people.get("Name"):
continue continue
if StringUtils.is_chinese(people.get("Name")): if StringUtils.is_chinese(people.get("Name")):
peoples.append(people)
continue continue
if self._event.is_set():
logger.info(f"演职人员刮削服务停止")
return
info = self.__update_people(server=server, people=people, info = self.__update_people(server=server, people=people,
douban_actors=douban_actors) douban_actors=douban_actors)
if info: if info:
peoples.append(info) peoples.append(info)
else: elif not self._remove_nozh:
peoples.append(people) peoples.append(people)
# 保存媒体项信息 # 保存媒体项信息
if peoples: if peoples:
@ -356,23 +385,28 @@ class PersonMeta(_PluginBase):
# 更新季媒体项人物 # 更新季媒体项人物
peoples = [] peoples = []
if seasoninfo.get("People"): if seasoninfo.get("People"):
logger.info(f"开始更新季 {seasoninfo.get('Id')} 的人物信息 ...")
for people in seasoninfo["People"]: for people in seasoninfo["People"]:
if not people.get("Name"): if not people.get("Name"):
continue continue
if StringUtils.is_chinese(people.get("Name")): if StringUtils.is_chinese(people.get("Name")):
peoples.append(people)
continue continue
if self._event.is_set():
logger.info(f"演职人员刮削服务停止")
return
# 更新人物信息 # 更新人物信息
info = self.__update_people(server=server, people=people, info = self.__update_people(server=server, people=people,
douban_actors=season_actors) douban_actors=season_actors)
if info: if info:
peoples.append(info) peoples.append(info)
else: elif not self._remove_nozh:
peoples.append(people) peoples.append(people)
# 保存季媒体项信息 # 保存季媒体项信息
if peoples: if peoples:
seasoninfo["People"] = peoples seasoninfo["People"] = peoples
self.set_iteminfo(server=server, itemid=season.get("Id"), iteminfo=seasoninfo) self.set_iteminfo(server=server, itemid=season.get("Id"), iteminfo=seasoninfo)
logger.info(f"季 {seasoninfo.get('Id')} 的人物信息更新完成")
# 获取集媒体项 # 获取集媒体项
episodes = self.get_items(server=server, parentid=season.get("Id"), mtype="Episode") episodes = self.get_items(server=server, parentid=season.get("Id"), mtype="Episode")
if not episodes: if not episodes:
@ -387,23 +421,29 @@ class PersonMeta(_PluginBase):
continue continue
# 更新集媒体项人物 # 更新集媒体项人物
if episodeinfo.get("People"): if episodeinfo.get("People"):
logger.info(f"开始更新集 {episodeinfo.get('Id')} 的人物信息 ...")
peoples = [] peoples = []
for people in episodeinfo["People"]: for people in episodeinfo["People"]:
if not people.get("Name"): if not people.get("Name"):
continue continue
if StringUtils.is_chinese(people.get("Name")): if StringUtils.is_chinese(people.get("Name")):
peoples.append(people)
continue continue
if self._event.is_set():
logger.info(f"演职人员刮削服务停止")
return
# 更新人物信息 # 更新人物信息
info = self.__update_people(server=server, people=people, info = self.__update_people(server=server, people=people,
douban_actors=season_actors) douban_actors=season_actors)
if info: if info:
peoples.append(info) peoples.append(info)
else: elif not self._remove_nozh:
peoples.append(people) peoples.append(people)
# 保存集媒体项信息 # 保存集媒体项信息
if peoples: if peoples:
episodeinfo["People"] = peoples episodeinfo["People"] = peoples
self.set_iteminfo(server=server, itemid=episode.get("Id"), iteminfo=episodeinfo) self.set_iteminfo(server=server, itemid=episode.get("Id"), iteminfo=episodeinfo)
logger.info(f"集 {episodeinfo.get('Id')} 的人物信息更新完成")
def __update_people(self, server: str, people: dict, douban_actors: list = None) -> Optional[dict]: def __update_people(self, server: str, people: dict, douban_actors: list = None) -> Optional[dict]:
""" """
@ -434,25 +474,14 @@ class PersonMeta(_PluginBase):
# 查询媒体库人物详情 # 查询媒体库人物详情
personinfo = self.get_iteminfo(server=server, itemid=people.get("Id")) personinfo = self.get_iteminfo(server=server, itemid=people.get("Id"))
if not personinfo: if not personinfo:
logger.debug(f"未找到人物 {people.get('Id')} 的信息") logger.debug(f"未找到人物 {people.get('Name')} 的信息")
return None return None
# 是否更新标志 # 是否更新标志
updated_name = False updated_name = False
updated_overview = False updated_overview = False
profile_path = None profile_path = None
# 从豆瓣演员中匹配中文名称
if douban_actors:
for douban_actor in douban_actors:
if douban_actor.get("latin_name") == people.get("Name"):
# 名称
personinfo["Name"] = douban_actor.get("name")
ret_people["Name"] = douban_actor.get("name")
updated_name = True
# 图片
if douban_actor.get("avatar", {}).get("large"):
profile_path = douban_actor.get("avatar", {}).get("large")
break
if not updated_name or not updated_overview:
# 获取人物的TMDBID # 获取人物的TMDBID
person_tmdbid, person_imdbid = __get_peopleid(personinfo) person_tmdbid, person_imdbid = __get_peopleid(personinfo)
if person_tmdbid: if person_tmdbid:
@ -460,6 +489,7 @@ class PersonMeta(_PluginBase):
if person_tmdbinfo: if person_tmdbinfo:
cn_name = self.__get_chinese_name(person_tmdbinfo) cn_name = self.__get_chinese_name(person_tmdbinfo)
if cn_name: if cn_name:
logger.info(f"{people.get('Name')} 从TMDB获取到中文名：{cn_name}")
# 更新中文名 # 更新中文名
personinfo["Name"] = cn_name personinfo["Name"] = cn_name
ret_people["Name"] = cn_name ret_people["Name"] = cn_name
@ -471,6 +501,27 @@ class PersonMeta(_PluginBase):
personinfo["Overview"] = biography personinfo["Overview"] = biography
# 图片 # 图片
profile_path = f"https://image.tmdb.org/t/p/original{person_tmdbinfo.get('profile_path')}" profile_path = f"https://image.tmdb.org/t/p/original{person_tmdbinfo.get('profile_path')}"
if not updated_name or not updated_overview:
# 从豆瓣演员中匹配中文名称
if douban_actors:
for douban_actor in douban_actors:
if douban_actor.get("latin_name") == people.get("Name"):
logger.info(f"{people.get('Name')} 从豆瓣中获取到中文名：{douban_actor.get('name')}")
# 名称
personinfo["Name"] = douban_actor.get("name")
ret_people["Name"] = douban_actor.get("name")
updated_name = True
# 图片
if douban_actor.get("avatar", {}).get("large"):
profile_path = douban_actor.get("avatar", {}).get("large")
break
# 更新人物图片
if profile_path:
logger.info(f"更新人物 {people.get('Name')} 的图片：{profile_path}")
self.set_item_image(server=server, itemid=people.get("Id"), imageurl=profile_path)
# 锁定人物信息 # 锁定人物信息
if updated_name: if updated_name:
if "Name" not in personinfo["LockedFields"]: if "Name" not in personinfo["LockedFields"]:
@ -479,17 +530,14 @@ class PersonMeta(_PluginBase):
if "Overview" not in personinfo["LockedFields"]: if "Overview" not in personinfo["LockedFields"]:
personinfo["LockedFields"].append("Overview") personinfo["LockedFields"].append("Overview")
# 更新人物图片
if profile_path:
logger.info(f"更新人物 {people.get('Id')} 的图片：{profile_path}")
self.set_item_image(server=server, itemid=people.get("Id"), imageurl=profile_path)
# 更新人物信息 # 更新人物信息
if updated_name or updated_overview: if updated_name or updated_overview:
logger.info(f"更新人物 {people.get('Id')} 的信息：{personinfo}") logger.info(f"更新人物 {people.get('Name')} 的信息：{personinfo}")
ret = self.set_iteminfo(server=server, itemid=people.get("Id"), iteminfo=personinfo) ret = self.set_iteminfo(server=server, itemid=people.get("Id"), iteminfo=personinfo)
if ret: if ret:
return ret_people return ret_people
else:
logger.info(f"人物 {people.get('Name')} 未找到中文数据")
except Exception as err: except Exception as err:
logger.error(f"更新人物信息失败：{err}") logger.error(f"更新人物信息失败：{err}")
return None return None