随便存一下
# -*- encoding=utf8 -*-
__author__ = "GAME"
from airtest.utils.logger import get_logger
import logging
logger=get_logger("airtest")
logger.setLevel(logging.ERROR)
import sys
import json
from airtest.core.api import *
from airtest.cli.parser import cli_setup
from airtest.aircv import *
from PIL import Image
from ocrmode import *
# 初始化连接adb
auto_setup(__file__, logdir=False, project_root="D:/hzw_haidao",
devices=["Android:///"])
#dev = connect_device("Android:///127.0.0.1:5554?cap_method=javacap&touch_method=adb")
dev = connect_device("Android:///")
# 棋盘
num_xy = {"1": [815, 322], "3": [815, 522], "5": [683, 422], "7": [551, 322], "9": [551, 522], "10": [538, 226]}
# 待选区
'''
daixuan_xy = {"1": {"zuobiao": [175, 835], "name": "null"}, "2": {"zuobiao": [392, 835], "name": "null"},
"3": {"zuobiao": [623, 835], "name": "null"}, "4": {"zuobiao": [850, 835], "name": "null"},
"5": {"zuobiao": [1075, 835], "name": "null"}, "6": {"zuobiao": [1300, 835], "name": "null"},
"7": {"zuobiao": [1527, 835], "name": "null"}, "8": {"zuobiao": [1751, 835], "name": "null"}}
'''
#备战待选区
'''
daixuan_xy = {"1": {"zuobiao": [354, 835], "name": "null"}, "2": {"zuobiao": [528, 796], "name": "null"},
"3": {"zuobiao": [699, 835], "name": "null"}, "4": {"zuobiao": [878, 835], "name": "null"},
"5": {"zuobiao": [1046, 835], "name": "null"}, "6": {"zuobiao": [1222, 835], "name": "null"},
"7": {"zuobiao": [1392, 835], "name": "null"}, "8": {"zuobiao": [1569, 835], "name": "null"}}
'''
daixuan_xy = {"1": {"zuobiao": [353, 808], "name": "null"}, "2": {"zuobiao": [528, 808], "name": "null"},
"3": {"zuobiao": [703, 808], "name": "null"}, "4": {"zuobiao": [878, 808], "name": "null"},
"5": {"zuobiao": [1053, 808], "name": "null"}, "6": {"zuobiao": [1228, 808], "name": "null"},
"7": {"zuobiao": [1403, 808], "name": "null"}, "8": {"zuobiao": [1578, 808], "name": "null"}}
#待选区有人数据
dxyr_data=[]
#待选区无人数据
daixuan_data=["1","2","3","4","5","6","7","8"]
#dev.shell("input swipe %s %s %s %s 1000" % (1727,587, daixuan_xy["8"]["zuobiao"][0],835))
# 人物信息
rwinfo = {'阿金': {"weizhi": "1", "zuobiao": 0, "level": 2,"num":0,"maxnum":3}, '克里克': {"weizhi": "5", "zuobiao": 0, "level": 2,"num":0,"maxnum":3},
'强高': {"weizhi": "7", "zuobiao": 0, "level": 2,"num":0,"maxnum":3}, '帕鲁': {"weizhi": "3", "zuobiao": 0, "level": 2,"num":0,"maxnum":3}}
for key in list(rwinfo.keys()):
rwinfo[key]["zuobiao"] = num_xy.get(str(rwinfo[key]["weizhi"]))
# 现在的阵容情况
xianzaizr = {"1": "null", "3": "null", "5": "null", "7": "null", "9": "null"}
# 待选区信息
#daixuanlist = {"1": "null", "2": "null", "3": "null", "4": "null", "5": "null", "6": "null", "7": "null", "8": "null"}
#拖动
def tdxy(x_0,y_0,x_1,y_1,yanchi=300):
dev.shell("input swipe %s %s %s %s %s"%(x_0,y_0,x_1,y_1,yanchi))
# 区域截图类
def core_jt(x_0, y_0, x_1, y_1, filepath):
screen = G.DEVICE.snapshot()
screen = aircv.crop_image(screen, (x_0, y_0, x_1, y_1))
pil_image = cv2_2_pil(screen)
pil_image.save(filepath, quality=99, optimize=True)
return screen
# pytesseract识别数字
def tess_num(picture):
# 二值化后识别会更精确
img = Image.open(picture)
Img = img.convert('L')
threshold = 110
table = []
for i in range(256):
if i < threshold:
table.append(0)
else:
table.append(1)
photo = Img.point(table, '1')
photo.save(picture)
textImage = Image.open("test2.jpg")
text = pytesseract.image_to_string(textImage, lang='wnum',
config='--psm 10 --oem 2 -c tessedit_char_whitelist=0123456789')
return text
# 排队检查
def paidui():
while True:
if exists(Template(r"./picture/paidui/tpl1617260644200.png", record_pos=(0.022, -0.122),
resolution=(1920, 1080))):
print("wait....")
sleep(5.0)
else:
print("gameserver ok")
return 0
break
# 日常任务
def daily_task():
touch(Template(r"./picture/richang/tpl1617256338204.png", record_pos=(0.46, 0.239), resolution=(1920, 1080)))
# 查找日常任务图标
while True:
if exists(Template(r'./picture/richang/richang.png', record_pos=(0.125, 0.06), resolution=(1920, 1080))):
touch(Template(r'./picture/richang/richang.png', record_pos=(0.125, 0.06), resolution=(1920, 1080)))
while True:
try:
touch(Template(r"./picture/richang/tpl1617258247226.png", record_pos=(0.01, 0.183),
resolution=(1920, 1080)))
break
except:
touch(Template(r"./picture/richang/richang.png", record_pos=(0.125, 0.06), resolution=(1920, 1080)))
touch(Template(r"./picture/richang/tpl1617259072269.png", record_pos=(0.21, 0.12), resolution=(1920, 1080)))
break
else:
touch((1542, 518))
# 海岛购买人物类
def hd_hero(gm_num):
if exists(Template(r"tpl1617636327570.png", record_pos=(0.004, 0.248), resolution=(1920, 1080))):
print("开始购买人物")
heroname = ["克里克", "强高", "阿金", "帕鲁"]
name_x = [167, 627]
name_x1 = [355, 675]
# name_x=[174,628]
# name_x1=[381,670]
goumaicishu=1
break_flag = False
#循环购买两次，根据break_flag判断
while break_flag == False:
#if goumaicishu == 1:
#touch((1807, 997))
if goumaicishu>=gm_num:
break_flag = True
if goumaicishu >= 2:
touch((90, 983))
name_x = [167, 627]
name_x1 = [355, 675]
print("第%s次购买"%goumaicishu)
for i in range(0, 5):
core_jt(name_x[0], name_x[1], name_x1[0], name_x1[1], "score%s.png" % i)
# 识别人物名字
name_ocr = cnocr_str("score%s.png" % i)
# 判断是否有人物在商店
if name_ocr != 1:
if name_ocr in heroname:
for dx_id in daixuan_data:
if daixuan_xy[dx_id]["name"] != u"null":
#有人就从无人数据列表删除
daixuan_data.remove(dx_id)
print(dx_id, u"，这个位置有人了")
else:
#如果已经买了三次这个英雄，就不进行操作
if rwinfo[name_ocr]["num"]>=rwinfo[name_ocr]["maxnum"]:
print("买过三次了，等待升星")
else:
zba = daixuan_xy[dx_id]["zuobiao"]
print("拖动%s到待选区第%s格"%(name_ocr,dx_id))
#print("move %s,%s to %s,%s" % (name_x[0] + 80, name_x[1] - 40, zba[0], zba[1]))
xy_list = rwinfo[name_ocr]["zuobiao"]
tdxy(name_x[0] + 97, name_x[1] - 40, zba[0], zba[1])
sleep(0.3)
rwinfo[name_ocr]["num"]=rwinfo[name_ocr]["num"]+1 #购买英雄后计数，为升星做准备
dxyr_data.append(dx_id)#更新数据到有人列表
daixuan_xy[dx_id]["name"] = name_ocr # 更新阵容信息
print(daixuan_xy)
touch([1814, 1007])
daixuan_data.append(dx_id)
break
print("待选有人列表，%s"%dxyr_data)
else:
print("******不购买", name_ocr)
else:
print("******识别失败")
name_x[0] = name_x[0] + 370
name_x1[0] = name_x1[0] + 345
goumaicishu=goumaicishu+1
else:
print("不在商店")
# 拖动上阵
def hd_tuodong():
print("start")
touch((917, 913))
print("得到有人待选，%s"%dxyr_data)
for i in dxyr_data:
hero_name=daixuan_xy[i]["name"]
if hero_name!= "null":
hero_xy=daixuan_xy[i]["zuobiao"]# 人物待选坐标
weizhi = rwinfo[hero_name]["weizhi"] # 人物棋盘摆放位置
xy_list = rwinfo[hero_name]["zuobiao"] #棋盘对应坐标
zr_info = xianzaizr[str(weizhi)] #获取棋盘当前人物名字
if rwinfo[hero_name]["num"]==3:
touch((hero_xy[0],hero_xy[1]+30))
print("将%s合成二星"%hero_name)
for name_id in daixuan_xy:
if daixuan_xy[name_id]["name"]==hero_name:
print(daixuan_xy[name_id]["name"])
daixuan_xy[i]["name"] = "null"#更新待选位置人物名
dxyr_data.remove(i)#从有人列表删除
daixuan_data.append(i)#位置空了，添加到无人列表
else:
print("[获取待选栏人物]:%s"%hero_name)
# 如果棋盘位置是空的，就直接上阵
if zr_info == "null":
print("%s上阵到棋盘第%s个位置"%(hero_name,weizhi))
tdxy(hero_xy[0], hero_xy[1], xy_list[0], xy_list[1]) # 开始拖动
xianzaizr[weizhi] = hero_name # 更新新人物格子列表信息
daixuan_xy[i]["name"] = "null"#更新待选位置人物名
dxyr_data.remove(i)#从有人列表删除
daixuan_data.append(i)#位置空了，添加到无人列表
print("当前阵容%s"%xianzaizr)
else:
#优先级判断
level_1 = rwinfo[hero_name]["level"] # 待选栏优先级
level_2 = rwinfo[zr_info]["level"] # 当前阵容位置优先级
if level_1 < level_2: # 如果当前选择英雄优先级比阵上现有的小，开始拖动
print("%s的优先级比%s小，开始替换"%(hero_name,zr_info))
xy_list = rwinfo[name_ocr]["zuobiao"] # 坐标
tdxy(hero_xy[0], hero_xy[1], xy_list[0], xy_list[1]) # 开始拖动
daixuan_xy[i]["name"] = zr_info#更新待选位置人物名
dxyr_data.remove(i)#从有人列表删除
daixuan_data.append(i)#位置空了，添加到无人列表
#如果优先级相同
if level_1==level_2:
print("优先级相同，继续对比人物数量")
#如果待选栏人物数量比在阵上人物的数量多，就下阵棋盘人物
if rwinfo[hero_name]["num"]>=rwinfo[zr_info]["num"]:
print("把%s替换为%s"%(rwinfo[zr_info]["num"],rwinfo[hero_name]["num"]))
tdxy(hero_xy[0], hero_xy[1], xy_list[0], xy_list[1]) # 开始拖动
daixuan_xy[i]["name"] = zr_info#更新待选位置人物名
dxyr_data.remove(i)#从有人列表删除
daixuan_data.append(i)#位置空了，添加到无人列表
else:
# 如果优先级低，就卖掉人物
#如果还没买够升星需要的人物数量
touch((hero_xy[0], hero_xy[1]))#点击原待选位置获取人物信息界面
touch((1467, 928))#点击出售
daixuan_xy[i]["name"] = "null"#更新待选位置人物名
dxyr_data.remove(i)#从有人列表删除
daixuan_data.append(i)#位置空了，添加到无人列表
print("mai")
else:
# 如果优先级低，就卖掉人物
#如果还没买够升星需要的人物数量
touch((hero_xy[0], hero_xy[1]))#点击原待选位置获取人物信息界面
touch((1467, 928))#点击出售
daixuan_xy[i]["name"] = "null"#更新待选位置人物名
dxyr_data.remove(i)#从有人列表删除
daixuan_data.append(i)#位置空了，添加到无人列表
print("mai")
else:
print("******识别失败".encode("GB18030"))
sleep(10)
def haidao():
while True:
core_jt(931,10,995,55,"huihe.png")
huihe_list=cnocr_num("huihe.png")
if huihe_list:
print(huihe_list)
print("开始购买并上阵")
hd_hero(2)
hd_tuodong()
'''
if exists():
hd_hero()
# 如果不在战斗期间，获取回合数，然后升级
else:
# 如果回合数到了4回合以上就开始升级
if hd_huihe() == 4:
touch((1000, 968))
sleep(0.5)
touch((100, 1000), times=1)
if hd_huihe() == 6:
touch((1000, 968))
sleep(0.5)
touch((100, 1000), times=3)
hd_huihe()
'''
# like_num = find_template(screen_01,"duibi.png",-1)['confidence']
# print like_num
# hd_tuodong()
haidao()
#dev.shell("input swipe %s %s %s %s 1000" % (1727,587, 570,835))
# hd_hero()
# tdxy(724,379,698,652)#开始拖动
# tdxy(353,799,531,781)#开始拖动
# hd_rs()
# daily_task()
#haidao()
#获取金币和回合
#core_jt(328,45,398,80,"shangzhen.png")
#core_jt(1528,35,1596,75,"jinbi.png")
#print(cnocr_num("shangzhen.png"))
#print(cnocr_num("jinbi.png"))
文字识别模块
# -*- encoding=utf8 -*-
import mxnet as mx
from cnocr import CnOcr
from PIL import Image
def is_number(s):
try:
float(s)
return True
except ValueError:
pass
try:
import unicodedata
unicodedata.numeric(s)
return True
except (TypeError, ValueError):
pass
return False
def cnocr_str(picture):
ocr = CnOcr(name=picture)
img_fp = picture
img = mx.image.imread(img_fp, 1)
res = ocr.ocr(img)
name=""
try:
for i in res[0]:
name=name+i
return name
except:
return 1
def cnocr_num(picture):
img = Image.open(picture)
Img = img.convert('L')
threshold = 110
table = []
for i in range(256):
if i < threshold:
table.append(0)
else:
table.append(1)
photo = Img.point(table, '1')
photo.save(picture)
ocr = CnOcr(name=picture,cand_alphabet="0,1,2,3,4,5,6,7,8,9")
img_fp = picture
img = mx.image.imread(img_fp, 1)
res = ocr.ocr(img)
num_list=[]
try:
for i in res[0]:
if is_number(i):
num_list.append(i)
return num_list
except:
return []
浙公网安备 33010602011771号