部分python脚本支持多设备

gb112211 · gb112211 · commit 32d41d5c6c8d · 2016-07-19T13:22:53.000+08:00
diff --git a/README.md b/README.md
@@ -7,6 +7,11 @@ Android测试中常用到的脚本
 
 批量安装应用（支持以中文命名的 apk）、批量卸载、截屏、录制视频、获取当前应用的 apk 文件、包名、Activity 名等。<br>
 
+
+###2016.7.19
+增加部分python脚本对多设备的支持：
+
+![device_id](image/device_id.png)
 ###2016.04.22
 增加 `fps.py`,获取测试界面的 `fps`、`jankniess`。
 
diff --git a/image/device_id.png b/image/device_id.png
diff --git a/python/scriptUtils/utils.py b/python/scriptUtils/utils.py
@@ -11,16 +11,20 @@
 import re
 import subprocess
 import time
+import Tkinter  as tk
+import ttk
 
 import exception
 
+serialno_num = ""
+
 #判断系统类型，windows使用findstr，linux使用grep
 system = platform.system()
 if system is "Windows":
     find_util = "findstr"
 else:
     find_util = "grep"
-    
+
 #判断是否设置环境变量ANDROID_HOME
 if "ANDROID_HOME" in os.environ:
     if system == "Windows":
@@ -30,27 +34,94 @@
 else:
     raise EnvironmentError(
         "Adb not found in $ANDROID_HOME path: %s." %os.environ["ANDROID_HOME"])
-    
+
+
+def get_screen_size(window):
+    return window.winfo_screenwidth(),window.winfo_screenheight()
+
+def get_window_size(window):
+    return window.winfo_reqwidth(),window.winfo_reqheight()
+
+def center_window(root, width, height):
+    screenwidth = root.winfo_screenwidth()
+    screenheight = root.winfo_screenheight()
+    size = '%dx%d+%d+%d' % (width, height, (screenwidth - width)/2, (screenheight - height)/2)
+    root.geometry(size)
+
+class Window(object):
+    device_id = ""
+    device_id_list = []
+    root = None
+    box = None
+    def __init__(self, device_id_list, root):
+        self.device_id_list = device_id_list
+        self.device_id = device_id_list[0]
+        self.root = root
+        self.box = None
+
+    def show_window(self):
+        self.root.title(u'Serialno Number')
+        center_window(self.root, 300, 240)
+        self.root.maxsize(600, 400)
+        self.root.minsize(300, 240)
+
+        options = self.device_id_list
+        self.box = ttk.Combobox(values=options)
+        self.box.current(0)
+        self.box.pack(expand = tk.YES)
+        self.box.bind("<<ComboboxSelected>>", self.select)
+        ttk.Button(text=u"确定", command=self.ok).pack(expand = tk.YES)
+
+        self.root.mainloop()
+
+
+    def select(self, event=None):
+        self.device_id = self.box.selection_get()
+
+    def ok(self):
+        global serialno_num
+        serialno_num = self.device_id
+        self.root.destroy()
+
 #adb命令
 def adb(args):
-    cmd = "%s %s" %(command, str(args))
+    global serialno_num
+    if serialno_num == "":
+        devices = get_device_list()
+        if len(devices) == 1:
+            #global serialno_num
+            serialno_num = devices[0]
+        else:
+            root = tk.Tk()
+            window = Window(devices, root)
+            window.show_window()    
+    cmd = "%s -s %s %s" %(command, serialno_num, str(args))
     return subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
 
 #adb shell命令
 def shell(args):
-    cmd = "%s shell %s" %(command, str(args))
+    global serialno_num
+    if serialno_num == "":
+        devices = get_device_list()
+        if len(devices) == 1:
+            serialno_num = devices[0]
+        else:
+            root = tk.Tk()
+            window = Window(devices, root)
+            window.show_window()   
+    cmd = "%s -s %s shell %s" %(command, serialno_num, str(args))
     return subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
 
 #获取设备状态
 def get_state():
-    return os.popen("adb get-state").read().strip()
+    return os.popen("adb -s %s get-state" %serialno_num).read().strip()
 
 #获取对应包名的pid
-def get_app_pid(pkg_name):   
+def get_app_pid(pkg_name):
     if system is "Windows":
         string = shell("ps | findstr %s$" %pkg_name).stdout.read()
-    else:
-        string = shell("ps | grep -w %s" %pkg_name).stdout.read()
+
+    string = shell("ps | grep -w %s" %pkg_name).stdout.read()
 
     if string == '':
         return "the process doesn't exist."
@@ -64,13 +135,13 @@ def get_app_pid(pkg_name):
 #杀掉对应包名的进程
 def kill_process(pkg_name):
     pid = get_app_pid(pkg_name)
-    
+
     result = shell("kill %s" %str(pid)).stdout.read().split(": ")[-1]
-    
+
     if result != "":
         raise exception.SriptException("Operation not permitted or No such process")
 
-#获取设备上当前应用的包名与activity    
+#获取设备上当前应用的包名与activity
 def get_focused_package_and_activity():
     pattern = re.compile(r"[a-zA-Z0-9\.]+/.[a-zA-Z0-9\.]+")
     out = shell("dumpsys window w | %s \/ | %s name=" %(find_util, find_util)).stdout.read()
@@ -89,6 +160,18 @@ def get_current_activity():
 def timestamp():
     return time.strftime('%Y-%m-%d-%H-%M-%S',time.localtime(time.time()))
 
+def get_device_list():
+    devices = []
+    result = subprocess.Popen("adb devices", shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.readlines()
+    result.reverse()
+    for line in result[1:]:
+        if "attached" not in line.strip():
+            devices.append(line.split()[0])
+        else:
+            break
+    return devices
+    
+
 #连接设备
 # adb("kill-server").wait()
 # adb("start-server").wait()
@@ -97,10 +180,10 @@ def timestamp():
 if get_state() != "device":
     adb("kill-server").wait()
     adb("start-server").wait()
-    
+
 if get_state() != "device":
     raise exception.SriptException("Device not run")
-    
+
 
 if __name__ == "__main__":
     pass
