Merge pull request #15 from doronz88/refactor/afc

afc: refactor
doronz88 · Apr 20, 2021 · bdcf6fe · bdcf6fe
2 parents d72e0ad + 0c479f8
commit bdcf6fe
Show file tree

Hide file tree

Showing 4 changed files with 226 additions and 153 deletions.
diff --git a/pymobiledevice3/exceptions.py b/pymobiledevice3/exceptions.py
@@ -42,9 +42,21 @@ class DeviceNonConnectedError(PyMobileDevice3Exception):
     pass
 
 
-class MuxError(PyMobileDevice3Exception):
+class MuxException(PyMobileDevice3Exception):
     pass
 
 
-class MuxVersionError(MuxError):
+class MuxVersionError(MuxException):
+    pass
+
+
+class ArgumentError(PyMobileDevice3Exception):
+    pass
+
+
+class AfcException(PyMobileDevice3Exception):
+    pass
+
+
+class AfcFileNotFoundError(AfcException):
     pass
diff --git a/pymobiledevice3/service_connection.py b/pymobiledevice3/service_connection.py
@@ -47,7 +47,7 @@ def create(udid, port):
 
         try:
             socket = mux.connect(target_device, port)
-        except usbmux.MuxError:
+        except usbmux.MuxException:
             raise ConnectionFailedError(f'Connection to device port {port} failed')
 
         return ServiceConnection(socket)