summaryrefslogtreecommitdiffstats
path: root/DeDRM_plugin/kindlekey.py
diff options
context:
space:
mode:
Diffstat (limited to 'DeDRM_plugin/kindlekey.py')
-rw-r--r--DeDRM_plugin/kindlekey.py9
1 files changed, 7 insertions, 2 deletions
diff --git a/DeDRM_plugin/kindlekey.py b/DeDRM_plugin/kindlekey.py
index e8a0e65..6297959 100644
--- a/DeDRM_plugin/kindlekey.py
+++ b/DeDRM_plugin/kindlekey.py
@@ -62,7 +62,12 @@ except NameError:
# Routines common to Mac and PC
-from utilities import SafeUnbuffered
+try:
+ from utilities import SafeUnbuffered
+ from argv_utils import unicode_argv
+except:
+ from . import utilities, argv_utils
+
try:
from calibre.constants import iswindows, isosx
@@ -70,7 +75,7 @@ except:
iswindows = sys.platform.startswith('win')
isosx = sys.platform.startswith('darwin')
-from argv_utils import unicode_argv
+
class DrmException(Exception):
pass