877 lines
34 KiB
Python
877 lines
34 KiB
Python
import math
|
||
import os
|
||
import random
|
||
import sqlite3
|
||
import time
|
||
import traceback
|
||
import urllib
|
||
import numpy as np
|
||
from typing import List
|
||
from wordcloud import WordCloud
|
||
from PIL import Image, ImageDraw, ImageFilter
|
||
from pyrogram.types import Message
|
||
|
||
from defs.db2 import MysSign, GetDaily, cacheDB, GetMysInfo, errorDB, GetInfo, GetSpiralAbyssInfo, GetAward
|
||
from defs.event import ys_font
|
||
from genshinstats.genshinstats.daily import DailyRewardInfo
|
||
|
||
WEAPON_PATH = os.path.join("assets", 'weapon')
|
||
BG_PATH = os.path.join("assets", "bg")
|
||
CHAR_DONE_PATH = os.path.join("assets", 'char_done')
|
||
BG2_PATH = os.path.join("assets", "bg2")
|
||
CHAR_PATH = os.path.join("assets", "characters")
|
||
CHAR_IMG_PATH = os.path.join("assets", 'char_img')
|
||
REL_PATH = os.path.join("assets", "reliquaries")
|
||
avatar_json = {
|
||
"Albedo": "阿贝多",
|
||
"Ambor": "安柏",
|
||
"Barbara": "芭芭拉",
|
||
"Beidou": "北斗",
|
||
"Bennett": "班尼特",
|
||
"Chongyun": "重云",
|
||
"Diluc": "迪卢克",
|
||
"Diona": "迪奥娜",
|
||
"Eula": "优菈",
|
||
"Fischl": "菲谢尔",
|
||
"Ganyu": "甘雨",
|
||
"Hutao": "胡桃",
|
||
"Jean": "琴",
|
||
"Kazuha": "枫原万叶",
|
||
"Kaeya": "凯亚",
|
||
"Ayaka": "神里绫华",
|
||
"Keqing": "刻晴",
|
||
"Klee": "可莉",
|
||
"Lisa": "丽莎",
|
||
"Mona": "莫娜",
|
||
"Ningguang": "凝光",
|
||
"Noel": "诺艾尔",
|
||
"Qiqi": "七七",
|
||
"Razor": "雷泽",
|
||
"Rosaria": "罗莎莉亚",
|
||
"Sucrose": "砂糖",
|
||
"Tartaglia": "达达利亚",
|
||
"Venti": "温迪",
|
||
"Xiangling": "香菱",
|
||
"Xiao": "魈",
|
||
"Xingqiu": "行秋",
|
||
"Xinyan": "辛焱",
|
||
"Yanfei": "烟绯",
|
||
"Zhongli": "钟离",
|
||
"Yoimiya": "宵宫",
|
||
"Sayu": "早柚",
|
||
"Shogun": "雷电将军",
|
||
"Aloy": "埃洛伊",
|
||
"Sara": "九条裟罗",
|
||
"Kokomi": "珊瑚宫心海",
|
||
"Shenhe": "申鹤"
|
||
}
|
||
award_json = {
|
||
"Mail": "邮件奖励",
|
||
"Events": "活动奖励",
|
||
"Adventure": "冒险奖励",
|
||
"Daily Activity": "每日活跃",
|
||
"Quests": "任务奖励",
|
||
"Spiral Abyss": "深境螺旋",
|
||
"Other": "其他",
|
||
}
|
||
daily_im = '''
|
||
*数据刷新可能存在一定延迟，请以当前游戏实际数据为准{}
|
||
==============
|
||
原粹树脂：{}/{}{}
|
||
每日委托：{}/{} 奖励{}领取
|
||
周本减半：{}/{}
|
||
洞天宝钱：{}
|
||
探索派遣：
|
||
总数/完成/上限：{}/{}/{}
|
||
{}'''
|
||
month_im = '''
|
||
==============
|
||
{}
|
||
UID：{}
|
||
==============
|
||
本日获取原石：{}
|
||
本日获取摩拉：{}
|
||
==============
|
||
本月获取原石：{}
|
||
本月获取摩拉：{}
|
||
==============
|
||
上月获取原石：{}
|
||
上月获取摩拉：{}
|
||
==============
|
||
原石收入组成：
|
||
{}=============='''
|
||
|
||
|
||
def trans_award(string: str) -> str:
|
||
if string in award_json:
|
||
return award_json[string]
|
||
else:
|
||
return string
|
||
|
||
|
||
async def award(uid):
|
||
data = await GetAward(uid)
|
||
if data is None:
|
||
# 等级太低，或者不够活跃
|
||
return "暂无每月统计信息"
|
||
nickname = data['nickname']
|
||
day_stone = data['day_data']['current_primogems']
|
||
day_mora = data['day_data']['current_mora']
|
||
month_stone = data['month_data']['current_primogems']
|
||
month_mora = data['month_data']['current_mora']
|
||
lastmonth_stone = data['month_data']['last_primogems']
|
||
lastmonth_mora = data['month_data']['last_mora']
|
||
group_str = ''
|
||
for i in data['month_data']['group_by']:
|
||
group_str = group_str + \
|
||
trans_award(i['action']) + "：" + str(i['num']) + \
|
||
"（" + str(i['percent']) + "%）" + '\n'
|
||
|
||
im = month_im.format(nickname, uid, day_stone, day_mora, month_stone, month_mora,
|
||
lastmonth_stone, lastmonth_mora, group_str)
|
||
return im
|
||
|
||
|
||
# 签到函数
|
||
async def sign(uid):
|
||
try:
|
||
sign_data = await MysSign(uid)
|
||
if sign_data:
|
||
if isinstance(sign_data, DailyRewardInfo):
|
||
mes_im = "已经签到过了！"
|
||
get_im = f"本月已经签到了 {sign_data.claimed_rewards} 天"
|
||
else:
|
||
mes_im = "签到成功"
|
||
get_im = f"本次签到获得{sign_data['name']}x{sign_data['cnt']}"
|
||
im = mes_im + "!" + "\n" + get_im
|
||
else:
|
||
im = "签到失败，请检查Cookies是否失效。"
|
||
except:
|
||
im = "签到失败，请检查Cookies是否失效。"
|
||
return im
|
||
|
||
|
||
# 统计状态函数
|
||
async def daily(mode="push", uid=None):
|
||
def seconds2hours(seconds: int) -> str:
|
||
m, s = divmod(int(seconds), 60)
|
||
h, m = divmod(m, 60)
|
||
return "%02d:%02d:%02d" % (h, m, s)
|
||
|
||
temp_list = []
|
||
conn = sqlite3.connect('ID_DATA_OR.db')
|
||
c = conn.cursor()
|
||
if mode == "ask":
|
||
c_data = ([uid, 0, 0, 0, 0, 0, 0],)
|
||
else:
|
||
cursor = c.execute(
|
||
"SELECT * FROM NewCookiesTable WHERE StatusA != ?", ("off",))
|
||
c_data = cursor.fetchall()
|
||
|
||
for row in c_data:
|
||
try:
|
||
dailydata = await GetDaily(str(row[0]))
|
||
except Exception as e:
|
||
if str(e).find("用户信息不匹配"):
|
||
temp_list.append(
|
||
{"qid": row[2], "gid": row[3], "message": "你的推送状态有误；可能是uid绑定错误或没有在米游社打开“实时便筏”功能。"})
|
||
else:
|
||
traceback.print_exc()
|
||
else:
|
||
current_resin = dailydata['resin']
|
||
|
||
current_expedition_num = len(dailydata['expeditions'])
|
||
max_expedition_num = dailydata['max_expeditions']
|
||
finished_expedition_num = 0
|
||
expedition_info: List[str] = []
|
||
for expedition in dailydata['expeditions']:
|
||
avatar: str = expedition['icon'][92:-4]
|
||
try:
|
||
avatar_name: str = avatar_json[avatar]
|
||
except KeyError:
|
||
avatar_name: str = avatar
|
||
|
||
if expedition['status'] == 'Finished':
|
||
expedition_info.append(f"{avatar_name} 探索完成")
|
||
finished_expedition_num += 1
|
||
else:
|
||
remained_timed: str = seconds2hours(
|
||
expedition['remaining_time'])
|
||
expedition_info.append(
|
||
f"{avatar_name} 剩余时间{remained_timed}")
|
||
|
||
if current_resin >= row[6] or dailydata["max_realm_currency"] - dailydata[
|
||
"realm_currency"] <= 100 or finished_expedition_num > 0:
|
||
tip = ''
|
||
|
||
if current_resin >= row[6] != 0:
|
||
tip += "\n==============\n你的树脂快满了！"
|
||
if dailydata["max_realm_currency"] - dailydata["realm_currency"] <= 100:
|
||
tip += "\n==============\n你的洞天宝钱快满了！"
|
||
if finished_expedition_num > 0:
|
||
tip += "\n==============\n你有探索派遣完成了！"
|
||
max_resin = dailydata['max_resin']
|
||
rec_time = ''
|
||
# print(dailydata)
|
||
if current_resin < 160:
|
||
resin_recovery_time = seconds2hours(
|
||
dailydata['until_resin_limit'])
|
||
next_resin_rec_time = seconds2hours(
|
||
8 * 60 - ((dailydata['max_resin'] - dailydata['resin']) * 8 * 60 - int(
|
||
dailydata['until_resin_limit'])))
|
||
rec_time = f' ({next_resin_rec_time}/{resin_recovery_time})'
|
||
|
||
finished_task_num = dailydata['completed_commissions']
|
||
total_task_num = dailydata['total_commissions']
|
||
is_extra_got = '已' if dailydata['claimed_commission_reward'] else '未'
|
||
|
||
resin_discount_num_limit = dailydata['max_boss_discounts']
|
||
used_resin_discount_num = resin_discount_num_limit - \
|
||
dailydata['remaining_boss_discounts']
|
||
|
||
coin = f'{dailydata["realm_currency"]}/{dailydata["max_realm_currency"]}'
|
||
if dailydata["realm_currency"] < dailydata["max_realm_currency"]:
|
||
coin_rec_time = seconds2hours(int(dailydata["until_realm_currency_limit"]))
|
||
coin_add_speed = math.ceil((dailydata["max_realm_currency"] - dailydata["realm_currency"]) / (
|
||
int(dailydata["until_realm_currency_limit"]) / 60 / 60))
|
||
coin += f'（{coin_rec_time} 约{coin_add_speed}/h）'
|
||
|
||
expedition_data = "\n".join(expedition_info)
|
||
send_mes = daily_im.format(tip, current_resin, max_resin, rec_time,
|
||
finished_task_num, total_task_num,
|
||
is_extra_got, used_resin_discount_num,
|
||
resin_discount_num_limit, coin,
|
||
current_expedition_num, finished_expedition_num, max_expedition_num,
|
||
expedition_data)
|
||
|
||
temp_list.append(
|
||
{"qid": row[2], "gid": row[3], "message": send_mes})
|
||
return temp_list
|
||
|
||
|
||
def create_rounded_rectangle_mask(rectangle, radius):
|
||
solid_fill = (50, 50, 50, 255)
|
||
i = Image.new("RGBA", rectangle.size, (0, 0, 0, 0))
|
||
|
||
corner = Image.new('RGBA', (radius, radius), (0, 0, 0, 0))
|
||
draw = ImageDraw.Draw(corner)
|
||
draw.pieslice((0, 0, radius * 2, radius * 2), 180, 270, fill=solid_fill)
|
||
|
||
mx, my = rectangle.size
|
||
|
||
i.paste(corner, (0, 0), corner)
|
||
i.paste(corner.rotate(90), (0, my - radius), corner.rotate(90))
|
||
i.paste(corner.rotate(180), (mx - radius, my - radius), corner.rotate(180))
|
||
i.paste(corner.rotate(270), (mx - radius, 0), corner.rotate(270))
|
||
|
||
draw = ImageDraw.Draw(i)
|
||
draw.rectangle([(radius, 0), (mx - radius, my)], fill=solid_fill)
|
||
draw.rectangle([(0, radius), (mx, my - radius)], fill=solid_fill)
|
||
|
||
return i
|
||
|
||
|
||
def get_weapon_pic(url):
|
||
urllib.request.urlretrieve(url, os.path.join(WEAPON_PATH, url.split('/')[-1]))
|
||
|
||
|
||
def get_char_pic(id, url):
|
||
urllib.request.urlretrieve(f'{url}', os.path.join(CHAR_PATH, f'{id}.png'))
|
||
|
||
|
||
def get_charimg_pic(url):
|
||
urllib.request.urlretrieve(url, os.path.join(CHAR_IMG_PATH, url.split('/')[-1]))
|
||
|
||
|
||
def get_rel_pic(url):
|
||
urllib.request.urlretrieve(url, os.path.join(REL_PATH, url.split('/')[-1]))
|
||
|
||
|
||
async def draw_pic(uid, message: Message, nickname="1", mode=2, role_level=None):
|
||
# 获取Cookies，如果没有能使用的则return
|
||
while 1:
|
||
use_cookies = cacheDB(uid, mode - 1)
|
||
if use_cookies == '':
|
||
return "绑定记录不存在。"
|
||
elif use_cookies == "没有可以使用的Cookies！":
|
||
return "没有可以使用的Cookies！"
|
||
|
||
if mode == 3:
|
||
mys_data = await GetMysInfo(uid, use_cookies)
|
||
|
||
uid = mys_data[0]['uid']
|
||
nickname = mys_data[0]['nickname']
|
||
role_level = mys_data[0]['level']
|
||
|
||
try:
|
||
raw_data = await GetInfo(uid, use_cookies)
|
||
except Exception as e:
|
||
if str(e).find("Cannnot get data for more than 30 accounts per day.") != -1:
|
||
# return ("当前cookies已达到30人上限！")
|
||
errorDB(use_cookies, "limit30")
|
||
elif str(e).find("Login cookies have not been provided") != -1:
|
||
# return ("Cookie错误/过期，请重置Cookie")
|
||
errorDB(use_cookies, "error")
|
||
else:
|
||
traceback.print_exc()
|
||
return (
|
||
"Api报错\r\n"
|
||
+ "\r\n出现这种情况可能是UID输入错误 or 不存在"
|
||
)
|
||
else:
|
||
break
|
||
|
||
# 获取背景图片
|
||
bg2_path = os.path.join(BG_PATH, random.choice([x for x in os.listdir(BG_PATH)
|
||
if os.path.isfile(os.path.join(BG_PATH, x))]))
|
||
|
||
if message.media:
|
||
image_data = await message.download()
|
||
edit_bg = Image.open(image_data)
|
||
else:
|
||
edit_bg = Image.open(bg2_path)
|
||
|
||
# 获取背景主色
|
||
q = edit_bg.quantize(colors=3, method=2)
|
||
bg_num_temp = 0
|
||
for i in range(0, 3):
|
||
bg = tuple(q.getpalette()[i * 3:(i * 3) + 3])
|
||
bg_num = bg[0] + bg[1] + bg[2]
|
||
if bg_num >= bg_num_temp:
|
||
bg_num_temp = bg_num
|
||
bg_color = (bg[0], bg[1], bg[2])
|
||
|
||
# 通过背景主色（bg_color）确定文字主色
|
||
r = 140
|
||
if max(*bg_color) > 255 - r:
|
||
r *= -1
|
||
new_color = (math.floor(bg_color[0] + r if bg_color[0] + r <= 255 else 255),
|
||
math.floor(bg_color[1] + r if bg_color[1] + r <= 255 else 255),
|
||
math.floor(bg_color[2] + r if bg_color[2] + r <= 255 else 255))
|
||
|
||
# 确定texture2D路径
|
||
panle1_path = os.path.join(BG2_PATH, "panle_1.png")
|
||
panle3_path = os.path.join(BG2_PATH, "panle_3.png")
|
||
|
||
avatar_bg_path = os.path.join(BG2_PATH, "avatar_bg.png")
|
||
avatar_fg_path = os.path.join(BG2_PATH, "avatar_fg.png")
|
||
|
||
all_mask_path = os.path.join(BG2_PATH, "All_Mask.png")
|
||
|
||
# 记录数据
|
||
char_datas = raw_data["characters"]
|
||
|
||
# 确定角色占用行数
|
||
char_num = len(char_datas)
|
||
char_hang = 1 + (char_num - 1) // 6 if char_num > 8 else char_num
|
||
|
||
# 确定整体图片的长宽
|
||
based_w = 900
|
||
based_h = 890 + char_hang * 130 if char_num > 8 else 890 + char_hang * 110
|
||
based_scale = '%.3f' % (based_w / based_h)
|
||
|
||
# 通过确定的长宽比，缩放背景图片
|
||
w, h = edit_bg.size
|
||
scale_f = '%.3f' % (w / h)
|
||
new_w = math.ceil(based_h * float(scale_f))
|
||
new_h = math.ceil(based_w / float(scale_f))
|
||
if scale_f > based_scale:
|
||
bg_img2 = edit_bg.resize((new_w, based_h), Image.ANTIALIAS)
|
||
else:
|
||
bg_img2 = edit_bg.resize((based_w, new_h), Image.ANTIALIAS)
|
||
bg_img = bg_img2.crop((0, 0, 900, based_h))
|
||
|
||
# 转换遮罩的颜色、大小匹配，并paste上去
|
||
all_mask = Image.open(all_mask_path).resize(bg_img.size, Image.ANTIALIAS)
|
||
all_mask_img = Image.new("RGBA", (based_w, based_h), bg_color)
|
||
bg_img.paste(all_mask_img, (0, 0), all_mask)
|
||
|
||
# 操作图片
|
||
panle1 = Image.open(panle1_path)
|
||
panle3 = Image.open(panle3_path)
|
||
avatar_bg = Image.open(avatar_bg_path)
|
||
avatar_fg = Image.open(avatar_fg_path)
|
||
|
||
# 确定主体框架
|
||
avatar_bg_color = Image.new("RGBA", (316, 100), bg_color)
|
||
panle1_color = Image.new("RGBA", (900, 800), new_color)
|
||
bg_img.paste(panle1_color, (0, 0), panle1)
|
||
bg_img.paste(panle3, (0, char_hang * 130 + 800) if char_num > 8 else (0, char_hang * 110 + 800), panle3)
|
||
bg_img.paste(avatar_bg_color, (113, 98), avatar_bg)
|
||
bg_img.paste(avatar_fg, (114, 95), avatar_fg)
|
||
|
||
# 绘制基础信息文字
|
||
text_draw = ImageDraw.Draw(bg_img)
|
||
|
||
if role_level:
|
||
text_draw.text((140, 200), "冒险等级：" + f"{role_level}", new_color, ys_font(20))
|
||
|
||
text_draw.text((220, 123), f"{nickname}", new_color, ys_font(32))
|
||
text_draw.text((235, 163), 'UID ' + f"{uid}", new_color, ys_font(14))
|
||
|
||
# 活跃天数/成就数量/深渊信息
|
||
text_draw.text((640, 94.8), str(raw_data['stats']['active_days']), new_color, ys_font(26))
|
||
text_draw.text((640, 139.3), str(raw_data['stats']['achievements']), new_color, ys_font(26))
|
||
text_draw.text((640, 183.9), raw_data['stats']['spiral_abyss'], new_color, ys_font(26))
|
||
|
||
# 宝箱
|
||
text_draw.text((258, 375.4), str("未知"), new_color, ys_font(24))
|
||
text_draw.text((258, 425.4), str(raw_data['stats']['common_chests']), new_color, ys_font(24))
|
||
text_draw.text((258, 475.4), str(raw_data['stats']['exquisite_chests']), new_color, ys_font(24))
|
||
text_draw.text((258, 525.4), str(raw_data['stats']['precious_chests']), new_color, ys_font(24))
|
||
text_draw.text((258, 575.4), str(raw_data['stats']['luxurious_chests']), new_color, ys_font(24))
|
||
|
||
# 已获角色
|
||
text_draw.text((740, 547), str(raw_data['stats']['characters']), new_color, ys_font(24))
|
||
|
||
# 开启锚点和秘境数量
|
||
text_draw.text((258, 625.4), str(raw_data['stats']['unlocked_waypoints']), new_color, ys_font(24))
|
||
text_draw.text((258, 675.4), str(raw_data['stats']['unlocked_domains']), new_color, ys_font(24))
|
||
|
||
# 蒙德
|
||
text_draw.text((490, 370), str(raw_data['explorations'][4]['explored']) + '%', new_color,
|
||
ys_font(22))
|
||
text_draw.text((490, 400), 'lv.' + str(raw_data['explorations'][4]['level']), new_color, ys_font(22))
|
||
text_draw.text((513, 430), str(raw_data['stats']['anemoculi']), new_color, ys_font(22))
|
||
|
||
# 璃月
|
||
text_draw.text((490, 490), str(raw_data['explorations'][3]['explored']) + '%', new_color,
|
||
ys_font(22))
|
||
text_draw.text((490, 520), 'lv.' + str(raw_data['explorations'][3]['level']), new_color, ys_font(22))
|
||
text_draw.text((513, 550), str(raw_data['stats']['geoculi']), new_color, ys_font(22))
|
||
|
||
# 雪山
|
||
text_draw.text((745, 373.5), str(raw_data['explorations'][2]['explored']) + '%', new_color,
|
||
ys_font(22))
|
||
text_draw.text((745, 407.1), 'lv.' + str(raw_data['explorations'][2]['level']), new_color, ys_font(22))
|
||
|
||
# 稻妻
|
||
text_draw.text((490, 608), str(raw_data['explorations'][1]['explored']) + '%', new_color,
|
||
ys_font(22))
|
||
text_draw.text((490, 635), 'lv.' + str(raw_data['explorations'][1]['level']), new_color, ys_font(22))
|
||
text_draw.text((490, 662), 'lv.' + str(raw_data['explorations'][1]['offerings'][0]['level']), new_color,
|
||
ys_font(22))
|
||
text_draw.text((513, 689), str(raw_data['stats']['electroculi']), new_color, ys_font(22))
|
||
|
||
# 渊下宫
|
||
text_draw.text((745, 480), str(raw_data['explorations'][0]['explored']) + '%', new_color,
|
||
ys_font(22))
|
||
|
||
# 家园
|
||
if raw_data['teapot']:
|
||
text_draw.text((693, 582.4), 'lv.' + str(raw_data['teapot']['level']), new_color, ys_font(22))
|
||
text_draw.text((693, 620.4), str(raw_data['teapot']['visitors']), new_color, ys_font(22))
|
||
text_draw.text((693, 658.4), str(raw_data['teapot']['items']), new_color, ys_font(22))
|
||
text_draw.text((693, 696.4), str(raw_data['teapot']['comfort']), new_color, ys_font(22))
|
||
else:
|
||
text_draw.text((693, 582.4), "未开", new_color, ys_font(22))
|
||
text_draw.text((693, 620.4), "未开", new_color, ys_font(22))
|
||
text_draw.text((693, 658.4), "未开", new_color, ys_font(22))
|
||
text_draw.text((693, 696.4), "未开", new_color, ys_font(22))
|
||
|
||
# 确定texture2D路径
|
||
charpic_mask_path = os.path.join(BG2_PATH, "charpic_mask.png")
|
||
weaponpic_mask_path = os.path.join(BG2_PATH, "weaponpic_mask.png")
|
||
|
||
def getText(star, step):
|
||
return os.path.join(BG2_PATH, "{}s_{}.png".format(str(star), str(step)))
|
||
|
||
charpic_mask = Image.open(charpic_mask_path)
|
||
weaponpic_mask = Image.open(weaponpic_mask_path)
|
||
s5s1 = Image.open(getText(5, 1))
|
||
s5s2 = Image.open(getText(5, 2))
|
||
s5s3 = Image.open(getText(5, 3))
|
||
s5s4 = Image.open(getText(5, 4))
|
||
s4s1 = Image.open(getText(4, 1))
|
||
s4s2 = Image.open(getText(4, 2))
|
||
s4s3 = Image.open(getText(4, 3))
|
||
s4s4 = Image.open(getText(4, 4))
|
||
s3s3 = Image.open(getText(3, 3))
|
||
s2s3 = Image.open(getText(2, 3))
|
||
s1s3 = Image.open(getText(1, 3))
|
||
|
||
char_bg_path = os.path.join(BG2_PATH, "char_bg.png")
|
||
char_fg_path = os.path.join(BG2_PATH, "char_fg.png")
|
||
|
||
char_bg = Image.open(char_bg_path)
|
||
char_fg = Image.open(char_fg_path)
|
||
|
||
char_color = (math.floor(bg_color[0] + 10 if bg_color[0] + r <= 255 else 255),
|
||
math.floor(bg_color[1] + 10 if bg_color[1] + r <= 255 else 255),
|
||
math.floor(bg_color[2] + 10 if bg_color[2] + r <= 255 else 255))
|
||
|
||
charset_mask = Image.new("RGBA", (900, 130), char_color)
|
||
|
||
num = 0
|
||
char_datas.sort(key=lambda x: (-x['rarity'], -x['level'], -x['friendship']))
|
||
|
||
if char_num > 8:
|
||
for i in char_datas:
|
||
# char_mingzuo = 0
|
||
|
||
char_name = i["name"]
|
||
char_id = i["id"]
|
||
char_level = i["level"]
|
||
char_fetter = i['friendship']
|
||
char_rarity = i['rarity']
|
||
|
||
# char_weapon_star = i['weapon']['rarity']
|
||
# char_weapon_jinglian = i['weapon']['affix_level']
|
||
# char_weapon_icon = i['weapon']['icon']
|
||
|
||
# if not os.path.exists(os.path.join(WEAPON_PATH, str(char_weapon_icon.split('/')[-1]))):
|
||
# get_weapon_pic(char_weapon_icon)
|
||
if not os.path.exists(os.path.join(CHAR_PATH, str(i['id']) + ".png")):
|
||
get_char_pic(i['id'], i['icon'])
|
||
|
||
char = os.path.join(CHAR_PATH, str(char_id) + ".png")
|
||
# weapon = os.path.join(WEAPON_PATH, str(char_weapon_icon.split('/')[-1]))
|
||
|
||
char_img = Image.open(char)
|
||
char_img = char_img.resize((100, 100), Image.ANTIALIAS)
|
||
# weapon_img = Image.open(weapon)
|
||
# weapon_img = weapon_img.resize((47, 47), Image.ANTIALIAS)
|
||
|
||
charpic = Image.new("RGBA", (125, 140))
|
||
|
||
if char_rarity == 5:
|
||
charpic.paste(s5s1, (0, 0), s5s1)
|
||
baseda = Image.new("RGBA", (100, 100))
|
||
cc = Image.composite(char_img, baseda, charpic_mask)
|
||
charpic.paste(cc, (6, 15), cc)
|
||
charpic.paste(s5s2, (0, 0), s5s2)
|
||
# if char_weapon_star == 5:
|
||
# charpic.paste(s5s3, (0, 0), s5s3)
|
||
# elif char_weapon_star == 4:
|
||
# charpic.paste(s4s3, (0, 0), s4s3)
|
||
# elif char_weapon_star == 3:
|
||
# charpic.paste(s3s3, (0, 0), s3s3)
|
||
# elif char_weapon_star == 2:
|
||
# charpic.paste(s2s3, (0, 0), s2s3)
|
||
# elif char_weapon_star == 1:
|
||
# charpic.paste(s1s3, (0, 0), s1s3)
|
||
basedb = Image.new("RGBA", (47, 47))
|
||
# dd = Image.composite(weapon_img, basedb, weaponpic_mask)
|
||
# charpic.paste(dd, (69, 62), dd)
|
||
charpic.paste(s5s4, (0, 0), s5s4)
|
||
|
||
else:
|
||
charpic.paste(s4s1, (0, 0), s4s1)
|
||
baseda = Image.new("RGBA", (100, 100))
|
||
cc = Image.composite(char_img, baseda, charpic_mask)
|
||
charpic.paste(cc, (6, 15), cc)
|
||
charpic.paste(s4s2, (0, 0), s4s2)
|
||
# if char_weapon_star == 5:
|
||
# charpic.paste(s5s3, (0, 0), s5s3)
|
||
# elif char_weapon_star == 4:
|
||
# charpic.paste(s4s3, (0, 0), s4s3)
|
||
# elif char_weapon_star == 3:
|
||
# charpic.paste(s3s3, (0, 0), s3s3)
|
||
# elif char_weapon_star == 2:
|
||
# charpic.paste(s2s3, (0, 0), s2s3)
|
||
# elif char_weapon_star == 1:
|
||
# charpic.paste(s1s3, (0, 0), s1s3)
|
||
basedb = Image.new("RGBA", (47, 47))
|
||
# dd = Image.composite(weapon_img, basedb, weaponpic_mask)
|
||
# charpic.paste(dd, (69, 62), dd)
|
||
charpic.paste(s4s4, (0, 0), s4s4)
|
||
|
||
char_draw = ImageDraw.Draw(charpic)
|
||
char_draw.text((38, 106), f'Lv.{str(char_level)}', (21, 21, 21), ys_font(18))
|
||
# 无法读取 char_draw.text((104.5, 91.5), f'{str(char_weapon_jinglian)}', 'white', ys_font(10))
|
||
# 无法读取 char_draw.text((99, 19.5), f'{str(char_mingzuo)}', 'white', ys_font(18))
|
||
if str(i["friendship"]) == "10" or str(char_name) == "旅行者":
|
||
char_draw.text((98, 42), "♥", (21, 21, 21), ys_font(14))
|
||
else:
|
||
char_draw.text((100, 41), f'{str(char_fetter)}', (21, 21, 21), ys_font(16))
|
||
|
||
char_crop = (68 + 129 * (num % 6), 800 + 130 * (num // 6))
|
||
bg_img.paste(charpic, char_crop, charpic)
|
||
num = num + 1
|
||
else:
|
||
for i in char_datas:
|
||
# char_mingzuo = 0
|
||
|
||
char_name = i["name"]
|
||
char_id = i["id"]
|
||
char_level = i["level"]
|
||
char_fetter = i['friendship']
|
||
char_rarity = i['rarity']
|
||
char_img_icon = i["icon"].replace("https://upload-os-bbs.mihoyo.com/game_record/genshin/character_icon/",
|
||
"https://upload-bbs.mihoyo.com/game_record/genshin/character_image/")
|
||
char_img_icon = char_img_icon.replace(".png", "@2x.png")
|
||
|
||
# char_weapon_star = i['weapon']['rarity']
|
||
# char_weapon_jinglian = i['weapon']['affix_level']
|
||
# char_weapon_icon = i['weapon']['icon']
|
||
|
||
# if not os.path.exists(os.path.join(WEAPON_PATH, str(char_weapon_icon.split('/')[-1]))):
|
||
# get_weapon_pic(char_weapon_icon)
|
||
if not os.path.exists(os.path.join(CHAR_IMG_PATH, str(char_img_icon.split('/')[-1]))):
|
||
get_charimg_pic(char_img_icon)
|
||
if not os.path.exists(os.path.join(CHAR_PATH, str(i['id']) + ".png")):
|
||
get_char_pic(i['id'], i['icon'])
|
||
|
||
char = os.path.join(CHAR_PATH, str(char_id) + ".png")
|
||
# weapon = os.path.join(WEAPON_PATH, str(char_weapon_icon.split('/')[-1]))
|
||
char_stand_img = os.path.join(CHAR_IMG_PATH, str(char_img_icon.split('/')[-1]))
|
||
char_stand_mask = Image.open(os.path.join(BG2_PATH, "stand_mask.png"))
|
||
|
||
char_stand = Image.open(char_stand_img)
|
||
char_img = Image.open(char)
|
||
char_img = char_img.resize((100, 100), Image.ANTIALIAS)
|
||
# weapon_img = Image.open(weapon)
|
||
# weapon_img = weapon_img.resize((47, 47), Image.ANTIALIAS)
|
||
|
||
charpic = Image.new("RGBA", (900, 130))
|
||
charpic_temp = Image.new("RGBA", (900, 130))
|
||
|
||
charpic.paste(charset_mask, (0, 0), char_bg)
|
||
|
||
# weapon_bg = Image.open(getText(char_weapon_star, 3))
|
||
# charpic.paste(weapon_bg, (72, 10), weapon_bg)
|
||
charpic_temp.paste(char_img, (81, 13), charpic_mask)
|
||
charpic_temp.paste(char_stand, (335, -99), char_stand_mask)
|
||
charpic_temp.paste(char_fg, (0, 0), char_fg)
|
||
# charpic_temp.paste(weapon_img, (141, 72), weaponpic_mask)
|
||
# temp = Image.composite(weapon_img, basedb, weaponpic_mask)
|
||
charpic.paste(charpic_temp, (0, 0), charpic_temp)
|
||
|
||
# for _, k in enumerate(i["reliquaries"]):
|
||
# if not os.path.exists(os.path.join(REL_PATH, str(k["icon"].split('/')[-1]))):
|
||
# get_rel_pic(k["icon"])
|
||
# rel = os.path.join(REL_PATH, str(k["icon"].split('/')[-1]))
|
||
# rel_img = Image.open(rel).resize((43, 43), Image.ANTIALIAS)
|
||
# rel_bg = Image.open(getText(k["rarity"], 3))
|
||
#
|
||
# if k["pos_name"] == "生之花":
|
||
# charpic.paste(rel_bg, (287 + 55 * 0, -14), rel_bg)
|
||
# charpic.paste(rel_img, (360 + 55 * 0, 49), rel_img)
|
||
# elif k["pos_name"] == "死之羽":
|
||
# charpic.paste(rel_bg, (287 + 55 * 1, -14), rel_bg)
|
||
# charpic.paste(rel_img, (360 + 55 * 1, 49), rel_img)
|
||
# elif k["pos_name"] == "时之沙":
|
||
# charpic.paste(rel_bg, (287 + 55 * 2, -14), rel_bg)
|
||
# charpic.paste(rel_img, (360 + 55 * 2, 49), rel_img)
|
||
# elif k["pos_name"] == "空之杯":
|
||
# charpic.paste(rel_bg, (287 + 55 * 3, -14), rel_bg)
|
||
# charpic.paste(rel_img, (360 + 55 * 3, 49), rel_img)
|
||
# elif k["pos_name"] == "理之冠":
|
||
# charpic.paste(rel_bg, (287 + 55 * 4, -14), rel_bg)
|
||
# charpic.paste(rel_img, (360 + 55 * 4, 49), rel_img)
|
||
|
||
char_draw = ImageDraw.Draw(charpic)
|
||
|
||
char_draw.text((182, 39), i["name"], new_color, ys_font(22))
|
||
char_draw.text((272, 45), f'Lv.{str(char_level)}', new_color, ys_font(18))
|
||
|
||
# char_draw.text((104.5,91.5),f'{str(char_weapon_jinglian)}',new_color,ys_font(10))
|
||
# char_draw.text((267, 77), f'{str(char_mingzuo)}', new_color, ys_font(18))
|
||
|
||
char_draw.text((209, 77), f'{str(i["friendship"])}' if str(char_name) != "旅行者" else "10", new_color,
|
||
ys_font(18))
|
||
char_crop = (0, 800 + 110 * num)
|
||
num += 1
|
||
bg_img.paste(charpic, char_crop, charpic)
|
||
|
||
# 转换之后发送
|
||
bg_img = bg_img.convert('RGB')
|
||
bg_img.save(f"temp{os.sep}uid.jpg", format='JPEG', subsampling=0, quality=90)
|
||
try:
|
||
if message.media:
|
||
os.remove(image_data) # noqa
|
||
except:
|
||
pass
|
||
return f"temp{os.sep}uid.jpg"
|
||
|
||
|
||
async def draw_wordcloud(uid, message: Message, mode=2):
|
||
while 1:
|
||
use_cookies = cacheDB(uid, mode - 1)
|
||
if use_cookies == '':
|
||
return "绑定记录不存在。"
|
||
elif use_cookies == "没有可以使用的Cookies！":
|
||
return "没有可以使用的Cookies！"
|
||
|
||
if mode == 3:
|
||
mys_data = await GetMysInfo(uid, use_cookies)
|
||
|
||
uid = mys_data[0]['uid']
|
||
|
||
raw_Abyss_data = await GetSpiralAbyssInfo(uid, use_cookies)
|
||
try:
|
||
raw_data = await GetInfo(uid, use_cookies)
|
||
except Exception as e:
|
||
if str(e).find("Cannnot get data for more than 30 accounts per day.") != -1:
|
||
# return ("当前cookies已达到30人上限！")
|
||
errorDB(use_cookies, "limit30")
|
||
continue
|
||
elif str(e).find("Login cookies have not been provided") != -1:
|
||
# return ("Cookie错误/过期，请重置Cookie")
|
||
errorDB(use_cookies, "error")
|
||
continue
|
||
else:
|
||
traceback.print_exc()
|
||
return (
|
||
"Api报错\r\n"
|
||
+ "\r\n出现这种情况可能是UID输入错误 or 不存在"
|
||
)
|
||
break
|
||
|
||
char_datas = raw_data["characters"] # noqa
|
||
|
||
l1_size = 2
|
||
l2_size = 4
|
||
l3_size = 6
|
||
l4_size = 7
|
||
l5_size = 10
|
||
|
||
word_str = {}
|
||
|
||
g3d1 = 0
|
||
ly3c = 0
|
||
star5num = 0
|
||
star5numcon = 0
|
||
|
||
for i in char_datas:
|
||
if i["name"] in ['雷电将军', '温迪', '钟离', '枫原万叶']:
|
||
g3d1 += 1
|
||
if i["name"] in ['甘雨', '魈', '胡桃']:
|
||
ly3c += 1
|
||
if i['rarity'] == 5:
|
||
star5num += 1
|
||
if i['name'] != '旅行者':
|
||
star5numcon += 1
|
||
|
||
if i["level"] >= 80:
|
||
if i['name'] == "迪卢克":
|
||
word_str["落魄了家人们"] = l3_size
|
||
if i['name'] == "刻晴":
|
||
word_str["斩尽牛杂"] = l3_size
|
||
if i['name'] == "旅行者":
|
||
word_str["旅行者真爱党"] = l3_size
|
||
|
||
game_time = time.mktime(time.strptime('20200915', '%Y%m%d'))
|
||
now_time = time.time()
|
||
total_s = now_time - game_time
|
||
total_d = (((total_s) / 60) / 60) / 24
|
||
|
||
if math.floor(total_d) - 5 <= raw_data['stats']['active_days']:
|
||
word_str["开服玩家"] = l4_size
|
||
|
||
if g3d1 >= 4:
|
||
word_str["三神一帝"] = l3_size
|
||
if ly3c >= 3:
|
||
word_str["璃月3C"] = l3_size
|
||
if star5num >= 16:
|
||
word_str["五星众多"] = l3_size
|
||
|
||
if raw_data['stats']['anemoculi'] + raw_data['stats']['geoculi'] + raw_data['stats'][
|
||
'electroculi'] == 378:
|
||
word_str["全神瞳"] = l2_size
|
||
if raw_data['explorations'][4]['explored'] + \
|
||
raw_data['explorations'][3]['explored'] + \
|
||
raw_data['explorations'][2]['explored'] + \
|
||
raw_data['explorations'][1]['explored'] + \
|
||
raw_data['explorations'][0]['explored'] >= 495:
|
||
word_str["全探索"] = l4_size
|
||
if raw_data['stats']['achievements'] >= 510:
|
||
word_str["全成就"] = l5_size
|
||
elif raw_data['stats']['achievements'] >= 490:
|
||
word_str["成就达人"] = l3_size
|
||
if raw_data['stats']['spiral_abyss'] == '12-3':
|
||
word_str["深境的探究者"] = l2_size
|
||
if len(raw_data['characters']) >= 42:
|
||
word_str["全角色"] = l3_size
|
||
|
||
if raw_data['stats']['active_days'] <= 40:
|
||
word_str["刚入坑"] = l1_size
|
||
elif raw_data['stats']['active_days'] <= 100:
|
||
word_str["初心者"] = l2_size
|
||
elif raw_data['stats']['active_days'] <= 300:
|
||
word_str["老玩家"] = l2_size
|
||
if raw_data['stats']['active_days'] >= 365 and raw_data['stats'][
|
||
'common_chests'] + raw_data['stats']['exquisite_chests'] + \
|
||
raw_data['stats']['precious_chests'] + raw_data['stats']['luxurious_chests'] <= 2500:
|
||
word_str["老咸鱼"] = l3_size
|
||
if raw_data['teapot']:
|
||
if raw_data['teapot']['comfort'] >= 25000:
|
||
word_str["团雀附体"] = l2_size
|
||
|
||
if raw_Abyss_data['stats']['total_battles'] <= 12 and raw_Abyss_data['stats']['max_floor'] == '12-3':
|
||
word_str["PVP资格证"] = l4_size
|
||
if raw_Abyss_data['character_ranks']['most_damage_taken']:
|
||
try:
|
||
if raw_Abyss_data['character_ranks']['most_damage_taken'][0]["value"] >= 150000:
|
||
word_str["这一击，贯穿星辰"] = l4_size
|
||
except:
|
||
pass
|
||
|
||
bg_list = random.choice([x for x in os.listdir(BG_PATH)
|
||
if os.path.isfile(os.path.join(BG_PATH, x))])
|
||
|
||
bg2_path = os.path.join(BG_PATH, bg_list)
|
||
|
||
based_w = 900
|
||
based_h = 1000
|
||
based_scale = '%.3f' % (based_w / based_h)
|
||
|
||
is_edit = False
|
||
if message.media:
|
||
is_edit = await message.download()
|
||
|
||
if is_edit:
|
||
bg_path_edit = is_edit
|
||
else:
|
||
bg_path_edit = bg2_path
|
||
|
||
edit_bg = Image.open(bg_path_edit)
|
||
w, h = edit_bg.size
|
||
scale_f = '%.3f' % (w / h)
|
||
new_w = math.ceil(based_h * float(scale_f))
|
||
new_h = math.ceil(based_w / float(scale_f))
|
||
if scale_f > based_scale:
|
||
bg_img2 = edit_bg.resize((new_w, based_h), Image.ANTIALIAS)
|
||
else:
|
||
bg_img2 = edit_bg.resize((based_w, new_h), Image.ANTIALIAS)
|
||
|
||
bg_img = bg_img2.crop((0, 0, based_w, based_h))
|
||
|
||
x, y = 50, 153
|
||
radius = 50
|
||
cropped_img = bg_img.crop((x, y, x + 800, y + 800))
|
||
blurred_img = cropped_img.filter(ImageFilter.GaussianBlur(5), ).convert("RGBA")
|
||
bg_img.paste(blurred_img, (x, y), create_rounded_rectangle_mask(cropped_img, radius))
|
||
|
||
panle = Image.open(os.path.join(BG2_PATH, 'wordcloud_0.png'))
|
||
|
||
mask = np.array(Image.open(os.path.join(BG2_PATH, 'wordcloudmask.png')))
|
||
|
||
wc = WordCloud(
|
||
font_path=os.path.join("assets", "fonts", "ZhuZiAWan-2.ttc"),
|
||
mask=mask,
|
||
background_color="rgba(255, 255, 255, 0)",
|
||
mode="RGBA",
|
||
max_words=200,
|
||
max_font_size=80
|
||
# color_func=multi_color_func
|
||
# color_func=similar_color_func
|
||
).generate_from_frequencies(word_str, max_font_size=100)
|
||
|
||
image_produce = wc.to_image()
|
||
|
||
bg_img.paste(panle, (0, 0), panle)
|
||
bg_img.paste(image_produce, (0, 0), image_produce)
|
||
bg_img = bg_img.convert('RGB')
|
||
|
||
text_draw = ImageDraw.Draw(bg_img)
|
||
text_draw.text((450, 105), 'UID ' + f"{uid}", (40, 136, 168), ys_font(26), anchor="mm")
|
||
|
||
bg_img.save(f"temp{os.sep}cx.jpg", format='JPEG', subsampling=0, quality=90)
|
||
if is_edit:
|
||
try:
|
||
os.remove(is_edit)
|
||
except:
|
||
pass
|
||
return f"temp{os.sep}cx.jpg"
|