# coding: utf-8#
import uiautomator2 as u2
import time
from tqdm import tqdm
from loguru import logger
from ocr_test import OCR_UI2
class ClickTimeOutException(Exception):
def __init__(self, message="This is a custom exception"):
self.message = message
super().__init__(self.message)
def xpath是否存在(xpath,timeout=30):
for index in range(0,timeout):
wallet_ele = d.xpath(xpath).exists
if(wallet_ele):
logger.success(f"xpath已找到:{xpath}")
return True
else:
logger.debug(f"xpath未找到 {xpath} 等待:{index}")
time.sleep(1)
raise ClickTimeOutException(f"xpath等待超时:找不到 {xpath} ")
def click_5s_desc(xpath,timeout=20):
for index in range(0,timeout):
wallet_ele = d(description=xpath).exists
if(wallet_ele):
d(description=xpath).click()
logger.success(f"{xpath} --- desc 成功点击")
return True
else:
logger.debug(f"desc 未找到 {xpath} 等待:{index}")
time.sleep(1)
raise ClickTimeOutException(f"{xpath} --- desc 点击失败")
def OCR_click(xpath,timeout=20):
for index in range(0,timeout):
find = ocr.是否存在文本(xpath)
if(find):
pos_x,pos_y = ocr.查询文本坐标百分比(xpath)
d.click(pos_x,pos_y)
logger.success(f"{xpath} --- OCR 成功点击")
return True
else:
logger.debug(f"OCR 未找到 {xpath} 等待:{index}")
time.sleep(1)
raise ClickTimeOutException(f"{xpath} --- OCR 点击失败")
def 领取nft():
click_5s_desc("首页",20)
click_5s_desc("赚币中心")
time.sleep(2)
d(scrollable=True).scroll.toEnd()
time.sleep(2)
d.click(0.806, 0.413)
d(scrollable=True).scroll.toEnd()
d.xpath('//*[contains(@content-desc, "SecondLive")]//android.view.View[1]').click()
d(description="以后再说").click_exists()
claim = xpath是否存在('//*[@content-desc="SecondLive"]')
if claim:
time.sleep(1)
OCR_click("验证")
find = ocr.是否存在文本(d,"交互已完成")
if find:
return
time.sleep(5)
def click_5s(xpath,timeout=10,wait=0):
for index in range(0,timeout):
wallet_ele = d.xpath(xpath).exists
if(wallet_ele):
time.sleep(wait)
d.xpath(xpath).click()
logger.success(f"{xpath} --- 成功点击")
return True
else:
logger.debug(f"xpath未找到 {xpath} 等待:{index}")
time.sleep(1)
raise ClickTimeOutException(f"{xpath} --- 点击失败")
def 添加私钥(privatekey):
d.app_start("com.bitkeep.wallet", use_monkey=True)
d(description="暂不开启").click_exists()
click_5s('//*[contains(@content-desc, "钱") and contains(@content-desc, "包") and contains(@content-desc, "我")]/android.widget.ImageView[1]',20,2)
click_5s_desc("添加钱包")
click_5s_desc("助记词/私钥钱包\n支持12位、24位助记词导入或私钥导入")
for index in range(0, 6):
d(description="0").click()
click_5s('//android.widget.EditText')
time.sleep(2)
d.send_keys(privatekey, True)
#d.set_fastinput_ime(False)
click_5s_desc("确认导入")
#d(scrollable=True).scroll.to(description="Polygon")
click_5s('//*[@content-desc="Polygon"]')
click_5s_desc("确认")
click_5s_desc("开始探索")
return True
if __name__ == '__main__':
d = u2.connect("22b4097f")
ocr = OCR_UI2(d)
d.set_fastinput_ime(True)
print(d.info)
acc_list =[]
with open("address.txt","r") as f:
lines = f.readlines()[691:]
bar = tqdm(total = len(lines))
for line in lines:
line = line.replace("\n","")
addr,key = line.split("----")
acc = {
"private_key": key,
"address": addr
}
acc_list.append(acc)
for acc in acc_list:
try:
add_status = 添加私钥(acc['private_key'])
if add_status:
领取nft()
logger.success(f"{acc['address']} 领取nft成功!!!")
d.app_stop("com.bitkeep.wallet")
bar.update(1)
except BaseException as e:
logger.error(f"{acc['address']} 领取nft失败!!! ---- {e}")
finally:
try:
d.app_stop("com.bitkeep.wallet")
except:
pass
# 检查元素是否可点击
# coding: utf-8#
import uiautomator2 as u2
import time
from loguru import logger
from cnocr import CnOcr
class OCR_UI2():
def __init__(self,device):
self.device = device
self.ocr = CnOcr()
def 计算文本中心点坐标(self,points):
if len(points) == 0:
return None
num_points = len(points)
sum_x = sum(point[0] for point in points)
sum_y = sum(point[1] for point in points)
centroid_x = sum_x / num_points
centroid_y = sum_y / num_points
return (centroid_x, centroid_y)
# Example usage
def 坐标转百分比(self,x,y,image):
width = image.width
hight = image.height
pos_w = "{:.4f}".format(x/width)
pos_y = "{:.4f}".format(y/hight)
return (pos_w,pos_y)
def 是否存在文本(self,find_str):
time.sleep(1)
find = False
image = self.device.screenshot()
outs = self.ocr.ocr(image)
texts = [out['text'] for out in outs]
for text in texts:
if find_str in text:
logger.success(f"OCR查找成功:{text} --- 预期:{find_str}")
find = True
return find
def 查询文本坐标百分比(self,find_str):
image = self.device.screenshot()
outs = self.ocr.ocr(image)
for line in outs:
if find_str in line['text']:
(x,y) = self.计算文本中心点坐标(line['position'])
(pos_w,pos_y) = self.坐标转百分比(x,y,image)
logger.success(f"查询坐标百分比成功:{ line['text']} --- 预期:{find_str} ---坐标:{(x,y)} --- 坐标百分比:{(pos_w,pos_y)}")
return (float(pos_w),float(pos_y))
logger.debug(f"查询坐标百分比失败:预期:{find_str}")
return False
def 查询文本坐标(self,find_str):
image = self.device.screenshot()
outs = self.ocr.ocr(image)
for line in outs:
if find_str in line['text']:
(x,y) = self.计算文本中心点坐标(line['position'])
logger.success(f"查询坐标百分比成功:{ line['text']} --- 预期:{find_str} ---坐标:{(x,y)}")
return (int(x),int(y))
logger.debug(f"查询坐标失败:预期:{find_str}")
return False
if __name__ == '__main__':
d = u2.connect("5f83933a")
ocr = OCR_UI2(d)
ocr.查询文本坐标百分比("回家")
