120 lines
4.8 KiB
Python
120 lines
4.8 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
|
||
import requests, os
|
||
from configparser import RawConfigParser
|
||
from base64 import b64encode
|
||
from traceback import format_exc
|
||
from json import loads
|
||
from os.path import exists
|
||
|
||
# 检查“actor_photos”文件夹是否就绪
|
||
if not exists('actor_photos'):
|
||
print('\n“actor_photos” folder lost！Please put it to the same location with program\n')
|
||
os.system('pause')
|
||
# 读取配置文件，这个ini文件用来给用户设置emby网址和api id
|
||
print('Reading ini Setting...')
|
||
config_settings = RawConfigParser()
|
||
try:
|
||
config_settings.read('ap_config.ini', encoding='utf-8-sig')
|
||
url_emby = config_settings.get("emby/jellyfin", "website")
|
||
api_key = config_settings.get("emby/jellyfin", "api id")
|
||
bool_replace = True if config_settings.get("emby/jellyfin", "是否覆盖以前上传的头像？") == '是' else False
|
||
except:
|
||
print(format_exc())
|
||
print('Cannot read ini files!')
|
||
os.system('pause')
|
||
print('Read Success!\n')
|
||
# 修正用户输入的emby网址，无论是不是带“/”
|
||
if not url_emby.endswith('/'):
|
||
url_emby += '/'
|
||
# 成功的个数
|
||
num_suc = 0
|
||
num_fail = 0
|
||
num_exist = 0
|
||
sep = os.sep
|
||
try:
|
||
print('Getting emby/jellyfin Person`s form...')
|
||
# curl -X GET "http://localhost:8096/emby/Persons?api_key=3291434710e342089565ad05b6b2f499" -H "accept: application/json"
|
||
# 得到所有“人员” emby api没有细分“演员”还是“导演”“编剧”等等 下面得到的是所有“有关人员”
|
||
url_emby_persons = url_emby + 'emby/Persons?api_key=' + api_key # &PersonTypes=Actor
|
||
try:
|
||
rqs_emby = requests.get(url=url_emby_persons)
|
||
except requests.exceptions.ConnectionError:
|
||
print('Cannot connect to emby/jellyfin server ,Please check：', url_emby, '\n')
|
||
os.system('pause')
|
||
except:
|
||
print(format_exc())
|
||
print('Unkown Error ,Please submit screenshot to issues', url_emby, '\n')
|
||
os.system('pause')
|
||
# 401，无权访问
|
||
if rqs_emby.status_code == 401:
|
||
print('Please check API KEY！\n')
|
||
os.system('pause')
|
||
# print(rqs_emby.text)
|
||
try:
|
||
list_persons = loads(rqs_emby.text)['Items']
|
||
except:
|
||
print(rqs_emby.text)
|
||
print('Error! emby response：')
|
||
print('Please submit screenshot to issues！')
|
||
os.system('pause')
|
||
num_persons = len(list_persons)
|
||
print('There are currently' + str(num_persons) + ' People！\n')
|
||
# os.system('pause')
|
||
# 用户emby中的persons，在“actor_photos”文件夹中，已有头像的，记录下来
|
||
f_txt = open("included.txt", 'w', encoding="utf-8")
|
||
f_txt.close()
|
||
f_txt = open("no_included.txt", 'w', encoding="utf-8")
|
||
f_txt.close()
|
||
for dic_each_actor in list_persons:
|
||
actor_name = dic_each_actor['Name']
|
||
# 头像jpg/png在“actor_photos”中的路径
|
||
actor_pic_path = 'actor_photos' + sep + actor_name[0] + sep + actor_name
|
||
if exists(actor_pic_path + '.jpg'):
|
||
actor_pic_path = actor_pic_path + '.jpg'
|
||
header = {"Content-Type": 'image/jpeg', }
|
||
elif exists(actor_pic_path + '.png'):
|
||
actor_pic_path = actor_pic_path + '.png'
|
||
header = {"Content-Type": 'image/png', }
|
||
else:
|
||
print('>>No image：', actor_name)
|
||
f_txt = open("no_included.txt", 'a', encoding="utf-8")
|
||
f_txt.write(actor_name + '\n')
|
||
f_txt.close()
|
||
num_fail += 1
|
||
continue
|
||
# emby有某个演员，“actor_photos”文件夹也有这个演员的头像，记录一下
|
||
f_txt = open("included.txt", 'a', encoding="utf-8")
|
||
f_txt.write(actor_name + '\n')
|
||
f_txt.close()
|
||
# emby有某个演员，已经有他的头像，不再进行下面“上传头像”的操作
|
||
if dic_each_actor['ImageTags']: # emby已经收录头像
|
||
num_exist += 1
|
||
if not bool_replace: # 不需要覆盖已有头像
|
||
continue # 那么不进行下面的上传操作
|
||
f_pic = open(actor_pic_path, 'rb') # 二进制方式打开图文件
|
||
b6_pic = b64encode(f_pic.read()) # 读取文件内容，转换为base64编码
|
||
f_pic.close()
|
||
url_post_img = url_emby + 'emby/Items/' + dic_each_actor['Id'] + '/Images/Primary?api_key=' + api_key
|
||
requests.post(url=url_post_img, data=b6_pic, headers=header)
|
||
print('>>Success：', actor_name)
|
||
num_suc += 1
|
||
|
||
print('\nemby/jellyfin people: ', num_persons, '')
|
||
print('include photos: ', num_exist, '')
|
||
if bool_replace:
|
||
print('Mode：Overwrite existed images')
|
||
else:
|
||
print('Mode：Skip existed images')
|
||
print('Success Upload', num_suc, '个！')
|
||
print('No images', num_fail, '个！')
|
||
print('Saved to “no_included.txt”\n')
|
||
os.system('pause')
|
||
except:
|
||
print(format_exc())
|
||
os.system('pause')
|
||
|
||
|