227 lines
9.2 KiB
Python
227 lines
9.2 KiB
Python
# -*- coding:utf-8 -*-
|
||
# Gfriends Inputer / 女友头像仓库导入工具
|
||
# By xinxin8816, Many thanks for junerain123, ddd354, moyy996.
|
||
|
||
import requests, os, sys
|
||
from configparser import RawConfigParser
|
||
from base64 import b64encode
|
||
from traceback import format_exc
|
||
from json import loads
|
||
from PIL import Image, ImageFilter
|
||
from alive_progress import alive_bar
|
||
|
||
|
||
def fix_size(path):
|
||
pic = Image.open(path)
|
||
(x, y) = pic.size
|
||
if not 2 / 3 - 0.05 <= x / y <= 2 / 3 + 0.05: # 仅处理会过度拉伸的图片
|
||
fixed_pic = pic.resize((int(x), int(3 / 2 * x))) # 拉伸图片
|
||
fixed_pic = fixed_pic.filter(ImageFilter.GaussianBlur(radius=50)) # 高斯模糊
|
||
fixed_pic.paste(pic, (0, int((3 / 2 * x - y) / 2))) # 粘贴原图
|
||
fixed_pic.save(path)
|
||
|
||
|
||
def get_gfriends_map(repository_url):
|
||
print('下载头像仓库文件树...')
|
||
if repository_url == '默认/':
|
||
repository_url = 'https://raw.githubusercontent.com/xinxin8816/gfriends/master/'
|
||
github_template = repository_url + '{}/{}/{}'
|
||
request_url = repository_url + 'Filetree.json'
|
||
try:
|
||
if proxy == '不使用':
|
||
response = session.get(request_url)
|
||
else:
|
||
response = session.get(request_url, proxies=proxies)
|
||
except:
|
||
print(format_exc())
|
||
print('网络连接异常且重试 ' + max_retries + ' 次失败')
|
||
print('请尝试开启全局代理或配置 HTTP 局部代理；若已开启代理，请检查其可用性')
|
||
sys.exit()
|
||
if response.status_code != 200:
|
||
print('女友仓库返回了一个错误： {}'.format(response.status_code))
|
||
sys.exit()
|
||
|
||
map_json = loads(response.content)
|
||
output = {}
|
||
|
||
first_lvls = map_json.keys()
|
||
for first in first_lvls:
|
||
second_lvls = map_json[first].keys()
|
||
for second in second_lvls:
|
||
for k, v in map_json[first][second].items():
|
||
output[k[:-4]] = github_template.format(first, second, v)
|
||
|
||
print('读取头像仓库文件树完成')
|
||
print('当前仓库头像数量：' + str(response.text.count('\n')) + '枚\n')
|
||
return output
|
||
|
||
|
||
def get_gfriends_link(name):
|
||
if name in gfriends_map:
|
||
output = gfriends_map[name]
|
||
return output
|
||
else:
|
||
return None
|
||
|
||
|
||
def read_config():
|
||
if os.path.exists('config.ini'):
|
||
config_settings = RawConfigParser()
|
||
try:
|
||
config_settings.read('config.ini', encoding='UTF-8')
|
||
repository_url = config_settings.get("下载设置", "repository url")
|
||
host_url = config_settings.get("媒体服务器", "host url")
|
||
api_key = config_settings.get("媒体服务器", "api id")
|
||
max_retries = config_settings.get("下载设置", "max retry")
|
||
proxy = config_settings.get("下载设置", "proxy")
|
||
overwrite = True if config_settings.get("导入设置", "是否覆盖以前上传的头像？") == '是' else False
|
||
fixsize = True if config_settings.get("导入设置", "是否处理下载的头像？") == '是' else False
|
||
# 修正用户的URL
|
||
if not host_url.endswith('/'):
|
||
host_url += '/'
|
||
if not repository_url.endswith('/'):
|
||
repository_url += '/'
|
||
return (repository_url, host_url, api_key, overwrite, fixsize, max_retries, proxy)
|
||
except:
|
||
print(format_exc())
|
||
print('无法读取 config.ini')
|
||
os.system('pause')
|
||
else:
|
||
content = '''[媒体服务器]
|
||
# Emby/Jellyfin 服务器地址
|
||
host url = http://localhost:8096/
|
||
|
||
# Emby/Jellyfin API 密匙
|
||
api id =
|
||
|
||
[下载设置]
|
||
# 女友头像仓库地址，"默认"使用主分支：https://raw.githubusercontent.com/xinxin8816/gfriends/master/，网络不稳定可使用仓库备用镜像：https://gfriends.imfast.io/
|
||
repository url = 默认
|
||
|
||
# HTTP局部代理地址，格式为"IP:端口"，推荐开启全局代理而不是使用此处的局部代理
|
||
proxy = 不使用
|
||
|
||
# 最大重试次数，若网络连接不稳定，丢包率或延迟较高，可适当增加重试次数
|
||
max retry = 5
|
||
|
||
[导入设置]
|
||
# 处理以满足尺寸需求，能避免部分头像被拉伸但会牺牲一定质量
|
||
是否处理下载的头像？ = 是
|
||
|
||
是否覆盖以前上传的头像？ = 是'''
|
||
write_txt("config.ini", content)
|
||
print('没有找到配置文件 config.ini，已为阁下生成，请修改配置后重新运行程序\n')
|
||
os.system('pause')
|
||
sys.exit()
|
||
|
||
|
||
def read_persons(host_url, api_key):
|
||
print('读取 Emby / Jellyfin 演员...')
|
||
host_url_persons = host_url + 'emby/Persons?api_key=' + api_key # &PersonTypes=Actor
|
||
try:
|
||
rqs_emby = requests.get(url=host_url_persons)
|
||
except requests.exceptions.ConnectionError:
|
||
print('连接 Emby / Jellyfin 服务器失败，请检查：', host_url, '\n')
|
||
sys.exit()
|
||
except:
|
||
print(format_exc())
|
||
print('连接 Emby / Jellyfin 服务器未知错误', host_url, '\n')
|
||
sys.exit()
|
||
if rqs_emby.status_code == 401:
|
||
print('无权访问 Emby / Jellyfin 服务器，请检查 API 密匙\n')
|
||
sys.exit()
|
||
output = loads(rqs_emby.text)['Items']
|
||
print('读取 Emby / Jellyfin 演员完成\n')
|
||
return output
|
||
|
||
|
||
def write_txt(filename, content):
|
||
txt = open(filename, 'a', encoding="utf-8")
|
||
txt.write(content)
|
||
txt.close()
|
||
|
||
|
||
os.system('title Gfriends 一键导入工具')
|
||
print('读取配置文件 config.ini')
|
||
(repository_url, host_url, api_key, overwrite, fixsize, max_retries, proxy) = read_config()
|
||
os.system('cls')
|
||
|
||
num_suc = 0
|
||
num_fail = 0
|
||
num_exist = 0
|
||
try:
|
||
print('Gfriends 一键导入工具')
|
||
print('https://github.com/xinxin8816/gfriends')
|
||
# 局部代理
|
||
if proxy == '不使用':
|
||
print('推荐开启全局代理以加快下载速度\n')
|
||
else:
|
||
print('已配置 HTTP 局部代理：' + proxy + '，请确保其可用\n')
|
||
proxies = {
|
||
'http': 'http://' + proxy,
|
||
'https': 'https://' + proxy
|
||
}
|
||
os.system('pause')
|
||
# 持久会话
|
||
session = requests.Session()
|
||
session.mount('http://', requests.adapters.HTTPAdapter(max_retries=max_retries))
|
||
session.mount('https://', requests.adapters.HTTPAdapter(max_retries=max_retries))
|
||
list_persons = read_persons(host_url, api_key)
|
||
gfriends_map = get_gfriends_map(repository_url)
|
||
# 下载文件夹
|
||
folder_path = './Downloads/'
|
||
if os.path.exists(folder_path) == False:
|
||
os.makedirs(folder_path)
|
||
with alive_bar(len(list_persons), theme='ascii', enrich_print=False) as bar:
|
||
for dic_each_actor in list_persons:
|
||
actor_name = dic_each_actor['Name']
|
||
bar()
|
||
if get_gfriends_link(actor_name) == None:
|
||
print('>> 未收录：', actor_name)
|
||
write_txt("未收录的演员清单.txt", actor_name + '\n')
|
||
num_fail += 1
|
||
else:
|
||
write_txt("已匹配的演员清单.txt", actor_name + '\n')
|
||
if dic_each_actor['ImageTags']:
|
||
num_exist += 1
|
||
if not overwrite:
|
||
print('>> 跳过：', actor_name)
|
||
continue
|
||
print('>> 下载并导入：', get_gfriends_link(actor_name))
|
||
try:
|
||
if proxy == '不使用':
|
||
pic = session.get(get_gfriends_link(actor_name))
|
||
else:
|
||
pic = session.get(get_gfriends_link(actor_name), proxies=proxies)
|
||
except (KeyboardInterrupt):
|
||
sys.exit()
|
||
except:
|
||
with bar.pause():
|
||
print(format_exc())
|
||
print('网络连接异常且重试 ' + max_retries + ' 次失败')
|
||
print('请尝试开启全局代理或配置 HTTP 局部代理；若已开启代理，请检查其可用性')
|
||
print('继续运行则跳过下载此头像：' + actor_name)
|
||
os.system('pause')
|
||
continue
|
||
with open("Downloads/" + actor_name + ".jpg", "wb") as code:
|
||
code.write(pic.content)
|
||
if fixsize:
|
||
fix_size("Downloads/" + actor_name + ".jpg")
|
||
pic = open("Downloads/" + actor_name + ".jpg", 'rb')
|
||
b6_pic = b64encode(pic.read())
|
||
pic.close()
|
||
url_post_img = host_url + 'emby/Items/' + dic_each_actor['Id'] + '/Images/Primary?api_key=' + api_key
|
||
session.post(url=url_post_img, data=b6_pic, headers={"Content-Type": 'image/jpeg', })
|
||
num_suc += 1
|
||
print('\nEmby / Jellyfin 拥有演员', len(list_persons), '人，当前已有头像', num_exist, '人')
|
||
print('本次成功导入', num_suc, '人')
|
||
print('仓库未收录', num_fail, '人\n')
|
||
os.system('pause')
|
||
except (KeyboardInterrupt, SystemExit):
|
||
print('强制停止或已知致命错误！')
|
||
os.system('pause')
|
||
except:
|
||
print(format_exc())
|
||
print('未知致命错误！')
|
||
os.system('pause')
|