Gfriends Inputer.py

# -*- coding:utf-8 -*-
# Gfriends Inputer / 女友头像仓库导入工具
# Licensed under the MIT license.
# Designed by xinxin8816, many thanks for junerain123, ddd354, moyy996.
version = 'v2.81'

import requests, os, io, sys, time, re, threading, argparse
from alive_progress import alive_bar
from configparser import RawConfigParser
from traceback import format_exc
from hashlib import md5
from base64 import b64encode
from json import loads
from PIL import Image,ImageFilter
from aip import AipBodyAnalysis

def fix_size(type,path):
	try:
		pic = Image.open(path)
		if pic.mode != "RGB": pic = pic.convert('RGB') # 有些图片有P通道，base64编码后会出问题
		(wf,hf) = pic.size
		if not 2/3-0.02 <= wf/hf <= 2/3+0.02: # 仅处理会过度拉伸的图片
			if type == '1':
				fixed_pic = pic.resize((int(wf),int(3/2*wf))) # 拉伸图片
				fixed_pic = fixed_pic.filter(ImageFilter.GaussianBlur(radius=50)) # 高斯平滑滤镜
				fixed_pic.paste(pic,(0,int((3/2*wf-hf)/2))) # 粘贴原图
				fixed_pic.save(path, quality=95)
			elif type == '2':
				fixed_pic = pic.crop((int(wf/2-1/3*hf),0,int(wf/2+1/3*hf),int(hf))) # 像素中线向两边扩展
				fixed_pic.save(path, quality=95)
			elif type == '3':
				try:
					with open(path, 'rb') as fp:
						x_nose = int(BD_AI_client.bodyAnalysis(fp.read())["person_info"][0]['body_parts']['nose']['x']) # 返回鼻子横坐标
					if BD_VIP == '否':
						time.sleep(0.4) # 免费用户QPS=2
					else:
						time.sleep( 1/int(1.1*BD_VIP) )
					if x_nose + 1/3*hf > wf: # 判断鼻子在图整体的位置
						x_left = wf-2/3*hf # 以右为边
					elif x_nose - 1/3*hf < 0:
						x_left = 0 # 以左为边
					else:
						x_left = x_nose-1/3*hf # 以鼻子为中线向两边扩展
					fixed_pic = pic.crop((x_left,0,x_left+2/3*hf,hf))
					fixed_pic.save(path, quality=95)
				except (KeyboardInterrupt):
					sys.exit()
				except:
					print('!! '+ path +' AI 分析失败，跳过 AI 直接裁剪')
					fix_size('2',path)
			else:
				print('× 头像处理功能配置错误，没有此选项：' + str(type))
				sys.exit()
		return True
	except (KeyboardInterrupt, SystemExit):
		sys.exit()
	except:
		if 'pic' in vars(): del pic # 如果图片已打开，则关闭
		print('!! '+ path +' 可能已损坏，跳过。')
		failed_dir = re.sub(r'(.*/)(.*)', r'\1Failed/', path)
		failed_path = re.sub(r'(.*/)(.*)', r'\1Failed/\2', path)
		if not os.path.exists(failed_dir): os.makedirs(failed_dir)
		if os.path.exists(failed_path): os.remove(failed_path)
		os.rename(path,failed_path)
		return False

def get_gfriends_map(repository_url):
	rewriteable_word('>> 连接 Gfriends 女友头像仓库...')
	if repository_url == '默认/': repository_url = 'https://raw.githubusercontent.com/xinxin8816/gfriends/master/'
	github_template = repository_url+'{}/{}/{}'
	if aifix:
		request_url = repository_url+'Filetree.json'
	else:
		request_url = repository_url+'Filetree-NonFix.json'
	try:
		if proxy == '不使用':
			response = session.get(request_url)
		else:
			response = session.get(request_url, proxies = proxies)
		# 修复部分服务端返回 header 未指明编码使后续解析错误
		response.encoding = 'utf-8' 
	except:
		if debug: print(format_exc())
		print('× 网络连接异常且重试 ' + str(max_retries) + ' 次失败')
		print('× 请尝试开启全局代理或配置 HTTP 局部代理；若已开启代理，请检查其可用性')
		sys.exit()
	if response.status_code != 200:
		print('× 女友仓库返回了一个错误：' + str(response.status_code))
		sys.exit()
	if aifix:
		map_json = loads(response.text)
	else:
		map_json = loads(response.text.replace('AI-Fix-',''))
	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():
				secondstr = re.sub(".*-", "", second)
				if not secondstr in Black_List:
					output[k[:-4]] = github_template.format(first, second, v)
	print('√ 连接 Gfriends 女友头像仓库成功')
	print('   库存头像：' + str(response.text.count('\n')) + '枚\n')
	return output

def asyncc(f):
    def wrapper(*args, **kwargs):
        thr = threading.Thread(target=f, args=args, kwargs=kwargs)
        thr.start()
    return wrapper
	
@asyncc
def download_avatar(url,actor_name,proc_md5):
	if proxy == '不使用':
		response = session.get(url)
	else:
		response = session.get(url, proxies = proxies)
	pic_path = download_path+actor_name+".jpg"
	try:
		Image.open(io.BytesIO(response.content)).verify() # 校验下载的图片是否损坏
		with open(pic_path,"wb") as code:	
			code.write(response.content)
		proc_log.write(proc_md5+'\n')
	except:
		print('!! '+ pic_path +' 校验失败，可能是仓库收录的头像已损坏。')
	
@asyncc
def input_avatar(url,data):
	if proxy == '不使用' or '127.0.0.1' in host_url or 'localhost' in host_url or '192.168' in host_url:
		session.post(url, data=data, headers={"Content-Type": 'image/jpeg',"User-Agent": 'Gfriends_Inputer/'+version.replace('v','')})
	else:
		session.post(url, data=data, headers={"Content-Type": 'image/jpeg',"User-Agent": 'Gfriends_Inputer/'+version.replace('v','')}, proxies = proxies)

@asyncc
def del_avatar(url_post_img):
	if proxy == '不使用' or '127.0.0.1' in host_url or 'localhost' in host_url or '192.168' in host_url:
		session.delete(url=url_post_img)
	else:
		session.delete(url=url_post_img, proxies = proxies)
		
def get_gfriends_link(name):
	if name in gfriends_map:
		output = gfriends_map[name]
		return output
	else:
		return None

def argparse_function(ver: str) -> [str, str, bool]:
    parser = argparse.ArgumentParser()
    parser.add_argument("-c", "--config", default='config.ini', nargs='?', help="The config file Path.")
    parser.add_argument("-q", "--quiet", dest='quietflag', action="store_true", help="Assume Yes on all queries and Print logs to file.")
    parser.add_argument("-v", "--version", action="version", version=ver)
    args = parser.parse_args()
    return args.config, args.quietflag

def read_config(config_file):
	global config_settings
	rewriteable_word('>> 读取配置...')
	if os.path.exists(config_file):
		config_settings = RawConfigParser()
		try:
			config_settings.read('config.ini', encoding = 'UTF-8-SIG') # UTF-8-SIG 适配 Windows 记事本
			repository_url = config_settings.get("下载设置", "Repository_Url")
			host_url = config_settings.get("媒体服务器", "Host_Url")
			api_key = config_settings.get("媒体服务器", "Host_API")
			max_download_connect = config_settings.getint("下载设置", "MAX_DL")
			max_retries = config_settings.getint("下载设置", "MAX_Retry")
			proxy = config_settings.get("下载设置", "Proxy")
			download_path = config_settings.get("下载设置", "Download_Path")
			Black_List = config_settings.get("下载设置", "Black_List")
			max_upload_connect = config_settings.getint("导入设置", "MAX_UL")
			local_path = config_settings.get("导入设置", "Local_Path")
			BD_App_ID = config_settings.get("导入设置", "BD_App_ID")
			BD_API_Key = config_settings.get("导入设置", "BD_API_Key")
			BD_Secret_Key = config_settings.get("导入设置", "BD_Secret_Key")
			BD_VIP = config_settings.get("导入设置", "BD_VIP")
			aifix = True if config_settings.get("下载设置", "AI_Fix") == '是' else False
			overwrite = True if config_settings.get("导入设置", "OverWrite") == '是' else False
			debug = True if config_settings.get("调试功能", "DeBug") == '是' else False
			deleteall = True if config_settings.get("调试功能", "DEL_ALL") == '是' else False
			fixsize = config_settings.get("导入设置", "Size_Fix")
			# 修正用户的URL
			if not host_url.endswith('/'): host_url += '/'
			if not repository_url.endswith('/'): repository_url += '/'
			# 创建文件夹
			if not os.path.exists(download_path):
				os.makedirs(download_path)
				write_txt(download_path+"/README.txt",'本目录自动生成，用于存放从仓库下载和处理过的头像。')
			if not os.path.exists(local_path):
				os.makedirs(local_path)
				write_txt(local_path+"/README.txt",'本目录自动生成，您可以存放自己收集的头像，这些头像将被优先导入服务器。\n\n请自行备份您收集头像的副本，根据个人配置不同，该目录文件可能会被程序修改。\n\n仅支持JPG格式，且请勿再创建子目录。\n\n如果您收集的头像位于子目录，可通过 Move To Here.bat（Only for Windows） 工具将其全部提取到根目录。')
				write_txt(local_path+"/Move To Here.bat",'@echo off\necho This tool will help you move all files which in the subdirectory to this root directory\npause\nfor /f "delims=" %%a in ("dir /a-d /b /s ") do (\nmove "%%~a" ./ 2>nul\n)\n')
			if fixsize == '3':
				BD_AI_client = AipBodyAnalysis(BD_App_ID, BD_API_Key, BD_Secret_Key)
			else:
				BD_AI_client = None
			Black_List = Black_List.split('、')
			return (repository_url,host_url,api_key,overwrite,fixsize,max_retries,proxy,aifix,debug,deleteall,download_path,local_path,max_download_connect,max_upload_connect,BD_AI_client,BD_VIP,Black_List)
		except:
			print(format_exc())
			print('× 无法读取 config.ini。如果这是旧版本的配置文件，请删除后重试。\n')
			if WINOS: print('按任意键退出程序...'); os.system('pause>nul')
			sys.exit()
	else:
		content='''[媒体服务器]
# Emby / Jellyfin 服务器地址
Host_Url = http://localhost:8096/
	
# Emby / Jellyfin API 密钥
Host_API = 

[下载设置]
# 下载文件夹
Download_Path = ./Downloads/

# 下载线程数
# 若网络不稳定、丢包率或延迟较高，可适当减小下载线程数
MAX_DL = 5

# 下载失败重试数
# 若网络不稳定、丢包率或延迟较高，可适当增加失败重试数
MAX_Retry = 3

# 女友头像仓库源
# "默认"使用官方主仓库（已禁止浏览器访问）：https://raw.githubusercontent.com/xinxin8816/gfriends/master/
# 官方备用镜像（镜像下载线程数不允许大于5）：https://gfriends.imfast.io/，受服务提供商业务调整影响，镜像仓库将于 2020/12/31 关闭。
Repository_Url = 默认

# AI 优化（仅支持官方仓库）
# 在不可避免下载低质量头像时，自动挑选经 AI 算法放大优化的副本，质量更高但更占空间
AI_Fix = 是

# HTTP / Socks 局部代理
# 推荐开启全局代理而不是使用此局部代理
# HTTP 代理格式为 http://IP:端口 , 如 http://localhost:8088
# Socks 代理格式为 socks+协议版本://IP:端口 , 如 socks5://localhost:8087
Proxy = 不使用

# 厂牌黑名单（仅支持官方仓库）
# 请访问女友仓库 Content 文件夹确认最新的已收录的厂牌列表，下述收录列表更新有延迟：ラグジュTV、DMM(骑)、DMM(步)、痴女天堂、溜池ゴロー、无垢、WANZ、KMP、KiraKira、Ideapocket、DAS、BeFree、えむっ娘ラボ、OPPAI、Honnaka、桃太郎、Prestige、Madonna、Fitch、Attackers、未満、S1、Moodyz、Warashi、Premium、body、Kawaii、GRAPHIS、MUTEKI、Lovepop、Honey、FALENO、AVDBS、Derekhsu、Javrave、Nanairo
# 女友仓库默认提供质量优先的头像。但是这其中，某些厂牌官方给演员的头像可能真的很难看，如果你不想从仓库下载到某些厂牌官网的头像，请填写其厂牌名。多个厂牌请使用中文顿号（、）分隔。
Black_List = 无

[导入设置]
# 本地头像文件夹
# 将第三方头像包或自己收集的头像移动至该目录，可优先于仓库导入服务器。仅支持非子目录下的 jpg 格式。
Local_Path = ./Avatar/

# 覆盖已有头像
OverWrite = 是

# 导入线程数
# 导入至本地或内网服务器时，网络稳定可适当增大导入线程数（推荐：20-100）
# 导入至远程服务器时，可适当减小导入线程数（推荐：5-20）
MAX_UL = 20

# 头像尺寸优化
# 避免媒体服务器拉伸比例不符合 2:3 的头像
# 0 - 不处理直接导入
# 1 - 高斯平滑处理（填充毛玻璃样式）
# 2 - 直接裁剪处理（可能会裁剪到演员面部）
# 3 - AI检测并裁剪处理（需配置百度人体定位 AI，免费用户QPS=2，处理速度慢）
Size_Fix = 2

# 百度人体定位 AI
# 具体使用说明请参阅仓库项目 README
# 免费个人用户 QPS=2 处理速度慢。付费个人用户和企业用户请修改 BD_VIP 为您购买的 QPS 额度值，免费个人用户修改后会报错。
BD_VIP = 否
BD_App_ID = 
BD_API_Key = 
BD_Secret_Key = 

[调试功能]
# 删除所有头像
DEL_ALL = 否

# 输出详尽错误
DeBug = 否'''
		write_txt("config.ini", content)
		print('× 没有找到 config.ini。已为阁下生成，请修改配置后重新运行程序。\n')
		if WINOS: print('按任意键退出程序...');	os.system('pause>nul')
		sys.exit()

def read_persons(host_url,api_key,emby_flag):
	rewriteable_word('>> 连接 Emby / Jellyfin 服务器...')
	if emby_flag:
		host_url_persons = host_url + 'emby/Persons?api_key=' + api_key	 # &PersonTypes=Actor
	else:
		host_url_persons = host_url + 'jellyfin/Persons?api_key=' + api_key	 # &PersonTypes=Actor
	try:
		if proxy == '不使用' or '127.0.0.1' in host_url or 'localhost' in host_url or '192.168' in host_url:
			rqs_emby = session.get(url=host_url_persons, headers={"User-Agent": 'Gfriends_Inputer/'+version.replace('v','')})
		else:
			rqs_emby = session.get(url=host_url_persons, headers={"User-Agent": 'Gfriends_Inputer/'+version.replace('v','')}, proxies = proxies)	
	except session.exceptions.ConnectionError:
		print('× 连接 Emby / Jellyfin 服务器失败，请检查：', host_url, '\n')
		sys.exit()
	except:
		if debug: print(format_exc())
		print('× 连接 Emby / Jellyfin 服务器未知错误：', host_url, '\n')
		sys.exit()
	if rqs_emby.status_code == 401:
		print('× 无权访问 Emby / Jellyfin 服务器，请检查 API 密匙\n')
		sys.exit()
	if rqs_emby.status_code == 404 and emby_flag:
		rewriteable_word('>> 可能是新版 Jellyfin，尝试重新连接...')
		return read_persons(host_url,api_key,False)
	elif rqs_emby.status_code == 404 and not emby_flag:
		print('× 尝试读取 Emby / Jellyfin 演员列表但是未找到，可能是未适配的版本：', host_url, '\n')
		sys.exit()
	if rqs_emby.status_code != 200:
		print('× 连接 Emby / Jellyfin 服务器成功，但是服务器内部错误：' + str(response.status_code))
		sys.exit()
	if 'json' not in rqs_emby.headers['Content-Type']:
		print('× 连接 Emby / Jellyfin 服务器成功，但是服务器的演员列表不能识别：' + rqs_emby.headers['Content-Type'])
		sys.exit()
	output = sorted(loads(rqs_emby.text)['Items'], key = lambda x:x['Name']) # 按姓名排序
	print('√ 连接 Emby / Jellyfin 服务器成功')
	print('   演职人员：' + str(len(output)) + '人\n')
	return (output,emby_flag)

def write_txt(filename,content):
	txt = open(filename, 'a', encoding="utf-8")
	txt.write(content)
	txt.close()
		
def rewriteable_word(word):
	for t in ['', word]: sys.stdout.write('\033[K' + t + '\r')
		
def del_all():
	print('【调试模式】删除所有头像\n')
	(list_persons,emby_flag) = read_persons(host_url,api_key,True)
	rewriteable_word('按任意键开始...'); os.system('pause>nul') if WINOS else input('按任意键开始...')
	with alive_bar(len(list_persons), theme = 'ascii', enrich_print = False) as bar:
		for dic_each_actor in list_persons:
			bar(dic_each_actor['Name'])
			if dic_each_actor['ImageTags']:
				if emby_flag:
					url_post_img = host_url + 'emby/Items/' + dic_each_actor['Id'] + '/Images/Primary?api_key=' + api_key
				else:
					url_post_img = host_url + 'jellyfin/Items/' + dic_each_actor['Id'] + '/Images/Primary?api_key=' + api_key
				del_avatar(url_post_img)
				while True:
					if not threading.activeCount() > max_upload_connect + 1: break
	rewriteable_word('>> 即将完成')
	for thr_status in threading.enumerate():
		try:
			thr_status.join()
		except RuntimeError:
			continue
	print('√ 删除完成')
	if WINOS: print('按任意键退出程序...'); os.system('pause>nul')
	sys.exit()

def check_update():
	rewriteable_word('>> 检查更新...')
	try:
		if proxy == '不使用':
			response = session.get('https://api.github.com/repos/xinxin8816/gfriends/releases')
		else:
			response = session.get('https://api.github.com/repos/xinxin8816/gfriends/releases', proxies = proxies)
		response.encoding = 'utf-8'
		if response.status_code != 200:
			print('× 检查更新失败！返回了一个错误： {}\n'.format(response.status_code))
			rewriteable_word('按任意键跳过...'); os.system('pause>nul') if WINOS else input('按任意键跳过...')
		if version.replace('v','') < loads(response.text)[0]['tag_name'].replace('v',''):
			print(loads(response.text)[0]['tag_name']+' 新版本发布啦！\n')
			print(re.search('What\'s New?.*?(?=\r\n<details>)',loads(response.text)[0]['body'],flags=re.S).group(0).replace('*',''))
			print('请通过如下链接下载更新：\nhttps://git.io/JL0tk\n')		
			rewriteable_word('按任意键跳过更新...'); os.system('pause>nul') if WINOS else input('按任意键跳过更新...')
			print('即将跳过更新。不推荐跳过更新，如遇问题请及时更新。')
			time.sleep(5)
	except:
		if debug: print(format_exc())
		print('× 检查更新失败！\n')
		rewriteable_word('按任意键跳过...'); os.system('pause>nul') if WINOS else input('按任意键跳过更新...')
	if WINOS and not quiet_flag: os.system('cls')

WINOS = True if sys.platform.startswith('win') else False
if WINOS: os.system('title Gfriends Inputer '+version)
(config_file, quiet_flag) = argparse_function(version)
if quiet_flag: sys.stdout = open("quiet.log", "w", buffering = 1)
(repository_url,host_url,api_key,overwrite,fixsize,max_retries,proxy,aifix,debug,deleteall,download_path,local_path,max_download_connect,max_upload_connect,BD_AI_client,BD_VIP,Black_List) = read_config(config_file)

# 持久会话
session = requests.Session()
session.mount('http://', requests.adapters.HTTPAdapter(max_retries=max_retries))
session.mount('https://', requests.adapters.HTTPAdapter(max_retries=max_retries))

# 局部代理
if not proxy == '不使用': proxies={'http': proxy,'https': proxy}

# 检查更新
check_update()
if deleteall: del_all()


# 变量初始化
num_suc = num_fail = num_skip = num_exist = 0
pic_path_dict = {}
actor_dict = {}
link_dict = {}

print('Gfriends Inputer '+version)
print('https://git.io/gfriends\n')
if not proxy == '不使用': print('已配置局部代理 ' + proxy + '，请确保其可用\n')

if not quiet_flag:
	rewriteable_word('按任意键开始...'); os.system('pause>nul') if WINOS else input('按任意键开始...')

try:
	(list_persons,emby_flag) = read_persons(host_url,api_key,True)
	gfriends_map = get_gfriends_map(repository_url)
	if os.path.exists('未收录的演员清单.txt'): os.remove('未收录的演员清单.txt')
	write_txt("未收录的演员清单.txt",'【未收录的演员清单】\n（该清单仅供参考，下面可能还有导演、编剧、赞助商等其他人的名字，但是女友头像仓库只会收录日本女友。）\n\n')
	if os.path.exists('已匹配的演员清单.txt'): os.remove('已匹配的演员清单.txt')
	write_txt("已匹配的演员清单.txt",'【已匹配的演员清单】\n（该清单仅供参考，记录的名字都是女友头像仓库已经收录头像的。根据个人配置，可能会下载导入，也可能会跳过）\n\n')
	
	rewriteable_word('>> 引擎初始化...')
	proc_flag = False
	md5_persons = md5(str(list_persons).encode('UTF-8')).hexdigest()[14:-14]
	md5_config = md5(open('config.ini','rb').read()).hexdigest()[14:-14] # md5计算只支持字节流
	if os.path.exists('proc.tmp'): # 有中断记录，则逐行读取记录
		with open('proc.tmp', 'r', encoding='UTF-8') as file:
			proc_list = file.read().split('\n')
		if md5_persons in proc_list and md5_config in proc_list: # 上次中断后，演员列表和配置文件没有变化才尝试续传
			proc_flag = True
	proc_log = open('proc.tmp', 'w', encoding = "UTF-8", buffering = 1)
	proc_log.write('## Gfriends Inputer 断点记录 ##\n' + md5_persons + '\n' + md5_config + '\n')
	for dic_each_actor in list_persons:
		actor_name = dic_each_actor['Name']
		actor_id = dic_each_actor['Id']
		if dic_each_actor['ImageTags']: 
			num_exist += 1
			if not overwrite:
				write_txt("已匹配的演员清单.txt",'跳过：' + actor_name + '\n')
				num_skip += 1
				continue
		if not os.path.exists(local_path+actor_name+".jpg"):
			pic_link = get_gfriends_link(actor_name)
			if pic_link == None:
				actor_name = re.sub(r'（.*）','',actor_name)
				actor_name = re.sub(r'\(.*\)','',actor_name)
				pic_link = get_gfriends_link(actor_name)
				if pic_link == None:
					write_txt("未收录的演员清单.txt",'未找到：' + actor_name + '\n')
					num_fail += 1
					continue
			write_txt("已匹配的演员清单.txt",'下载：' + actor_name + '\n')
			link_dict[actor_name] = pic_link
		actor_dict[actor_name] = actor_id
	print('√ 引擎初始化成功，尝试从上次中断位置继续') if proc_flag else print('√ 引擎初始化成功')
	
	print('\n>> 下载头像...')
	if quiet_flag:
		for actor_name,link in link_dict.items():
			try:
				proc_md5 = md5((actor_name+'+1').encode('UTF-8')).hexdigest()[14:-14]
				if (proc_flag and not proc_md5 in proc_list) or not proc_flag:
					download_avatar(link,actor_name,proc_md5) # 记录下载完成的操作放到子线程中，以防没下完中断的断点没记录到
				else:
					proc_log.write(proc_md5+'\n')
				while True:
					if threading.activeCount() > max_download_connect + 1:
						time.sleep(0.01)
					else:
						break
			except (KeyboardInterrupt):
				sys.exit()
			except:
				if debug: print(format_exc())
				print('× 网络连接异常，跳过下载：'+ str(actor_name)+'\n')
				continue
	else:
		with alive_bar(len(link_dict), theme = 'ascii', enrich_print = False) as bar:
			for actor_name,link in link_dict.items():
				try:
					bar(re.sub(r'（.*）','',actor_name)) if '（' in actor_name else bar(actor_name)
					proc_md5 = md5((actor_name+'+1').encode('UTF-8')).hexdigest()[14:-14]
					if (proc_flag and not proc_md5 in proc_list) or not proc_flag:
						download_avatar(link,actor_name,proc_md5) # 记录下载完成的操作放到子线程中，以防没下完中断的断点没记录到
					else:
						proc_log.write(proc_md5+'\n')
					while True:
						if threading.activeCount() > max_download_connect + 1:
							time.sleep(0.01)
						else:
							break
				except (KeyboardInterrupt):
					sys.exit()
				except:
					with bar.pause():
						if debug: print(format_exc())
						print('× 网络连接异常且重试 ' + str(max_retries) + ' 次失败')
						print('× 请尝试开启全局代理或配置 HTTP 局部代理；若已开启代理，请检查其可用性')
						print('× 按任意键继续运行则跳过下载这些头像：'+ str(actor_name)+'\n')
						os.system('pause>nul') if WINOS else input()
					continue
		rewriteable_word('>> 即将完成')
	for thr_status in threading.enumerate(): # 等待子线程运行结束
		try:
			thr_status.join()
		except RuntimeError:
			continue
	print('√ 下载完成')
	
	# 构建路径映射
	for filename in os.listdir(download_path):
		if '.jpg' in filename and filename.replace('.jpg','') in actor_dict:
			pic_path_dict[filename] = download_path + filename
	for filename in os.listdir(local_path):
		if '.jpg' in filename and filename.replace('.jpg','') in actor_dict:
			pic_path_dict[filename] = local_path + filename
	
	if fixsize != '0':
		print('\n>> 尺寸优化...')
		if quiet_flag:
			for filename,pic_path in pic_path_dict.items():
				proc_md5 = md5((filename+'+2').encode('UTF-8')).hexdigest()[14:-14]
				if (proc_flag and not proc_md5 in proc_list) or not proc_flag:
					result = fix_size(fixsize,pic_path)
					if not result: pic_path_dict.pop(filename)
				proc_log.write(proc_md5+'\n')
		else:
			with alive_bar(len(pic_path_dict), theme = 'ascii', enrich_print = False) as bar:
				for filename,pic_path in pic_path_dict.items():
					bar(re.sub(r'（.*）','',filename).replace('.jpg','')) if '（' in filename else bar(filename.replace('.jpg',''))
					proc_md5 = md5((filename+'+2').encode('UTF-8')).hexdigest()[14:-14]
					if (proc_flag and not proc_md5 in proc_list) or not proc_flag:
						result = fix_size(fixsize,pic_path)
						if not result: pic_path_dict.pop(filename)
					proc_log.write(proc_md5+'\n')
		print('√ 优化完成')
	
	print('\n>> 导入头像...')	
	if quiet_flag:
		for filename,pic_path in pic_path_dict.items():
			proc_md5 = md5((filename+'+3').encode('UTF-8')).hexdigest()[14:-14]
			if (proc_flag and not proc_md5 in proc_list) or not proc_flag:
				with open(pic_path, 'rb') as pic_bit:
					b6_pic = b64encode(pic_bit.read())
				if emby_flag:
					url_post_img = host_url + 'emby/Items/' + actor_dict[filename.replace('.jpg','')] + '/Images/Primary?api_key=' + api_key
				else:
					url_post_img = host_url + 'jellyfin/Items/' + actor_dict[filename.replace('.jpg','')] + '/Images/Primary?api_key=' + api_key
				input_avatar(url_post_img,b6_pic)
			proc_log.write(proc_md5+'\n')
			while True:
				if threading.activeCount() > max_upload_connect + 1:
					time.sleep(0.01)
				else:
					break
			num_suc += 1
	else:
		with alive_bar(len(pic_path_dict), theme = 'ascii', enrich_print = False) as bar:
			for filename,pic_path in pic_path_dict.items():
				bar(re.sub(r'（.*）','',filename).replace('.jpg','')) if '（' in filename else bar(filename.replace('.jpg',''))
				proc_md5 = md5((filename+'+3').encode('UTF-8')).hexdigest()[14:-14]
				if (proc_flag and not proc_md5 in proc_list) or not proc_flag:
					with open(pic_path, 'rb') as pic_bit:
						b6_pic = b64encode(pic_bit.read())
					if emby_flag:
						url_post_img = host_url + 'emby/Items/' + actor_dict[filename.replace('.jpg','')] + '/Images/Primary?api_key=' + api_key
					else:
						url_post_img = host_url + 'jellyfin/Items/' + actor_dict[filename.replace('.jpg','')] + '/Images/Primary?api_key=' + api_key
					input_avatar(url_post_img,b6_pic)
				proc_log.write(proc_md5+'\n')
				while True:
					if threading.activeCount() > max_upload_connect + 1:
						time.sleep(0.01)
					else:
						break
				num_suc += 1	
		rewriteable_word('>> 即将完成')
	for thr_status in threading.enumerate(): # 等待子线程运行结束
		try:
			thr_status.join()
		except RuntimeError:
			continue
	proc_log.close()
	os.remove('proc.tmp')
	print('√ 导入完成')
	print('\nEmby / Jellyfin 演职人员共 ' + str(len(list_persons)) + ' 人，其中 ' + str(num_exist) + ' 人之前已有头像')
	print('本次导入 ' + str(num_suc) + ' 人，还有 ' + str(num_fail) + ' 人没有头像\n')
	if not overwrite: print('-- 未开启覆盖已有头像，所以跳过了一些演员，详见记录清单')
except (KeyboardInterrupt, SystemExit):
	print('× 用户强制停止或已知错误。')
except:
	if debug: print(format_exc())
	print('× 未知错误，在配置文件中开启 Debug 可获得错误详情。')
if WINOS and not quiet_flag: print('按任意键退出程序...'); os.system('pause>nul')