Don't rule out running from the command line

4 years ago · 0e0d7d8b14
parent 981aadc497
commit 0e0d7d8b14
1 changed files with 4 additions and 1 deletions
--- a/DeDRM_plugin/kfxdedrm.py
+++ b/DeDRM_plugin/kfxdedrm.py
@ -11,7 +11,10 @@ import shutil
 import zipfile

 from io import BytesIO
-from calibre_plugins.dedrm.ion import DrmIon, DrmIonVoucher
+try:
+    from ion import DrmIon, DrmIonVoucher
+except:
+    from calibre_plugins.dedrm.ion import DrmIon, DrmIonVoucher


 __license__ = 'GPL v3'