Mostly Mac fixes. mobidedrm.py now works, and k4mobidedrm for at least some input. kindlekey.py should be working too. But lots more changes and testing to do.

1 files changed, 4 insertions, 5 deletions

diff --git a/DeDRM_plugin/kfxdedrm.py b/DeDRM_plugin/kfxdedrm.py
index 875c4a1..6c1d86a 100644
--- a/DeDRM_plugin/kfxdedrm.py
+++ b/DeDRM_plugin/kfxdedrm.py
@@ -18,11 +18,6 @@ except ImportError:
     except ImportError:
         from io import StringIO
 
-try:
-    from calibre_plugins.dedrm import ion
-except ImportError:
-    import ion
-
 
 __license__ = 'GPL v3'
 __version__ = '2.0'
@@ -38,6 +33,10 @@ class KFXZipBook:
         return (None, None)
 
     def processBook(self, totalpids):
+        try:
+            import ion
+        except:
+            from calibre_plugins.dedrm import ion
         with zipfile.ZipFile(self.infile, 'r') as zf:
             for filename in zf.namelist():
                 with zf.open(filename) as fh: