change: get adb devices in cli

This commit is contained in:
lilong.129
2025-02-07 16:20:07 +08:00
parent ae5a137353
commit b314a2024c
3 changed files with 27 additions and 33 deletions
+20 -23
View File
@@ -10,14 +10,9 @@ import (
"github.com/spf13/cobra" "github.com/spf13/cobra"
"github.com/httprunner/httprunner/v5/internal/sdk" "github.com/httprunner/httprunner/v5/internal/sdk"
"github.com/httprunner/httprunner/v5/pkg/uixt" "github.com/httprunner/httprunner/v5/pkg/gadb"
) )
func format(data map[string]string) string {
result, _ := json.MarshalIndent(data, "", "\t")
return string(result)
}
var listAndroidDevicesCmd = &cobra.Command{ var listAndroidDevicesCmd = &cobra.Command{
Use: "devices", Use: "devices",
Short: "List all Android devices", Short: "List all Android devices",
@@ -31,34 +26,36 @@ var listAndroidDevicesCmd = &cobra.Command{
}) })
}() }()
deviceList, err := uixt.GetAndroidDevices(serial) deviceList, err := getAndroidDevices()
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
os.Exit(0) os.Exit(1)
} }
for _, d := range deviceList { for _, d := range deviceList {
if isDetail { fmt.Println(format(d.DeviceInfo()))
fmt.Println(format(d.DeviceInfo()))
} else {
if usb, err := d.Usb(); err != nil {
fmt.Println(d.Serial())
} else {
fmt.Println(d.Serial(), usb)
}
}
} }
return nil return nil
}, },
} }
var ( func format(data map[string]string) string {
serial string result, _ := json.MarshalIndent(data, "", "\t")
isDetail bool return string(result)
) }
func getAndroidDevices() (devices []*gadb.Device, err error) {
var adbClient gadb.Client
if adbClient, err = gadb.NewClient(); err != nil {
return nil, err
}
if devices, err = adbClient.DeviceList(); err != nil {
return nil, err
}
return devices, nil
}
func init() { func init() {
listAndroidDevicesCmd.Flags().StringVarP(&serial, "serial", "s", "", "filter by device's serial")
listAndroidDevicesCmd.Flags().BoolVarP(&isDetail, "detail", "d", false, "print device's detail")
androidRootCmd.AddCommand(listAndroidDevicesCmd) androidRootCmd.AddCommand(listAndroidDevicesCmd)
} }
+6 -9
View File
@@ -1,29 +1,26 @@
package adb package adb
import ( import (
"fmt"
"github.com/spf13/cobra" "github.com/spf13/cobra"
"github.com/httprunner/httprunner/v5/pkg/gadb"
"github.com/httprunner/httprunner/v5/pkg/uixt" "github.com/httprunner/httprunner/v5/pkg/uixt"
"github.com/httprunner/httprunner/v5/pkg/uixt/option"
) )
var serial string
var androidRootCmd = &cobra.Command{ var androidRootCmd = &cobra.Command{
Use: "adb", Use: "adb",
Short: "simple utils for android device management", Short: "simple utils for android device management",
PersistentPreRun: func(cmd *cobra.Command, args []string) {}, PersistentPreRun: func(cmd *cobra.Command, args []string) {},
} }
func getDevice(serial string) (*gadb.Device, error) { func getDevice(serial string) (*uixt.AndroidDevice, error) {
devices, err := uixt.GetAndroidDevices(serial) device, err := uixt.NewAndroidDevice(option.WithSerialNumber(serial))
if err != nil { if err != nil {
return nil, err return nil, err
} }
if len(devices) > 1 { return device, nil
return nil, fmt.Errorf("found multiple attached devices, please specify android serial")
}
return devices[0], nil
} }
func Init(rootCmd *cobra.Command) { func Init(rootCmd *cobra.Command) {
+1 -1
View File
@@ -1 +1 @@
v5.0.0+2502071516 v5.0.0+2502071620