95 lines
3.8 KiB
Python
95 lines
3.8 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
import os
|
|
import sys
|
|
|
|
from analyze import Analyze
|
|
from tmdb import Tmdb
|
|
from utils.LoggerUtil import Logger
|
|
|
|
|
|
def __init_logger(log_file="tmdb.log", level="info", back_count=3):
|
|
"""
|
|
服务日志记录对象
|
|
:param log_file: 日志文件名
|
|
:param level: 日志记录级别。debug info warning error crit
|
|
:param back_count: 日志文件备份天数
|
|
:return: 日志对象
|
|
"""
|
|
# 获取当前文件路径
|
|
current_path = os.path.abspath(__file__)
|
|
# 获取当前文件的父目录
|
|
father_path = os.path.abspath(os.path.dirname(current_path) + os.path.sep + ".")
|
|
# (必填)日志文件名 log_file="/data/ws_env/logtest/process.log"
|
|
log_file_abspath = os.path.join(father_path, "logs", log_file)
|
|
return Logger(log_file_abspath, level=level, backCount=back_count)
|
|
|
|
|
|
|
|
|
|
|
|
def __check_version():
|
|
version_info = sys.version_info
|
|
if 3 > version_info.major:
|
|
log.logger.error("当前Python版本不能小于3!")
|
|
raise SystemExit(1)
|
|
else:
|
|
if 8 > version_info.minor:
|
|
log.logger.error("当前Python版本不能小于3.8!")
|
|
raise SystemExit(1)
|
|
elif 8 == version_info.minor:
|
|
log.logger.warn("推荐使用Python 3.9 及以上版本!")
|
|
|
|
|
|
def __execute(dir_path, output, tmdb_token):
|
|
log.logger.info("------------------- 开始获取演员元数据及海报 -------------------")
|
|
__file_paths = []
|
|
log.logger.info("当前执行元数据刮削识别的根文件夹:{0}".format(dir_path))
|
|
for folder in os.listdir(dir_path):
|
|
__folder2 = os.path.join(dir_path, folder)
|
|
# 判断是否文件夹
|
|
if os.path.isdir(__folder2):
|
|
for nfo_file in os.listdir(__folder2):
|
|
__child_file_path = os.path.join(__folder2, nfo_file)
|
|
if ".nfo" in os.path.basename(__child_file_path):
|
|
__file_paths.append(__child_file_path)
|
|
elif os.path.isfile(__folder2):
|
|
__file_name = os.path.basename(__folder2)
|
|
if ".nfo" in __file_name:
|
|
__file_paths.append(__folder2)
|
|
for __file_path in __file_paths:
|
|
log.logger.info("开始处理元数据刮削识别:{0}".format(__file_path))
|
|
# __file_path = "example/神出鬼没 (2023) - 2160p.nfo"
|
|
__nfo_data = Analyze(file_path=__file_path).analyze()
|
|
for __actor in __nfo_data["actors"]:
|
|
__tmdbid = __actor["tmdbid"]
|
|
__actor_name = __actor["name"]
|
|
__name = __actor_name[1].lower()
|
|
__full_actor_name = __actor_name + "-tmdb-" + __tmdbid
|
|
__path_dir = os.path.join(output, __name, __full_actor_name)
|
|
if not os.path.exists(__path_dir):
|
|
os.makedirs(__path_dir)
|
|
# 如果存在元数据则不再进行刮削
|
|
if "person.nfo" not in os.listdir(__path_dir):
|
|
Tmdb(log=log, tmdb_id=__tmdbid, actor_path=__path_dir, tmdb_token=tmdb_token).create_actor_nfo()
|
|
# 如果存在海报则不再进行刮削
|
|
if "folder.jpg" not in os.listdir(__path_dir):
|
|
Tmdb(log=log, tmdb_id=__tmdbid, actor_path=__path_dir, tmdb_token=tmdb_token).get_actor_image()
|
|
log.logger.info("------------------- 结束获取演员元数据及海报 -------------------")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# 扫描目录
|
|
__dir_path = "example/tvs"
|
|
# 输出演员元数据目录
|
|
__output = "data/metadata/person"
|
|
# TMDB API TOKEN
|
|
__tmdb_token = "tmdb_token"
|
|
# 初始化日志
|
|
log = __init_logger()
|
|
# 检查python版本
|
|
__check_version()
|
|
# 开始执行主程序
|
|
# 默认 language="zh-CN" (简体中文),可以通过修改 "language" 的值变更获取元数据的语言类别
|
|
__execute(dir_path=__dir_path, output=__output, tmdb_token=__tmdb_token)
|