diff --git a/devicemanagement/constants.py b/devicemanagement/constants.py index 9967548..b4c3caa 100644 --- a/devicemanagement/constants.py +++ b/devicemanagement/constants.py @@ -2,8 +2,9 @@ from enum import Enum from pymobiledevice3.lockdown import LockdownClient class Device: - def __init__(self, uuid: int, name: str, version: str, build: str, model: str, hardware: str, cpu: str, locale: str, ld: LockdownClient): + def __init__(self, uuid: int, usb: bool, name: str, version: str, build: str, model: str, hardware: str, cpu: str, locale: str, ld: LockdownClient): self.uuid = uuid + self.connected_via_usb = usb self.name = name self.version = version self.build = build diff --git a/devicemanagement/device_manager.py b/devicemanagement/device_manager.py index ca3073f..85d7f49 100644 --- a/devicemanagement/device_manager.py +++ b/devicemanagement/device_manager.py @@ -109,6 +109,7 @@ class DeviceManager: pass dev = Device( uuid=device.serial, + usb=device.is_usb, name=vals['DeviceName'], version=vals['ProductVersion'], build=vals['BuildVersion'], diff --git a/gui/main_window.py b/gui/main_window.py index 9fc883a..8695738 100644 --- a/gui/main_window.py +++ b/gui/main_window.py @@ -275,7 +275,13 @@ class MainWindow(QtWidgets.QMainWindow): self.ui.devicePicker.setEnabled(True) # populate the ComboBox with device names for device in self.device_manager.devices: - self.ui.devicePicker.addItem(device.name) + tag = "" + if self.device_manager.apply_over_wifi: + if device.connected_via_usb: + tag = " (@ USB)" + else: + tag = " (@ WiFi)" + self.ui.devicePicker.addItem(f"{device.name}{tag}") # show all pages self.ui.sidebarDiv1.show()