# -*- 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')