From 7f4e6698efac2f57b660fcc07fc128989a0377cf Mon Sep 17 00:00:00 2001
From: Aldo Bleeker <mail@ableeker.demon.nl>
Date: Fri, 6 Nov 2020 23:49:18 +0100
Subject: [PATCH] More Python 3 fixes for Customize plugin dialog

---
 DeDRM_plugin/config.py   | 29 ++++++++++++++++-------------
 DeDRM_plugin/erdr2pml.py |  8 ++++----
 2 files changed, 20 insertions(+), 17 deletions(-)
 mode change 100644 => 100755 DeDRM_plugin/config.py
 mode change 100644 => 100755 DeDRM_plugin/erdr2pml.py

diff --git a/DeDRM_plugin/config.py b/DeDRM_plugin/config.py
old mode 100644
new mode 100755
index c319a6e..c81b615
--- a/DeDRM_plugin/config.py
+++ b/DeDRM_plugin/config.py
@@ -6,7 +6,7 @@ __license__ = 'GPL v3'
 # Python 3, September 2020
 
 # Standard Python modules.
-import os, traceback, json
+import os, traceback, json, binascii
 
 from PyQt5.Qt import (Qt, QWidget, QHBoxLayout, QVBoxLayout, QLabel, QLineEdit,
                       QGroupBox, QPushButton, QListWidget, QListWidgetItem,
@@ -378,7 +378,7 @@ class ManageKeysDialog(QDialog):
                 with open(fpath,'rb') as keyfile:
                     new_key_value = keyfile.read()
                 if self.binary_file:
-                    new_key_value = new_key_value.encode('hex')
+                    new_key_value = binascii.b2a_hex(new_key_value)
                 elif self.json_file:
                     new_key_value = json.loads(new_key_value)
                 elif self.android_file:
@@ -431,17 +431,20 @@ class ManageKeysDialog(QDialog):
         defaultname = "{0}.{1}".format(keyname, self.keyfile_ext)
         filename = choose_save_file(self, unique_dlg_name,  caption, filters, all_files=False, initial_filename=defaultname)
         if filename:
-            with open(filename, 'w') as fname:
-                if self.binary_file:
-                    fname.write(self.plugin_keys[keyname].decode('hex'))
-                elif self.json_file:
+            if self.binary_file:
+                with open(filename, 'wb') as fname:
+                    fname.write(binascii.a2b_hex(self.plugin_keys[keyname]))
+            elif self.json_file:
+                with open(filename, 'w') as fname:
                     fname.write(json.dumps(self.plugin_keys[keyname]))
-                elif self.android_file:
+            elif self.android_file:
+                with open(filename, 'w') as fname:
                     for key in self.plugin_keys[keyname]:
-                        fname.write(key)
-                        fname.write("\n")
-                else:
-                    fname.write(self.plugin_keys[keyname])
+                        fname.write(key.decode('utf-8'))
+                        fname.write('\n')
+            else:
+                with open(filename, 'w') as fname:
+                    fname.write(self.plugin_keys[keyname].decode('utf-8'))
 
 
 
@@ -670,7 +673,7 @@ class AddEReaderDialog(QDialog):
     @property
     def key_value(self):
         from calibre_plugins.dedrm.erdr2pml import getuser_key as generate_ereader_key
-        return generate_ereader_key(self.user_name,self.cc_number).encode('hex')
+        return binascii.b2a_hex(generate_ereader_key(self.user_name, self.cc_number))
 
     @property
     def user_name(self):
@@ -752,7 +755,7 @@ class AddAdeptDialog(QDialog):
 
     @property
     def key_value(self):
-        return self.default_key.encode('hex')
+        return binascii.b2a_hex(self.default_key)
 
 
     def accept(self):
diff --git a/DeDRM_plugin/erdr2pml.py b/DeDRM_plugin/erdr2pml.py
old mode 100644
new mode 100755
index 6c65ac2..6e1054e
--- a/DeDRM_plugin/erdr2pml.py
+++ b/DeDRM_plugin/erdr2pml.py
@@ -542,10 +542,10 @@ def usage():
     print("  It's enough to enter the last 8 digits of the credit card number")
     return
 
-def getuser_key(name,cc):
+def getuser_key(name, cc):
     newname = "".join(c for c in name.lower() if c >= 'a' and c <= 'z' or c >= '0' and c <= '9')
     cc = cc.replace(" ","")
-    return struct.pack('>LL', binascii.crc32(newname) & 0xffffffff,binascii.crc32(cc[-8:])& 0xffffffff)
+    return struct.pack('>LL', binascii.crc32(bytes(newname.encode('utf-8'))) & 0xffffffff, binascii.crc32(bytes(cc[-8:].encode('utf-8'))) & 0xffffffff)
 
 def cli_main():
     print("eRdr2Pml v{0}. Copyright © 2009–2020 The Dark Reverser et al.".format(__version__))
@@ -580,9 +580,9 @@ def cli_main():
     elif len(args)==4:
         infile, outpath, name, cc = args
 
-    print(getuser_key(name,cc).encode('hex'))
+    print(bin2ascii.b2a_hex(getuser_key(name, cc)))
 
-    return decryptBook(infile, outpath, make_pmlz, getuser_key(name,cc))
+    return decryptBook(infile, outpath, make_pmlz, bin2ascii.b2a_hex(getuser_key(name, cc)))
 
 
 if __name__ == "__main__":