diff --git a/README.md b/README.md index 40e6f21d..975686b7 100644 --- a/README.md +++ b/README.md @@ -46,6 +46,16 @@ $ pip3 install -r requirements.txt ## 更新记录 +**（作者已经转用NoneBot2，Hoshino的更新可能未经测试，有bug及时提Issues！）** + +#### 2021-11-21 + +新增：`查询词云`功能 + +优化：`命座\dxx` + +修改：默认背景图 + #### 2021-11-18 修复：最新的深渊查询返回数据 @@ -54,8 +64,6 @@ $ pip3 install -r requirements.txt 调整：默认背景图 -（作者已经转用NoneBot2，Hoshino的更新可能未经测试，有bug及时提Issues！） - #### 2021-11-14 新增：NoneBot2分支（前往NoneBot分支查看） @@ -211,6 +219,7 @@ $ pip3 install -r requirements.txt | 武器 | | 查询武器信息 | 武器天空之卷 | | | 命座\d | | 查询角色命座信息 | 命座6可莉 | | | 角色 | | 查询角色简略信息 | 角色可莉 | | +| 查询词云 | | 查询绑定角色的词云 | 查询词云 | 必须绑定过UID/MYSID | ### 深渊查询： @@ -224,6 +233,10 @@ $ pip3 install -r requirements.txt ![3](https://raw.githubusercontent.com/KimigaiiWuyi/GenshinUID/main/readme/4.png) +### 查询词云： + +![3](https://raw.githubusercontent.com/KimigaiiWuyi/GenshinUID/main/readme/5.png) + ## 相关仓库 - [PaimonBot](https://github.com/XiaoMiku01/PaimonBot) - 插件原始代码来自于它 diff --git a/__init__.py b/__init__.py index 4b0dc657..e1837be5 100644 --- a/__init__.py +++ b/__init__.py @@ -1,4 +1,4 @@ -from .getImg import draw_pic,draw_abyss_pic,draw_abyss0_pic +from .getImg import draw_pic,draw_abyss_pic,draw_abyss0_pic,draw_wordcloud from .getDB import connectDB,selectDB,cookiesDB,cacheDB,deletecache,CheckDB,TransDB,OpenPush,GetMysInfo,GetDaily,GetSignList,MysSign,GetSignInfo,OpCookies,GetAward,GetWeaponInfo,GetCharInfo from nonebot import * @@ -129,10 +129,13 @@ async def _(bot:HoshinoBot, ev: CQEvent): @sv.on_prefix('命座') async def _(bot:HoshinoBot, ev: CQEvent): message = ev.message.extract_plain_text() - num = int(re.findall(r"\d+", message)[0]) # str + num = int(re.findall(r"\d+", message)[0]) m = ''.join(re.findall('[\u4e00-\u9fa5]',message)) - im = await char_wiki(m,2,num) - await bot.send(ev,im,at_sender=True) + if num<= 0 or num >6: + await bot.send(ev,"你家{}有{}命？".format(m,num),at_sender = True) + else: + im = await char_wiki(m,2,num) + await bot.send(ev,im,at_sender=True) #每日零点清空cookies使用缓存 @sv.scheduled_job('cron', hour='0') @@ -430,6 +433,12 @@ async def _(bot, ev): await bot.send(ev, im, at_sender=True) except: await bot.send(ev,'深渊输入错误！') + elif m == "词云": + try: + im = await draw_wordcloud(uid[0],image,uid[1]) + await bot.send(ev,im, at_sender=True) + except: + await bot.send(ev,'遇到错误！') elif m == "": try: bg = await draw_pic(uid[0],nickname,image,uid[1]) diff --git a/getImg.py b/getImg.py index 01495126..e43e0dd0 100644 --- a/getImg.py +++ b/getImg.py @@ -4,6 +4,7 @@ from io import BytesIO import urllib import math +from wordcloud import WordCloud from PIL import Image, ImageDraw, ImageFont, ImageFilter @@ -57,6 +58,269 @@ def get_chardone_pic(id,url,star): def get_weapon_pic(url): urllib.request.urlretrieve(url, os.path.join(WEAPON_PATH, url.split('/')[-1])) +async def draw_wordcloud(uid,image = None,mode = 2): + if mode == 3: + mys_data = await GetMysInfo(uid) + mysid_data = mys_data[1] + mys_data = mys_data[0] + for i in mys_data['data']['list']: + if i['data'][0]['name'] != '活跃天数': + mys_data['data']['list'].remove(i) + uid = mys_data['data']['list'][0]['game_role_id'] + nickname = mys_data['data']['list'][0]['nickname'] + role_level = mys_data['data']['list'][0]['level'] + raw_data = await GetInfo(uid,"cn_gf01","1",mysid_data) + raw_Abyss_data = await GetSpiralAbyssInfo(uid,"cn_gf01","1",mysid_data) + else: + raw_Abyss_data = await GetSpiralAbyssInfo(uid,"cn_gf01","1") + raw_data = await GetInfo(uid) + + if (raw_data["retcode"] != 0): + if (raw_data["retcode"] == 10001): + return ("Cookie错误/过期，请重置Cookie") + elif (raw_data["retcode"] == 10101): + return ("当前cookies已达到30人上限！") + elif (raw_data["retcode"] == 10102): + return ("当前查询id已经设置了隐私，无法查询！") + return ( + "Api报错，返回内容为：\r\n" + + str(raw_data) + "\r\n出现这种情况可能的UID输入错误 or 不存在" + ) + else: + pass + + raw_Abyss_data = raw_Abyss_data['data'] + raw_data = raw_data['data'] + + char_data = raw_data["avatars"] + char_num = len(raw_data["avatars"]) + if mode == 2: + char_ids = [] + char_rawdata = [] + + for i in char_data: + char_ids.append(i["id"]) + + char_rawdata = await GetCharacter(uid,char_ids) + char_datas = char_rawdata["data"]["avatars"] + + weapons_datas = [] + for i in char_datas: + weapons_datas.append(i['weapon']) + elif mode == 3: + char_ids = [] + char_rawdata = [] + + for i in char_data: + char_ids.append(i["id"]) + + char_rawdata = await GetCharacter(uid,char_ids,"cn_gf01",mysid_data) + char_datas = char_rawdata["data"]["avatars"] + + weapons_datas = [] + for i in char_datas: + weapons_datas.append(i['weapon']) + + l1_size = 2 + l2_size = 4 + l3_size = 6 + l4_size = 7 + l5_size = 10 + + word_str = {} + + star4weapon = 0 + star5weapon = 0 + star5weaponcon = 0 + star3weapon = 0 + for i in weapons_datas: + if i['rarity'] == 5: + star5weapon += 1 + star5weaponcon += i['affix_level'] + elif i['rarity'] == 4: + star4weapon += 1 + elif i['rarity'] == 3: + star4weapon += 1 + + g3d1 = 0 + ly3c = 0 + star5num = 0 + star5numcon = 0 + + for i in raw_data['avatars']: + if i["name"] in ['雷电将军','温迪','钟离','枫原万叶']: + g3d1 += 1 + if i["name"] in ['甘雨','魈','胡桃']: + ly3c += 1 + if i['rarity'] == 5: + star5num += 1 + if i['name'] != '旅行者': + star5numcon += 1 + i['actived_constellation_num'] + + 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 + + if i['actived_constellation_num'] == 6: + if i['rarity'] == 5: + if i['name'] == "旅行者": + word_str["满命{}".format(i['name'])] = l1_size + if i['name'] == "魈": + word_str['魈深氪的救赎'] = l5_size + if i['name'] == "甘雨": + word_str['璃月自走归终机'] = l5_size + if i['name'] == "胡桃": + word_str['一波送走全送走'] = l5_size + else: + word_str["满命{}".format(i['name'])] = l5_size + else: + word_str["满命{}".format(i['name'])] = l2_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_day_number']: + 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 len(weapons_datas) - star4weapon <= 3: + word_str["武器基本四星"] = l3_size + + if raw_data['stats']['achievement_number'] // (star5weaponcon + star5numcon) >= 23: + word_str["平民玩家"] = l2_size + elif raw_data['stats']['achievement_number'] // (star5weaponcon + star5numcon) <= 15: + word_str["氪金玩家"] = l3_size + + if raw_data['stats']['anemoculus_number'] + raw_data['stats']['geoculus_number'] + raw_data['stats']['electroculus_number'] == 378: + word_str["全神瞳"] = l2_size + if raw_data['world_explorations'][3]['exploration_percentage']+raw_data['world_explorations'][2]['exploration_percentage']+raw_data['world_explorations'][1]['exploration_percentage']+raw_data['world_explorations'][0]['exploration_percentage'] >= 3950: + word_str["全探索"] = l4_size + if raw_data['stats']['achievement_number'] >= 510: + word_str["全成就"] = l5_size + elif raw_data['stats']['achievement_number'] >= 490: + word_str["成就达人"] = l3_size + if raw_data['stats']['spiral_abyss'] == '12-3': + word_str["深境的探究者"] = l2_size + if len(raw_data['avatars']) >= 42: + word_str["全角色"] = l3_size + + if raw_data['stats']['active_day_number'] <= 40: + word_str["刚入坑"] = l1_size + elif raw_data['stats']['active_day_number'] <= 100: + word_str["初心者"] = l2_size + elif raw_data['stats']['active_day_number'] <= 300: + word_str["老玩家"] = l2_size + + if raw_data['stats']['active_day_number'] >=365 and raw_data['stats']['magic_chest_number'] + raw_data['stats']['common_chest_number']+raw_data['stats']['exquisite_chest_number']+raw_data['stats']['precious_chest_number']+raw_data['stats']['luxurious_chest_number'] <=2500: + word_str["老咸鱼"] = l3_size + if raw_data['stats']['magic_chest_number'] >= 46: + word_str["迷失在黑夜里"] = l2_size + if raw_data['homes'][0]['comfort_num'] >= 25000: + word_str["团雀附体"] = l2_size + + if raw_Abyss_data['total_battle_times'] <= 12 and raw_Abyss_data['max_floor'] == '12-3': + word_str["PVP资格证"] = l4_size + if raw_Abyss_data["damage_rank"][0]["value"] >= 150000: + word_str["这一击，贯穿星辰"] = l4_size + + 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 image: + 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 + + if is_edit == True: + bg_path_edit = os.path.join(TEXT_PATH,f"{nickname}.png") + 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(TEXT_PATH,'wordcloud_0.png')) + + mask = np.array(Image.open(os.path.join(TEXT_PATH,'wordcloudmask.png'))) + + wc = WordCloud( + font_path = os.path.join(FILE2_PATH,"yuanshen.ttf"), + 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) + + text_draw = ImageDraw.Draw(bg_img) + text_draw.text((450, 105), 'UID ' + f"{uid}", (40,136,168), ys_font(26), anchor="mm") + + result_buffer = BytesIO() + bg_img.save(result_buffer, format='JPEG', subsampling=0, quality=90) + imgmes = 'base64://' + b64encode(result_buffer.getvalue()).decode() + resultmes = f"[CQ:image,file={imgmes}]" + return resultmes + +def similar_color_func(word=None, font_size=None, + position=None, orientation=None, + font_path=None, random_state=None): + h = 40 # 0 - 360 + s = 100 # 0 - 100 + l = random_state.randint(30, 70) # 0 - 100 + return "hsl({}, {}%, {}%)".format(h, s, l) + +def multi_color_func(word=None, font_size=None, + position=None, orientation=None, + font_path=None, random_state=None): + colors = [[4, 77, 82], + [25, 74, 85], + [82, 43, 84], + [158, 48, 79]] + rand = random_state.randint(0, len(colors) - 1) + return "hsl({}, {}%, {}%)".format(colors[rand][0], colors[rand][1], colors[rand][2]) + async def draw_abyss0_pic(uid,nickname,image = None,mode = 2,date = "1"): if mode == 3: mys_data = await GetMysInfo(uid) diff --git a/mys/bg/3.jpg b/mys/bg/3.jpg new file mode 100644 index 00000000..ddf4dc4e Binary files /dev/null and b/mys/bg/3.jpg differ diff --git a/mys/bg/5.jpg b/mys/bg/5.jpg new file mode 100644 index 00000000..f8bce886 Binary files /dev/null and b/mys/bg/5.jpg differ diff --git a/mys/texture2d/wordcloud_0.png b/mys/texture2d/wordcloud_0.png new file mode 100644 index 00000000..76372ce0 Binary files /dev/null and b/mys/texture2d/wordcloud_0.png differ diff --git a/mys/texture2d/wordcloudmask.png b/mys/texture2d/wordcloudmask.png new file mode 100644 index 00000000..a264a595 Binary files /dev/null and b/mys/texture2d/wordcloudmask.png differ diff --git a/readme/5.PNG b/readme/5.PNG new file mode 100644 index 00000000..9498859f Binary files /dev/null and b/readme/5.PNG differ