-
Notifications
You must be signed in to change notification settings - Fork 1
/
sign_in.py
175 lines (160 loc) · 6.63 KB
/
sign_in.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
162
163
164
165
166
167
168
169
170
171
172
173
174
175
from datetime import datetime, timedelta
from typing import Optional
import cv2
import numpy as np
from arknights_mower.utils.email import send_message
from arknights_mower.utils.graph import SceneGraphSolver
from arknights_mower.utils.image import cropimg
from arknights_mower.utils.log import logger
from arknights_mower.utils.scene import Scene
from arknights_mower.utils.vector import va
class TaskManager:
def __init__(self):
self.task_list = ["back_to_index"]
@property
def task(self):
return self.task_list[0] if self.task_list else None
def add(self, task: str, year: int, month: int, day: int):
if datetime.now() > datetime(year, month, day):
return
self.task_list.insert(-1, task)
def complete(self, task: Optional[str]):
task = task or self.task
if task in self.task_list:
self.task_list.remove(task)
brilliant_sun = "brilliant_sun"
orundum = "orundum"
headhunting = "headhunting"
class SignInSolver(SceneGraphSolver):
def run(self) -> None:
logger.info("Start: 签到活动")
self.scene_graph_navigation(Scene.INDEX)
self.tm = TaskManager()
self.tm.add(brilliant_sun, 2024, 8, 15) # 沉沙赫日签到活动
self.tm.add(orundum, 2024, 8, 15) # 大巴扎许愿墙
self.tm.add(headhunting, 2024, 8, 15) # 每日赠送单抽
self.failure = 0
self.in_progress = False
self.start_time = datetime.now()
return super().run()
def notify(self, msg, level="INFO"):
logger.info(msg)
self.recog.save_screencap("sign_in")
send_message(msg, level=level, attach_image=self.recog.img)
def handle_unknown(self):
self.failure += 1
if self.failure > 30:
self.notify("签到任务执行失败！", level="ERROR")
self.scene_graph_navigation(Scene.INDEX)
return True
self.sleep()
def transition(self) -> bool:
if datetime.now() - self.start_time > timedelta(minutes=2):
self.notify("签到任务超时！", level="ERROR")
self.scene_graph_navigation(Scene.INDEX)
return True
if not self.tm.task:
return True
if self.find("connecting"):
return self.handle_unknown()
elif self.recog.detect_index_scene():
if self.tm.task == brilliant_sun:
if pos := self.find("@hot/brilliant_sun/entry"):
self.in_progress = False
self.ctap(pos)
else:
self.notify("未检测到沉沙赫日签到活动入口！")
self.tm.complete(brilliant_sun)
elif self.tm.task == orundum:
if pos := self.find("@hot/orundum/entry"):
self.in_progress = False
self.ctap(pos)
else:
self.notify("未检测到大巴扎许愿墙活动入口！")
self.tm.complete(orundum)
elif self.tm.task == headhunting:
self.tap_index_element("headhunting")
else:
self.tm.complete("back_to_index")
elif self.find("@hot/brilliant_sun/banner"):
if self.tm.task == brilliant_sun:
top_left = 677, 333
img = cropimg(self.recog.img, (top_left, (1790, 565)))
img = cv2.cvtColor(img, cv2.COLOR_RGB2HSV)
img = cv2.inRange(img, (5, 100, 100), (15, 255, 255))
tpl = np.zeros((100, 100), dtype=np.uint8)
tpl[:] = (255,)
result = cv2.matchTemplate(img, tpl, cv2.TM_CCORR_NORMED)
min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(result)
if max_val > 0.9:
self.in_progress = True
self.ctap(va(va(max_loc, top_left), (50, 50)))
else:
if not self.in_progress:
self.notify("沉沙赫日签到奖励已领完")
self.tm.complete(brilliant_sun)
self.back()
else:
self.back()
elif self.find("@hot/orundum/banner"):
if self.tm.task == orundum:
if self.find("@hot/orundum/complete"):
if self.in_progress:
self.sleep()
else:
self.tm.complete(orundum)
self.notify("今日许愿墙奖励已领完")
self.back()
for x in range(445, 1520, 213):
if self.find("@hot/orundum/choose"):
self.tap((x, 415))
elif pos := self.find("@hot/orundum/confirm"):
self.in_progress = True
self.ctap(pos)
break
else:
self.sleep()
break
else:
self.back()
elif self.find("materiel_ico"):
self.sleep()
if self.tm.task == brilliant_sun:
self.notify("沉沙赫日活动签到成功")
elif self.tm.task == orundum:
self.notify("成功领取许愿墙奖励")
self.tm.complete(orundum)
else:
self.notify("物资领取")
self.tap((960, 960))
elif pos := self.find("pull_once"):
if self.tm.task == headhunting:
if self.find("@hot/headhunting/banner"):
if self.find("@hot/headhunting/available"):
self.tap(pos)
else:
self.notify("今日赠送单抽已抽完")
self.tm.complete(headhunting)
else:
self.notify("在流沙上刻印卡池已关闭")
self.back()
else:
self.back()
elif pos := self.find("double_confirm/main"):
if not self.find("@hot/headhunting/free"):
return self.handle_unknown()
if self.tm.task == headhunting:
self.tap(pos, x_rate=1)
else:
self.tap(pos, x_rate=0)
elif pos := self.find("skip"):
self.ctap(pos)
elif pos := self.find("@hot/headhunting/contract"):
if self.tm.task == headhunting:
self.notify("成功抽完赠送单抽")
self.tm.complete(headhunting)
self.tap((960, 540))
elif pos := self.recog.check_announcement():
self.tap(pos)
else:
return self.handle_unknown()