-
Notifications
You must be signed in to change notification settings - Fork 0
/
gacha.py
161 lines (142 loc) · 6.1 KB
/
gacha.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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
import requests, json,re
from hoshino import R,MessageSegment
from hoshino.typing import CQEvent, Message
import hoshino
from .util import Dict
from hoshino import aiorequests
from .gacha_role import init_user_info, user_info, save_user_info
from .gacha_wish import more_ten
sv=hoshino.Service('原神抽卡-惜月')
#activity = 301 限定卡池
#activity2 = 400 限定卡池2
#weapon = 302 武器卡池
#permanent = 200 常驻卡池
@sv.on_rex(r'^抽((?P<num>\d+)|(?:.*))十连(?P<pool>.*?)$')
async def gacha(bot, ev):
uid = ev.user_id
init_user_info(uid)
sd = ev.sender
num = ev['match'].group('num')
pool = ev['match'].group('pool')
if num:
if num.isdigit():
num = int(num)
if num > 5:
await bot.send(ev, '最多只能同时5十连哦', at_sender=True)
return
elif num >= 3:
await bot.send(ev, '抽卡图正在生成中，请稍候', at_sender=True)
else:
num = 1
else:
num = 1
if not pool:
pool = '角色1'
gacha_type = gacha_type_by_name(pool)
if gacha_type == 0:
await bot.finish(ev, '卡池名称出错,请输入角色1|角色2|武器|常驻', at_sender=True)
return
data = await gacha_info_list()
f = lambda x: x.gacha_type == gacha_type
gacha_data = sorted(list(filter(f, data)), key=lambda x: x.end_time)[-1]
gacha_id = gacha_data.gacha_id
#gacha_name = gacha_data.gacha_name
#gacha_type = gacha_data.gacha_type
gacha_data = await gacha_info(gacha_id)
img = await more_ten(uid,gacha_data, num, sd)
save_user_info()
await bot.send(ev, MessageSegment.image(img), at_sender=True)
@sv.on_prefix('查看抽卡记录')
async def gacharecord(bot, ev):
uid = ev.user_id
init_user_info(uid)
if user_info[uid]['gacha_list']['wish_total'] == 0:
await bot.send(ev,'你此前并没有抽过卡哦',at_sender=True)
return
msg = ev.message.extract_plain_text().strip()
if msg == '角色' or msg == '武器':
res = await getrwrecord(msg,uid)
else:
data = user_info[uid]['gacha_list']
res = '你的模拟抽卡记录如下:\n'
res += '你在本频道总共抽卡{%s}次\n其中五星共{%s}个,四星共{%s}个\n' % (user_info[uid]['gacha_list']['wish_total'],user_info[uid]['gacha_list']['wish_5'],user_info[uid]['gacha_list']['wish_4'])
try:
t5 = '{:.2f}%'.format(data['wish_5']/(data['wish_total']-data['gacha_5_role']-data['gacha_5_weapon']-data['gacha_5_permanent'])*100)
except:
t5 = '0.00%'
try:
u5 = '{:.2f}%'.format(data['wish_5_up']/data['wish_5']*100)
except:
u5 = '0.00%'
try:
t4 = '{:.2f}%'.format(data['wish_4']/(data['wish_total']-data['gacha_4_role']-data['gacha_4_weapon']-data['gacha_4_permanent'])*100)
except:
t4 = '0.00%'
try:
u4 = '{:.2f}%'.format(data['wish_4_up']/data['wish_4']*100)
except:
u4 = '0.00%'
res += '五星出货率为{%s} up率为{%s}\n四星出货率为{%s} up率为{%s}\n' %(t5,u5,t4,u4)
res += '·|角色池|·\n目前{%s}抽未出五星 {%s}抽未出四星\n下次五星是否up:{%s}\n' % (data['gacha_5_role'],data['gacha_4_role'],data['is_up_5_role'])
res += '·|武器池|·\n目前{%s}抽未出五星 {%s}抽未出四星\n下次五星是否up:{%s}\n' % (data['gacha_5_weapon'],data['gacha_4_weapon'],data['is_up_5_weapon'])
res += '·|常驻池|·\n目前{%s}抽未出五星 {%s}抽未出四星\n' % (data['gacha_5_permanent'],data['gacha_4_permanent'])
await bot.send(ev,res,at_sender=True)
async def getrwrecord(msg,uid):
if msg == '角色':
if not len(user_info[uid]['role_list']):
res = '你还没有角色'
else:
res = '你所拥有的角色如下:\n'
for role in user_info[uid]['role_list'].items():
if len(role[1]['星级']) == 5:
res += '%s%s 数量: %s 出货: %s\n' % (role[1]['星级'],role[0],role[1]['数量'],role[1]['出货'])
else:
res += '%s%s 数量: %s\n' % (role[1]['星级'],role[0],role[1]['数量'])
else:
if not len(user_info[uid]['weapon_list']):
res = '你还没有武器'
else:
res = '你所拥有的武器如下:\n'
for wp in user_info[uid]['weapon_list'].items():
if len(wp[1]['星级']) == 5:
res += '%s%s 数量: %s 出货: %s\n' % (wp[1]['星级'],wp[0],wp[1]['数量'],wp[1]['出货'])
else:
res += '%s%s 数量: %s\n' % (wp[1]['星级'],wp[0],wp[1]['数量'])
res = res.replace('[','')
res = res.replace(']','')
res = res.replace(',',' ')
return res
@sv.on_fullmatch('清空抽卡记录')
async def deleterecord(bot,ev):
uid = ev.user_id
if uid not in user_info:
await bot.send(ev,'你此前并没有抽卡记录哦',at_sender=True)
return
try:
del user_info[uid]
save_user_info()
await bot.send(ev,'你的抽卡记录清空成功',at_sender=True)
except:
await bot.send(ev,'你的抽卡记录清空失败',at_sender=True)
def gacha_type_by_name(gacha_type):
if re.match(r'^角色1|限定1(?:池)$', gacha_type):
return 301
if re.match(r'^角色2|限定2(?:池)$', gacha_type):
return 400
if re.match(r'^武器|武器池$', gacha_type):
return 302
if re.match(r'^常驻|普(?:池)$', gacha_type):
return 200
return 0
BASE_URL = 'https://webstatic.mihoyo.com/hk4e/gacha_info/cn_gf01/%s'
async def gacha_info_list():
res = await aiorequests.get(BASE_URL % 'gacha/list.json')
json_data = await res.json(object_hook=Dict)
if json_data.retcode != 0:
raise Exception(json_data.message)
return json_data.data.list
async def gacha_info(gacha_id):
res = await aiorequests.get(BASE_URL % gacha_id + '/zh-cn.json')
if res.status_code != 200:
raise Exception("error gacha_id: %s" % gacha_id)
return await res.json(object_hook=Dict)