From 73af5d355d17ca487d939272c248f2ea0e522396 Mon Sep 17 00:00:00 2001
From: Apprentice Harper <apprenticeharper@gmail.com>
Date: Sun, 11 Apr 2021 16:43:16 +0100
Subject: [PATCH] whitespace and some unicode/bytes

Minor changes.
---
 DeDRM_plugin/__init__.py | 3 ++-
 DeDRM_plugin/ion.py      | 6 ++++++
 DeDRM_plugin/kfxdedrm.py | 7 ++++---
 Obok_plugin/__init__.py  | 4 ++--
 4 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/DeDRM_plugin/__init__.py b/DeDRM_plugin/__init__.py
index 173db21..8af84b1 100644
--- a/DeDRM_plugin/__init__.py
+++ b/DeDRM_plugin/__init__.py
@@ -5,7 +5,7 @@
 # Copyright © 2008-2020 Apprentice Harper et al.
 
 __license__   = 'GPL v3'
-__version__ = '7.2.0'
+__version__ = '7.2.1'
 __docformat__ = 'restructuredtext en'
 
 
@@ -76,6 +76,7 @@ __docformat__ = 'restructuredtext en'
 #   7.0.3 - More Python 3 changes. Integer division in ineptpdf.py
 #   7.1.0 - Full release for calibre 5.x
 #   7.2.0 - Update for latest KFX changes, and Python 3 Obok fixes.
+#   7.2.1 - Whitespace!
 
 """
 Decrypt DRMed ebooks.
diff --git a/DeDRM_plugin/ion.py b/DeDRM_plugin/ion.py
index 9e9dc0c..f102ec5 100644
--- a/DeDRM_plugin/ion.py
+++ b/DeDRM_plugin/ion.py
@@ -848,6 +848,12 @@ class DrmIonVoucher(object):
     def __init__(self, voucherenv, dsn, secret):
         self.dsn, self.secret = dsn, secret
 
+        if isinstance(dsn, str):
+            self.dsn = dsn.encode('ASCII')
+
+        if isinstance(secret, str):
+            self.secret = secret.encode('ASCII')
+
         self.lockparams = []
 
         self.envelope = BinaryIonParser(voucherenv)
diff --git a/DeDRM_plugin/kfxdedrm.py b/DeDRM_plugin/kfxdedrm.py
index d6b0e1f..67b10f8 100644
--- a/DeDRM_plugin/kfxdedrm.py
+++ b/DeDRM_plugin/kfxdedrm.py
@@ -5,6 +5,7 @@
 
 #  2.0   - Python 3 for calibre 5.0
 #  2.1   - Some fixes for debugging
+#  2.1.1 - Whitespace!
 
 
 import os
@@ -67,7 +68,7 @@ class KFXZipBook:
         print("Decrypting KFX DRM voucher: {0}".format(info.filename))
 
         for pid in [''] + totalpids:
-        	# Belt and braces. PIDs should be unicode strings, but just in case...
+            # Belt and braces. PIDs should be unicode strings, but just in case...
             if isinstance(pid, bytes):
                 pid = pid.decode('ascii')
             for dsn_len,secret_len in [(0,0), (16,0), (16,40), (32,40), (40,0), (40,40)]:
@@ -82,8 +83,8 @@ class KFXZipBook:
                 voucher.decryptvoucher()
                 break
             except:
-   				traceback.print_exc()
-                    pass
+                traceback.print_exc()
+                pass
         else:
             raise Exception("Failed to decrypt KFX DRM voucher with any key")
 
diff --git a/Obok_plugin/__init__.py b/Obok_plugin/__init__.py
index 722c088..60daad3 100644
--- a/Obok_plugin/__init__.py
+++ b/Obok_plugin/__init__.py
@@ -3,7 +3,7 @@ from __future__ import (unicode_literals, division, absolute_import,
                         print_function)
 
 __license__   = 'GPL v3'
-__version__ = '7.2.0'
+__version__ = '7.2.1'
 __docformat__ = 'restructuredtext en'
 
 #####################################################################
@@ -20,7 +20,7 @@ except NameError:
 PLUGIN_NAME = 'Obok DeDRM'
 PLUGIN_SAFE_NAME = PLUGIN_NAME.strip().lower().replace(' ', '_')
 PLUGIN_DESCRIPTION = _('Removes DRM from Kobo kepubs and adds them to the library.')
-PLUGIN_VERSION_TUPLE = (7, 2, 0)
+PLUGIN_VERSION_TUPLE = (7, 2, 1)
 PLUGIN_VERSION = '.'.join([str(x) for x in PLUGIN_VERSION_TUPLE])
 HELPFILE_NAME = PLUGIN_SAFE_NAME + '_Help.htm'
 PLUGIN_AUTHORS = 'Anon'