mirror of
https://github.com/KimigaiiWuyi/GenshinUID.git
synced 2025-06-01 13:09:56 +08:00
340 lines
14 KiB
Python
340 lines
14 KiB
Python
import time
|
||
from base64 import b64encode
|
||
from io import BytesIO
|
||
|
||
import urllib
|
||
import math
|
||
|
||
from PIL import Image, ImageDraw, ImageFont, ImageFilter
|
||
|
||
from .getData import GetInfo,GetCharacter
|
||
|
||
import os
|
||
import json
|
||
import random
|
||
|
||
FILE_PATH = os.path.dirname(__file__)
|
||
FILE2_PATH = os.path.join(FILE_PATH,'mys')
|
||
CHAR_PATH = os.path.join(FILE2_PATH,'chars')
|
||
CHAR_WEAPON_PATH = os.path.join(FILE2_PATH,'char_weapon')
|
||
TEXT_PATH = os.path.join(FILE2_PATH,'texture2d')
|
||
WEAPON_PATH = os.path.join(FILE2_PATH,'weapon')
|
||
|
||
def ys_font(size):
|
||
return ImageFont.truetype(os.path.join(FILE2_PATH,"yuanshen.ttf"), size=size)
|
||
|
||
async def draw_pic(uid,nickname,image = None):
|
||
is_edit = False
|
||
if image != None:
|
||
image_file= image.group(1)
|
||
image_data = image.group(2)
|
||
urllib.request.urlretrieve(f'{image_data}', os.path.join(TEXT_PATH,nickname + '.png'))
|
||
is_edit = True
|
||
|
||
raw_data = await GetInfo(uid)
|
||
|
||
if (raw_data["retcode"] != 0):
|
||
if (raw_data["retcode"] == 10001):
|
||
return ("Cookie错误/过期，请重置Cookie")
|
||
if (raw_data["retcode"] == 10101):
|
||
return ("当前cookies已达到30人上限！")
|
||
if (raw_data["retcode"] == 10102):
|
||
return ("当前查询id已经设置了隐私，无法查询！")
|
||
return (
|
||
"Api报错，返回内容为：\r\n"
|
||
+ str(raw_data) + "\r\n出现这种情况可能的UID输入错误 or 不存在"
|
||
)
|
||
else:
|
||
pass
|
||
|
||
bg_list = ['bg_img_1.png','bg_img_2.png','bg_img_3.png']
|
||
bg_path = os.path.join(TEXT_PATH,"bg.png")
|
||
bg2_path = os.path.join(TEXT_PATH,random.choice(bg_list))
|
||
fg_path = os.path.join(TEXT_PATH,"fg_img_1.png")
|
||
mask_path = os.path.join(TEXT_PATH,"mask_img.png")
|
||
|
||
if is_edit == True:
|
||
bg_path_edit = os.path.join(TEXT_PATH,f"{nickname}.png")
|
||
edit_bg = Image.open(bg_path_edit)
|
||
w, h = edit_bg.size
|
||
scale_f = w / h
|
||
scale = '%.4f' % scale_f
|
||
new_w = math.ceil(1200*float(scale))
|
||
new_h = math.ceil(800/float(scale))
|
||
if w > h:
|
||
bg_img2 = edit_bg.resize((new_w, 1200),Image.ANTIALIAS)
|
||
elif float(scale) > 0.66:
|
||
bg_img2 = edit_bg.resize((new_w, 1200),Image.ANTIALIAS)
|
||
else:
|
||
bg_img2 = edit_bg.resize((800, new_h),Image.ANTIALIAS)
|
||
bg_img = bg_img2.crop((0, 0, 800, 1200))
|
||
else:
|
||
bg_img = Image.open(bg2_path)
|
||
|
||
raw_data = raw_data['data']
|
||
|
||
fg_img = Image.open(fg_path)
|
||
mask_img = Image.open(mask_path)
|
||
|
||
#img = Image.open(bg_path)
|
||
area = (37, 268, 764, 1154)
|
||
img_bb = bg_img.crop(area)
|
||
img_blur = img_bb.filter(ImageFilter.GaussianBlur(4))
|
||
|
||
bg_img.paste(img_blur, (37,268),mask_img)
|
||
|
||
bg_img.paste(fg_img,(0,0),fg_img)
|
||
text_draw = ImageDraw.Draw(bg_img)
|
||
|
||
text_draw.text((192.6,128.3), f"{nickname}", (217,217,217), ys_font(32))
|
||
text_draw.text((210.6, 165.3), 'UID ' + f"{uid}", (217,217,217), ys_font(14))
|
||
|
||
char_data = raw_data["avatars"]
|
||
|
||
index_char = ["神里绫华","琴","迪卢克","温迪","魈","可莉","钟离","达达利亚","七七","甘雨","阿贝多","莫娜","刻晴","胡桃","枫原万叶","优菈","丽莎","芭芭拉","凯亚","雷泽","安柏","香菱","北斗","行秋","凝光","菲谢尔","班尼特","诺艾尔","重云","迪奥娜","砂糖","辛焱","罗莎莉亚","烟绯","旅行者"]
|
||
|
||
five_star_char = ["神里绫华","琴","迪卢克","温迪","魈","可莉","钟离","达达利亚","七七","甘雨","阿贝多","莫娜","刻晴","胡桃","枫原万叶","优菈","宵宫"]
|
||
four_star_char = ["丽莎","芭芭拉","凯亚","雷泽","安柏","香菱","北斗","行秋","凝光","菲谢尔","班尼特","诺艾尔","重云","迪奥娜","砂糖","辛焱","罗莎莉亚","烟绯","早柚"]
|
||
|
||
for k in char_data:
|
||
if k['name'] in five_star_char:
|
||
k['star'] = 5
|
||
if k['name'] in four_star_char:
|
||
k['star'] = 4
|
||
if k['name'] == '旅行者':
|
||
k['star'] = 3
|
||
|
||
#char_data.sort(key=lambda x: index_char.index(x['name']))
|
||
|
||
char_data.sort(key=lambda x: (-x['star'],-x['level'],-x['fetter']))
|
||
|
||
text_draw.text((590, 94.8),str(raw_data['stats']['active_day_number']), (21, 21, 21), ys_font(26))
|
||
text_draw.text((590, 139.3),str(raw_data['stats']['achievement_number']), (21, 21, 21), ys_font(26))
|
||
text_draw.text((590, 183.9),raw_data['stats']['spiral_abyss'], (21, 21, 21), ys_font(26))
|
||
|
||
|
||
text_draw.text((635.4, 462.8), str(raw_data['stats']['anemoculus_number']), 'white', ys_font(26))
|
||
text_draw.text((635.4, 493.6), str(raw_data['stats']['geoculus_number']), 'white', ys_font(26))
|
||
text_draw.text((635.4, 524.8), str(raw_data['stats']['electroculus_number']), 'white', ys_font(26))
|
||
|
||
text_draw.text((224.6, 379.1),str(raw_data['stats']['common_chest_number']),'white', ys_font(28))
|
||
text_draw.text((436.5, 379.1),str(raw_data['stats']['exquisite_chest_number']),'white', ys_font(28))
|
||
text_draw.text((224.6, 416),str(raw_data['stats']['precious_chest_number']), 'white', ys_font(28))
|
||
text_draw.text((436.5, 416),str(raw_data['stats']['luxurious_chest_number']), 'white', ys_font(28))
|
||
|
||
text_draw.text((646.3, 379.1),str(raw_data['stats']['avatar_number']),'white', ys_font(26))
|
||
text_draw.text((646.3, 416),str(raw_data['stats']['way_point_number']),'white', ys_font(26))
|
||
|
||
|
||
text_draw.text((228.9, 502),str(raw_data['world_explorations'][3]['exploration_percentage']/10) + '%','white', ys_font(20))
|
||
text_draw.text((228.9, 538),'lv.' + str(raw_data['world_explorations'][3]['level']),'white', ys_font(20))
|
||
|
||
text_draw.text((451.1, 502),str(raw_data['world_explorations'][2]['exploration_percentage']/10) + '%','white', ys_font(20))
|
||
text_draw.text((451.1, 538),'lv.' + str(raw_data['world_explorations'][2]['level']),'white', ys_font(20))
|
||
|
||
text_draw.text((228.9, 606.4),str(raw_data['world_explorations'][1]['exploration_percentage']/10) + '%','white', ys_font(20))
|
||
text_draw.text((228.9, 641),'lv.' + str(raw_data['world_explorations'][1]['level']),'white', ys_font(20))
|
||
|
||
text_draw.text((451.1, 597),str(raw_data['world_explorations'][0]['exploration_percentage']/10) + '%','white', ys_font(20))
|
||
text_draw.text((451.1, 624),'lv.' + str(raw_data['world_explorations'][0]['level']),'white', ys_font(20))
|
||
text_draw.text((451.1, 652),'lv.' + str(raw_data['world_explorations'][0]['offerings'][0]['level']),'white', ys_font(20))
|
||
|
||
if len(raw_data['homes']):
|
||
text_draw.text((594.5, 565),'lv.' + str(raw_data['homes'][0]['level']),'white', ys_font(26))
|
||
text_draw.text((594.5, 598),str(raw_data['homes'][0]['visit_num']),'white', ys_font(26))
|
||
text_draw.text((594.5, 630),str(raw_data['homes'][0]['item_num']),'white', ys_font(26))
|
||
text_draw.text((594.5, 662),str(raw_data['homes'][0]['comfort_num']),'white', ys_font(26))
|
||
else:
|
||
text_draw.text((594.5, 616.6),'未开',(0, 0, 0), ys_font(26))
|
||
|
||
num = 0
|
||
|
||
for i in raw_data['avatars']:
|
||
if num < 15:
|
||
char = os.path.join(CHAR_PATH,str(char_data[num]['id']) + ".png")
|
||
char_img = Image.open(char)
|
||
char_draw = ImageDraw.Draw(char_img)
|
||
char_draw.text((40,108),f'Lv.{str(char_data[num]["level"])}',(21,21,21),ys_font(18))
|
||
char_draw.text((95.3,19),f'{str(char_data[num]["actived_constellation_num"])}','white',ys_font(18))
|
||
if str(char_data[num]["fetter"]) == "10":
|
||
char_draw.text((95.3,40.5),"F",(21,21,21),ys_font(17))
|
||
else:
|
||
char_draw.text((95.3,40.5),f'{str(char_data[num]["fetter"])}',(21,21,21),ys_font(18))
|
||
|
||
if num < 5:
|
||
char_crop = (74+133*num,750)
|
||
bg_img.paste(char_img,char_crop,char_img)
|
||
elif num >= 5 and num < 10:
|
||
char_crop = (74+133*(num-5),875)
|
||
bg_img.paste(char_img,(char_crop),char_img)
|
||
elif num >= 10 and num < 15:
|
||
char_crop = (74+133*(num-10),1000)
|
||
bg_img.paste(char_img,(char_crop),char_img)
|
||
else:
|
||
break
|
||
num = num+1
|
||
|
||
result_buffer = BytesIO()
|
||
bg_img.save(result_buffer, format='png')
|
||
imgmes = 'base64://' + b64encode(result_buffer.getvalue()).decode()
|
||
resultmes = f"[CQ:image,file={imgmes}]"
|
||
#return 'base64://' + b64encode(result_buffer.getvalue()).decode()
|
||
return resultmes
|
||
|
||
def get_weapon_pic():
|
||
f2 = open(os.path.join(WEAPON_PATH,"weapons.json"),"r")
|
||
ikk = json.load(f2)
|
||
for k in ikk:
|
||
m = ikk[f'{k}']['icon']
|
||
urllib.request.urlretrieve(f'{m}', os.path.join(WEAPON_PATH, m.split('/')[-1]))
|
||
print(k)
|
||
time.sleep(2)
|
||
|
||
async def draw_char_pic(uid,nickname,image = None):
|
||
is_edit = False
|
||
if image != None:
|
||
image_file= image.group(1)
|
||
image_data = image.group(2)
|
||
urllib.request.urlretrieve(f'{image_data}', os.path.join(TEXT_PATH,nickname + '.png'))
|
||
is_edit = True
|
||
|
||
raw_data = await GetInfo(uid)
|
||
|
||
if (raw_data["retcode"] != 0):
|
||
if (raw_data["retcode"] == 10001):
|
||
return ("Cookie错误/过期，请重置Cookie")
|
||
return (
|
||
"Api报错，返回内容为：\r\n"
|
||
+ str(raw_data) + "\r\n出现这种情况可能的UID输入错误 or 不存在"
|
||
)
|
||
else:
|
||
pass
|
||
|
||
bg_list = ['bg_img_1.png','bg_img_2.png',"bg_img_3.png"]
|
||
bg_path = os.path.join(TEXT_PATH,"bg.png")
|
||
bg2_path = os.path.join(TEXT_PATH,random.choice(bg_list))
|
||
fg_path = os.path.join(TEXT_PATH,"char_fg_img.png")
|
||
mask_path = os.path.join(TEXT_PATH,"char_mask_img.png")
|
||
|
||
if is_edit == True:
|
||
bg_path_edit = os.path.join(TEXT_PATH,f"{nickname}.png")
|
||
edit_bg = Image.open(bg_path_edit)
|
||
w, h = edit_bg.size
|
||
scale_f = w / h
|
||
scale = '%.4f' % scale_f
|
||
new_w = math.ceil(1200*float(scale))
|
||
new_h = math.ceil(800/float(scale))
|
||
if w > h:
|
||
bg_img2 = edit_bg.resize((new_w, 1200),Image.ANTIALIAS)
|
||
elif float(scale) > 0.66:
|
||
bg_img2 = edit_bg.resize((new_w, 1200),Image.ANTIALIAS)
|
||
else:
|
||
bg_img2 = edit_bg.resize((800, new_h),Image.ANTIALIAS)
|
||
bg_img = bg_img2.crop((0, 0, 800, 1200))
|
||
else:
|
||
bg_img = Image.open(bg2_path)
|
||
|
||
fg_img = Image.open(fg_path)
|
||
mask_img = Image.open(mask_path)
|
||
|
||
#img = Image.open(bg_path)
|
||
area = (37, 58, 764, 1142)
|
||
img_bb = bg_img.crop(area)
|
||
img_blur = img_bb.filter(ImageFilter.GaussianBlur(4))
|
||
|
||
bg_img.paste(img_blur, (37,268),mask_img)
|
||
|
||
bg_img.paste(fg_img,(0,0),fg_img)
|
||
text_draw = ImageDraw.Draw(bg_img)
|
||
|
||
#text_draw.text((192.6,128.3), f"{nickname}", (217,217,217), ys_font(32))
|
||
text_draw.text((297.7, 83.8), 'UID ' + f"{uid}", (245,245,245), ys_font(26))
|
||
|
||
|
||
charid_data = raw_data['data']["avatars"]
|
||
char_ids = []
|
||
char_rawdata = []
|
||
|
||
for i in charid_data:
|
||
char_ids.append(i["id"])
|
||
|
||
char_rawdata = await GetCharacter(uid,char_ids)
|
||
char_data = char_rawdata["data"]["avatars"]
|
||
|
||
bar_5_path = os.path.join(TEXT_PATH,"jinglian_5.png")
|
||
bar_5 = Image.open(bar_5_path)
|
||
#bar_5.thumbnail((13, 30))
|
||
bar_4_path = os.path.join(TEXT_PATH,"jinglian_4.png")
|
||
bar_4 = Image.open(bar_4_path)
|
||
#bar_4.thumbnail((13, 30))
|
||
|
||
num = 0
|
||
|
||
char_data.sort(key=lambda x: (-x['rarity'],-x['level'],-x['fetter']))
|
||
|
||
for i in char_data:
|
||
char_mingzuo = 0
|
||
for k in i['constellations']:
|
||
if k['is_actived'] == True:
|
||
char_mingzuo += 1
|
||
char_id = i["id"]
|
||
char_level = i["level"]
|
||
char_fetter = i['fetter']
|
||
char_rarity = i['rarity']
|
||
|
||
char_weapon = i['weapon']
|
||
char_weapon_name = i['weapon']['name']
|
||
char_weapon_level = i['weapon']['level']
|
||
char_weapon_jinglian = i['weapon']['affix_level']
|
||
char_weapon_icon = i['weapon']['icon']
|
||
|
||
|
||
char = os.path.join(CHAR_WEAPON_PATH,str(char_id) + ".png")
|
||
weapon = os.path.join(WEAPON_PATH, str(char_weapon_icon.split('/')[-1]))
|
||
|
||
char_img = Image.open(char)
|
||
weapon_img = Image.open(weapon)
|
||
weapon_img.thumbnail((57, 57))
|
||
char_draw = ImageDraw.Draw(char_img)
|
||
|
||
char_draw.text((47,109),f'Lv.{str(char_level)}',(21,21,21),ys_font(18))
|
||
|
||
char_draw.text((110,19.5),f'{str(char_mingzuo)}','white',ys_font(18))
|
||
|
||
if str(char_data[num]["fetter"]) == "10":
|
||
char_draw.text((110,41.5),"F",(21,21,21),ys_font(17))
|
||
else:
|
||
char_draw.text((110,41.5),f'{str(char_fetter)}',(21,21,21),ys_font(18))
|
||
|
||
char_img.paste(weapon_img,(70,52),weapon_img)
|
||
|
||
#if char_rarity == 4:
|
||
# for j in range(0,char_weapon_jinglian):
|
||
# char_img.paste(bar_4,(116,97-8*j),bar_4)
|
||
#elif char_rarity == 5:
|
||
# for j in range(0,char_weapon_jinglian):
|
||
# char_img.paste(bar_5,(116,97-8*j),bar_5)
|
||
|
||
if char_rarity == 4:
|
||
char_img.paste(bar_4,(110,92),bar_4)
|
||
char_draw.text((112,91),f'{str(char_weapon_jinglian)}','black',ys_font(14))
|
||
elif char_rarity == 5:
|
||
char_img.paste(bar_5,(110,92),bar_5)
|
||
char_draw.text((112,91),f'{str(char_weapon_jinglian)}','black',ys_font(14))
|
||
|
||
if num < 35:
|
||
char_crop = (53+137*(num%5),181+131*(num//5))
|
||
bg_img.paste(char_img,char_crop,char_img)
|
||
else:
|
||
break
|
||
num = num+1
|
||
|
||
result_buffer = BytesIO()
|
||
bg_img.save(result_buffer, format='png')
|
||
imgmes = 'base64://' + b64encode(result_buffer.getvalue()).decode()
|
||
resultmes = f"[CQ:image,file={imgmes}]"
|
||
#return 'base64://' + b64encode(result_buffer.getvalue()).decode()
|
||
return resultmes
|
||
|
||
async def get_char_data(uid,nickname,image = None):
|
||
pass |