class ADBShell(object):
def __init__(self, adb_host=ADB_HOST):
self.SCREEN_SHOOT_SAVE_PATH = os.path.abspath(
SCREEN_SHOOT_SAVE_PATH) + "\\"
# os.chdir(ADB_ROOT)
self.ADB_ROOT = ADB_ROOT
self.ADB_HOST = adb_host
self.__buffer = ""
self.shell_color = ShellColor()
self.__adb_tools = ""
self.__adb_command = ""
self.DEVICE_NAME = self.__adb_device_name_detector()
self.__command = "\"" + self.ADB_ROOT + "\\adb.exe\" -s " + self.DEVICE_NAME + " {tools} {command} "
# 命令格式 "D:\Program Files\Nox\bin\adb.exe" -s 127.0.0.1:62001 shell am start ...
# Linux 和 Mac 机器我不太清楚咋整. 不过好像大家目前还没这个需求
# self.__adb_connect()
def __adb_device_name_detector(self):
self.__command = "\"" + self.ADB_ROOT + "\\adb.exe\" {tools} {command}"
self.__adb_tools = "devices"
content = self.run_cmd(DEBUG_LEVEL=1).strip().split("\n")
content.pop(0)
if content.__len__() == 1:
device_name = content[0].split("\t")[0]
else:
self.shell_color.helper_text("[!] 检测到多台设备，根据 ADB_HOST 参数将自动选择设备")
device_name = ""
for c in range(content.__len__()):
print("[*] " + c.__str__() + "\t" + content[c])
if self.ADB_HOST == content[c].split("\t")[0]:
device_name = self.ADB_HOST
if device_name == "":
print("自动选择设备失败，请根据上述内容自行输入数字并选择")
input_valid_flag = False
num = "0"
while not input_valid_flag:
num = input(">")
if num.isnumeric() and int(num) <= content.__len__():
input_valid_flag = True
else:
print("输入不合法，请重新输入")
device_name = content[int(num)].split("\t")[0]
self.shell_color.helper_text("[+] 确认设备名称\t" + device_name)
return device_name
def __adb_connect(self):
self.__adb_tools = "connect"
self.__adb_command = self.DEVICE_NAME
self.run_cmd(DEBUG_LEVEL=1)
if "device" in self.__buffer or "already connected to {}".format(
self.DEVICE_NAME) in self.__buffer:
self.shell_color.warning_text(
"[+] Connect to DEVICE {} Success".format(self.DEVICE_NAME))
else:
self.shell_color.failure_text(
"[-] Connect to DEVICE {} Failed".format(self.DEVICE_NAME))
def run_cmd(self, DEBUG_LEVEL=2):
"""
:param DEBUG_LEVEL:
0 : cannot get any info
1 : use get_buffer() to get info
2 : print command and can use get_buffer() to get info
3 : print command and print the return content
:return:
"""
if DEBUG_LEVEL == 3:
print(self.shell_color.H_OK_BLUE + self.__command.format(
tools=self.__adb_tools, command=self.__adb_command) +
self.shell_color.E_END)
self.__buffer = os.popen(
self.__command.format(tools=self.__adb_tools,
command=self.__adb_command)).read()
self.get_buffer()
elif DEBUG_LEVEL == 2:
print(self.shell_color.H_OK_BLUE + self.__command.format(
tools=self.__adb_tools, command=self.__adb_command) +
self.shell_color.E_END)
self.__buffer = os.popen(
self.__command.format(tools=self.__adb_tools,
command=self.__adb_command)).read()
elif DEBUG_LEVEL == 1:
self.__buffer = os.popen(
self.__command.format(tools=self.__adb_tools,
command=self.__adb_command)).read()
return self.__buffer
else:
os.system(
self.__command.format(tools=self.__adb_tools,
command=self.__adb_command))
def get_buffer(self, n=1024, BUFFER_OUT_PUT_LEVEL=1):
"""
:param n: buffer size default 1024 chars
:param BUFFER_OUT_PUT_LEVEL:
1 INFO - color blue
0 HELPER - color green
-1 WEARING - color red
:return:
"""
if BUFFER_OUT_PUT_LEVEL == 1:
print(self.shell_color.H_OK_BLUE + "[+] DEBUG INFO " +
self.shell_color.E_END + "\n" + self.__buffer[0:n])
elif BUFFER_OUT_PUT_LEVEL == -1:
print(self.shell_color.H_FAIL + "[+] DEBUG WARNING " +
self.shell_color.E_END + "\n" + self.__buffer[0:n])
elif BUFFER_OUT_PUT_LEVEL == 0:
print(self.shell_color.H_OK_GREEN + "[+] DEBUG HELPER " +
self.shell_color.E_END + "\n" + self.__buffer[0:n])
return self.__buffer[0:n]
def get_sub_screen(self, file_name, screen_range):
i = Image.open(self.SCREEN_SHOOT_SAVE_PATH + file_name)
i.crop(
(screen_range[0][0], screen_range[0][1],
screen_range[0][0] + screen_range[1][0], screen_range[0][1] +
screen_range[1][1])).save(self.SCREEN_SHOOT_SAVE_PATH + file_name)
def get_screen_shoot(self, file_name="1.png", screen_range=None):
sleep(1)
if screen_range is None:
screen_range = []
self.__adb_tools = "shell"
self.__adb_command = "/system/bin/screencap -p /sdcard/screenshot.png"
self.run_cmd(1)
self.__adb_tools = "pull"
self.__adb_command = "/sdcard/screenshot.png \"{}\"".format(
self.SCREEN_SHOOT_SAVE_PATH + file_name)
self.run_cmd(1)
self.__adb_tools = "shell"
self.__adb_command = "rm /sdcard/screen.png"
self.run_cmd(1)
# print(self.SCREEN_SHOOT_SAVE_PATH + file_name)
if screen_range.__len__() == 2:
self.get_sub_screen(file_name, screen_range)
def get_mouse_swipe(self, XY_mXmY=None, FLAG=None):
sleep(1)
XY, mXmY = XY_mXmY
self.__adb_tools = "shell"
self.__adb_command = "input swipe {X1} {Y1} {X2} {Y2}".format(
X1=XY[0], Y1=XY[1], X2=XY[0] + mXmY[0], Y2=XY[1] + mXmY[1])
self.run_cmd(DEBUG_LEVEL=0)
def get_mouse_click(self, XY=None, FLAG=None):
# sleep(10)
sleep(0.5)
self.__adb_tools = "shell"
if FLAG is not None:
# self.shell_color.info_text(XY.__str__())
# self.__adb_tools = "shell"
self.__adb_command = "input tap {} {}".format(
XY[0] + randint(-FLAG[0], FLAG[0]),
XY[1] + randint(-FLAG[1], FLAG[1]))
# self.run_cmd(DEBUG_LEVEL=0)
else:
# self.shell_color.info_text(XY.__str__())
# self.__adb_tools = "shell"
self.__adb_command = "input tap {} {}".format(
XY[0] + randint(-1, 1), XY[1] + randint(-1, 1))
# self.run_cmd(DEBUG_LEVEL=0)
# print(self.__adb_command)
# 如果你遇到了问题，可以把这百年输出并把日志分享到群里。
self.run_cmd(DEBUG_LEVEL=0)
def mv_file(self, file_name, file_path="/sdcard/", RM=False):
self.__adb_tools = "pull"
self.__adb_command = "{} {}".format(file_path + file_name,
SCREEN_SHOOT_SAVE_PATH + file_name)
self.run_cmd(DEBUG_LEVEL=0)
if RM:
self.__adb_tools = "shell"
self.__adb_command = "rm {}".format(file_path + file_name)
@staticmethod
def img_difference(img1, img2):
img1 = Image.open(img1).convert('1')
img2 = Image.open(img2).convert('1')
hist1 = list(img1.getdata())
hist2 = list(img2.getdata())
sum1 = 0
for i in range(len(hist1)):
if hist1[i] == hist2[i]:
sum1 += 1
else:
sum1 += 1 - float(abs(hist1[i] - hist2[i])) / max(
hist1[i], hist2[i])
return sum1 / len(hist1)
def ch_tools(self, tools):
self.__adb_tools = tools
def ch_abd_command(self, command):
self.__adb_command = command
class ADBShell(object):
def __init__(self):
self.resolution = [1440,810]
self.SCREEN_SHOOT_SAVE_PATH = SCREEN_SHOOT_SAVE_PATH
os.chdir(ADB_ROOT)
self.ADB_ROOT = ADB_ROOT
self.ADB_HOST = ADB_HOST
self.__command = "adb {tools} {command}"
self.__buffer = ""
self.shell_color = ShellColor()
self.__adb_tools = ""
self.__adb_command = ""
self.__adb_connect()
def __adb_connect(self):
self.__adb_tools = "connect"
self.__adb_command = self.ADB_HOST
self.run_cmd(DEBUG_LEVEL=1)
if "device" in self.__buffer or "already connected to {}".format(self.ADB_HOST) in self.__buffer:
self.shell_color.warning_text("[+] Connect to DEVICE {} Success".format(self.ADB_HOST))
else:
self.shell_color.failure_text("[-] Connect to DEVICE {} Failed".format(self.ADB_HOST))
def run_cmd(self, DEBUG_LEVEL=2):
"""
:param DEBUG_LEVEL:
0 : cannot get any info
1 : use get_buffer() to get info
2 : print command and can use get_buffer() to get info
3 : print command and print the return content
:return:
"""
if DEBUG_LEVEL == 3:
print(self.shell_color.H_OK_BLUE +
self.__command.format(
tools=self.__adb_tools,
command=self.__adb_command) + self.shell_color.E_END
)
self.__buffer = os.popen(self.__command.format(
tools=self.__adb_tools,
command=self.__adb_command
)).read()
self.get_buffer()
elif DEBUG_LEVEL == 2:
print(self.shell_color.H_OK_BLUE +
self.__command.format(
tools=self.__adb_tools,
command=self.__adb_command) + self.shell_color.E_END
)
self.__buffer = os.popen(self.__command.format(
tools=self.__adb_tools,
command=self.__adb_command
)).read()
elif DEBUG_LEVEL == 1:
self.__buffer = os.popen(self.__command.format(
tools=self.__adb_tools,
command=self.__adb_command
)).read()
else:
os.system(self.__command.format(
tools=self.__adb_tools,
command=self.__adb_command
))
def get_buffer(self, n=1024, BUFFER_OUT_PUT_LEVEL=1):
"""
:param n: buffer size default 1024 chars
:param BUFFER_OUT_PUT_LEVEL:
1 INFO - color blue
0 HELPER - color green
-1 WEARING - color red
:return:
"""
if BUFFER_OUT_PUT_LEVEL == 1:
print(self.shell_color.H_OK_BLUE + "[+] DEBUG INFO " + self.shell_color.E_END + "\n" +
self.__buffer[0:n])
elif BUFFER_OUT_PUT_LEVEL == -1:
print(self.shell_color.H_FAIL + "[+] DEBUG WARNING " + self.shell_color.E_END + "\n" +
self.__buffer[0:n])
elif BUFFER_OUT_PUT_LEVEL == 0:
print(self.shell_color.H_OK_GREEN + "[+] DEBUG HELPER " + self.shell_color.E_END + "\n" +
self.__buffer[0:n])
@staticmethod
def get_sub_screen(src_name, screen_range, save_name=""):
if save_name == "":
save_name = src_name
i = Image.open(SCREEN_SHOOT_SAVE_PATH + src_name)
i.crop(
(
screen_range[0][0],
screen_range[0][1],
screen_range[0][0] + screen_range[1][0],
screen_range[0][1] + screen_range[1][1]
)
).save(SCREEN_SHOOT_SAVE_PATH + save_name)
def get_screen_shoot(self, file_name="screenshoot.png", screen_range=None):
if screen_range is None:
screen_range = []
self.__adb_tools = "shell"
self.__adb_command = "/system/bin/screencap -p /sdcard/screenshot.png"
self.run_cmd(1)
self.__adb_tools = "pull"
self.__adb_command = "/sdcard/screenshot.png {}".format(self.SCREEN_SHOOT_SAVE_PATH + file_name)
self.run_cmd(1)
self.__adb_tools = "shell"
self.__adb_command = "rm /sdcard/screen.png"
self.run_cmd(1)
if screen_range.__len__() == 2:
self.get_sub_screen(file_name, screen_range)
def get_mouse_swipe(self, start_point, end_point, FLAG=None):
# XY = list(XY[0], XY[1])
# mXmY = list(mXmY[0], mXmY[1])
# if FLAG is not None:
# FLAG_XY, FLAG_mXmY = FLAG
# XY[0] = XY[0] + randint(-FLAG_XY[0], FLAG_XY[0])
# XY[1] = XY[0] + randint(-FLAG_XY[1], FLAG_XY[1])
# mXmY[0] = mXmY[0] + randint(-FLAG_mXmY[0], FLAG_mXmY[0])
# mXmY[1] = mXmY[1] + randint(-FLAG_mXmY[1], FLAG_mXmY[1])
self.__adb_tools = "shell"
self.__adb_command = "input swipe {X1} {Y1} {X2} {Y2}".format(
X1=start_point[0], Y1=start_point[1], X2=end_point[0], Y2=end_point[1]
)
self.run_cmd(DEBUG_LEVEL=0)
def get_mouse_click(self, XY=None, FLAG=None):
sleep(1)
if XY is None:
XY = [0, 0]
# else:
# XY = [XY[0], XY[1]]
# if FLAG is None:
# FLAG = FLAGS_CLICK_BIAS_TINY
# print(FLAG)
# XY[0] = XY[0] + randint(-FLAG[0], FLAG[0])
# XY[1] = XY[0] + randint(-FLAG[1], FLAG[1])
self.__adb_tools = "shell"
self.__adb_command = "input tap {} {}".format(XY[0], XY[1])
self.run_cmd(DEBUG_LEVEL=0)
def get_mouse_click_random(self, box=None, FLAG=None):
if box is None:
box = [[0, 0], [0, 0]]
self.__adb_tools = "shell"
self.__adb_command = "input tap {} {}".format(box[0][0]+int(box[1][0]*random()), box[0][1]+int(box[1][1]*random()))
self.run_cmd(DEBUG_LEVEL=0)
def mv_file(self, file_name, file_path="/sdcard/", RM=False):
self.__adb_tools = "pull"
self.__adb_command = "{} {}".format(
file_path + file_name,
SCREEN_SHOOT_SAVE_PATH + file_name
)
self.run_cmd(DEBUG_LEVEL=0)
if RM:
self.__adb_tools = "shell"
self.__adb_command = "rm {}".format(file_path + file_name)
@staticmethod
def img_difference(img1, img2):
img1 = Image.open(img1).convert('1')
img2 = Image.open(img2).convert('1')
hist1 = list(img1.getdata())
hist2 = list(img2.getdata())
sum1 = 0
for i in range(len(hist1)):
if (hist1[i] == hist2[i]):
sum1 += 1
else:
sum1 += 1 - float(abs(hist1[i] - hist2[i])) / max(hist1[i], hist2[i])
return sum1 / len(hist1)
def ch_tools(self, tools):
self.__adb_tools = tools
def ch_abd_command(self, command):
self.__adb_command = command
@staticmethod
def wait(min_time, prefix_time):
wait_time = min_time + prefix_time * random()
time.sleep(wait_time)
def start_app(self, packet_name):
self.__adb_tools = "shell"
self.__adb_command = "am start -n {}".format(packet_name)
self.run_cmd(DEBUG_LEVEL=0)
def stop_app(self, packet_name):
self.__adb_tools = "shell"
self.__adb_command = "am force-stop {}".format(packet_name)
self.run_cmd(DEBUG_LEVEL=0)