-
Notifications
You must be signed in to change notification settings - Fork 1
/
search.py
132 lines (116 loc) · 5.53 KB
/
search.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
import json
import os
import re
from PIL import Image, ImageDraw, ImageFont
from hoshino import aiorequests, R
from prettytable import PrettyTable
# 设置当前源
async def set_source(source_name: str) -> bool:
f_data = await get_source()
source_list = dict(f_data['source_list'])
if source_name not in source_list:
return False
f_data['current'] = source_name
current_dir = os.path.join(os.path.dirname(__file__), 'data_source.json')
with open(current_dir, 'w', encoding='UTF-8') as f:
json.dump(f_data, f, indent=4, ensure_ascii=False)
return True
# 获取源
async def get_source() -> dict:
current_dir = os.path.join(os.path.dirname(__file__), 'data_source.json')
with open(current_dir, 'r', encoding='UTF-8') as af:
f_data = json.load(af)
return f_data
# 获取当前的数据源详情
async def get_source_detail(f_data: dict) -> dict:
current = f_data['current']
source_list = f_data.get('source_list', {})
return source_list.get(current, {})
# 通用头
async def get_headers(f_data: dict) -> dict:
source_detail = await get_source_detail(f_data)
headers = {
'Accept': 'application/json, text/javascript, */*; q=0.01',
'Accept-Encoding': 'gzip, deflate, br',
'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6',
'Custom-Source': 'Kyaru',
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
'AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.141 Safari/537.36',
'Content-Type': 'application/json',
'Origin': f'https://{source_detail["domain"]}',
'Referer': f'https://{source_detail["domain"]}/'
}
return headers
# 获取最新数据的时间档
async def get_current_time(server, f_data):
source_detail = await get_source_detail(f_data)
url = source_detail['api'] + '/current/getalltime/tw'
time_tmp = await aiorequests.get(url, headers=await get_headers(f_data), timeout=10)
alltime = await time_tmp.json()
all_days = alltime['data'][server].keys()
up_day = list(all_days)[-1]
up_hour = list(alltime['data'][server][up_day])[-1]
up_time = str(up_day) + str(up_hour)
return up_time
# 返回查询信息
async def get_search_rank(server: str, uptime: str, f_data: dict, search_type: str = None, search_param: str = ''):
source_detail = await get_source_detail(f_data)
url = source_detail['api'] + '/search/' + search_type
file_tmp = 'tw/' + str(server) + '/' + str(uptime)
filename_tmp = 'tw-' + str(server) + '-' + str(uptime) + '-' + str(search_param) + '.png'
params = {
'filename': file_tmp,
'search': search_param,
'page': 0,
'page_limit': 10
}
clan_score_tmp = await aiorequests.post(url, headers=await get_headers(f_data), json=params, timeout=10)
clan_score = await clan_score_tmp.json()
return clan_score, filename_tmp
# 生成图片
async def create_img(info_data, filename_tmp, is_all):
if is_all:
field_names = ('全服排名', '排名', '公会名', '人数', '会长名', '分数', '周目', 'BOSS', '剩余血量', '上期排名')
else:
field_names = ('排名', '公会名', '人数', '会长名', '分数', '周目', 'BOSS', '剩余血量', '上期排名')
table = PrettyTable(field_names=field_names)
# 输入参数
allid = info_data['data'].keys()
for id in allid:
rank = info_data['data'][str(id)]['rank']
clan_name = info_data['data'][str(id)]['clan_name']
# 去除特殊字符，只保留中英日韩数字
clan_name = re.sub(u"([^\u4e00-\u9fa5\u0030-\u0039\u0041-\u005a\u0061-\u007a\uAC00-\uD7AF\u3040-\u31FF])", "",
clan_name)
member_num = str(info_data['data'][str(id)]['member_num']).replace('.0', '')
leader_name = info_data['data'][str(id)]['leader_name']
# 去除特殊字符，只保留中英日韩数字
leader_name = re.sub(u"([^\u4e00-\u9fa5\u0030-\u0039\u0041-\u005a\u0061-\u007a\uAC00-\uD7AF\u3040-\u31FF])", "",
leader_name)
damage = info_data['data'][str(id)]['damage']
lap = info_data['data'][str(id)]['lap']
boss_id = info_data['data'][str(id)]['boss_id']
remain = info_data['data'][str(id)]['remain']
grade_rank = str(info_data['data'][str(id)]['grade_rank']).replace('.0', '')
if is_all:
all_server_rank = str(info_data['data'][str(id)]['all_server_rank'])
table.add_row(
[all_server_rank, rank, clan_name, member_num, leader_name, damage, lap, boss_id, remain, grade_rank])
else:
table.add_row([rank, clan_name, member_num, leader_name, damage, lap, boss_id, remain, grade_rank])
# 制作图片
table_info = str(table)
space = 5
current_dir = os.path.join(os.path.dirname(__file__), 'simhei.ttf')
font = ImageFont.truetype(current_dir, 20, encoding='utf-8')
im = Image.new('RGB', (10, 10), (255, 255, 255, 0))
draw = ImageDraw.Draw(im, 'RGB')
img_size = draw.multiline_textsize(table_info, font=font)
im_new = im.resize((img_size[0] + space * 2, img_size[1] + space * 2))
del draw, im
draw = ImageDraw.Draw(im_new, 'RGB')
draw.multiline_text((space, space), table_info, fill=(0, 0, 0), font=font)
save_dir = R.img('clan_rank_tw').path
path_dir = os.path.join(save_dir, filename_tmp)
im_new.save(path_dir, 'PNG')
del draw