add: 补充README.md说明

change: 优化person.py代码
This commit is contained in:
Carol 2023-09-06 10:59:34 +08:00
parent e7c7d22400
commit 140c08898f
2 changed files with 24 additions and 17 deletions

View File

@ -45,6 +45,10 @@ pip3 install json
* 例如电影存放于 `./movies/` 下,则该路径填写 `./movies` 的完整路径。
* 为了兼容 `电视剧` 中不刮削 `季` 中的 `.nfo` 内容,因此只刮削 `__dir_path` 路径下一层文件夹及当前层下的 `.nfo` 文件(兼容./movies 下同级存放的视频及.nfo文件)
* __output: 演员信息、图片输出路径
* __tmdb_token: TMDB 开发者API调用token
* 登录 [TMDB](https://www.themoviedb.org/login)
* 访问 [API](https://www.themoviedb.org/settings/api)
* 复制 `API 读访问令牌`
**`目录结构说明`**
- ./movies
@ -60,15 +64,8 @@ pip3 install json
- Season 1 (不刮削)
### 运行
**`方法一:`**
修改 `person.py` 文件中 `if __name__ == '__main__':` 方法中 `__dir_path` 参数值和 `__output` 参数值
**`方法二:`**
> 通过命令行方式设置参数
```python
python3 person.py {"__dir_path":"/volume1/video/movies", "__output":"/volume1/metadata/person"}
```

View File

@ -35,16 +35,17 @@ class Analyze:
class Tmdb:
def __init__(self, tmdb_id, actor_path):
def __init__(self, tmdb_id, actor_path, tmdb_token):
self.image_path = None
self.tmdb_id = tmdb_id
self.actor_path = actor_path
self.tmdb_token = tmdb_token
def get_actor_info(self):
url = "https://api.themoviedb.org/3/person/" + self.tmdb_id + "?language=zh-CN"
headers = {
"accept": "application/json",
"Authorization": "Bearer eyJhbGciOiJIUzI1NiJ9.eyJhdWQiOiIxYTU4ODAxMGY5OTUwYWEyNThhYjFhYjJlMjI4NGVmYSIsInN1YiI6IjYxYmRmOGNjMzgzZGYyMDA0MjIzNDhjOSIsInNjb3BlcyI6WyJhcGlfcmVhZCJdLCJ2ZXJzaW9uIjoxfQ.RPG8F8AELlK7MgrXDR2U0YRv61VteZZ9ponilnkQqkE"
"Authorization": "Bearer " + self.tmdb_token
}
response = requests.get(url, headers=headers)
return response.text
@ -60,12 +61,10 @@ class Tmdb:
f.write(response.content)
if __name__ == '__main__':
__dir_path = "example/movies"
__output = "data/metadata/person"
def __execute(dir_path, output, tmdb_token):
__file_paths = []
for folder in os.listdir(__dir_path):
__folder2 = os.path.join(__dir_path, folder)
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):
@ -85,11 +84,22 @@ if __name__ == '__main__':
__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)
__path_dir = os.path.join(output, __name, __full_actor_name)
if not os.path.exists(__path_dir):
os.makedirs(__path_dir)
if ".nfo" not in os.listdir(__path_dir):
__actor_info = Tmdb(tmdb_id=__tmdbid, actor_path=__path_dir).get_actor_info()
__actor_info = Tmdb(tmdb_id=__tmdbid, actor_path=__path_dir, tmdb_token=tmdb_token).get_actor_info()
print(__actor_info)
if "folder" not in os.listdir(__path_dir):
Tmdb(tmdb_id=__tmdbid, actor_path=__path_dir).get_actor_image()
Tmdb(tmdb_id=__tmdbid, actor_path=__path_dir, tmdb_token=tmdb_token).get_actor_image()
if __name__ == '__main__':
# 扫描目录
__dir_path = "example/movies"
# 输出演员元数据目录
__output = "data/metadata/person"
# TMDB API TOKEN
__tmdb_token = "eyJhbGciOiJIUzI1NiJ9.eyJhdWQiOiIxYTU4ODAxMGY5OTUwYWEyNThhYjFhYjJlMjI4NGVmYSIsInN1YiI6IjYxYmRmOGNjMzgzZGYyMDA0MjIzNDhjOSIsInNjb3BlcyI6WyJhcGlfcmVhZCJdLCJ2ZXJzaW9uIjoxfQ.RPG8F8AELlK7MgrXDR2U0YRv61VteZZ9ponilnkQqkE"
# 开始执行主程序
__execute(dir_path=__dir_path, output=__output, tmdb_token=__tmdb_token)