whitespace and some unicode/bytes

Minor changes.
This commit is contained in:
Apprentice Harper 2021-04-11 16:43:16 +01:00
parent 45a1a64db5
commit 73af5d355d
4 changed files with 14 additions and 6 deletions

View file

@ -5,7 +5,7 @@
# Copyright © 2008-2020 Apprentice Harper et al. # Copyright © 2008-2020 Apprentice Harper et al.
__license__ = 'GPL v3' __license__ = 'GPL v3'
__version__ = '7.2.0' __version__ = '7.2.1'
__docformat__ = 'restructuredtext en' __docformat__ = 'restructuredtext en'
@ -76,6 +76,7 @@ __docformat__ = 'restructuredtext en'
# 7.0.3 - More Python 3 changes. Integer division in ineptpdf.py # 7.0.3 - More Python 3 changes. Integer division in ineptpdf.py
# 7.1.0 - Full release for calibre 5.x # 7.1.0 - Full release for calibre 5.x
# 7.2.0 - Update for latest KFX changes, and Python 3 Obok fixes. # 7.2.0 - Update for latest KFX changes, and Python 3 Obok fixes.
# 7.2.1 - Whitespace!
""" """
Decrypt DRMed ebooks. Decrypt DRMed ebooks.

View file

@ -848,6 +848,12 @@ class DrmIonVoucher(object):
def __init__(self, voucherenv, dsn, secret): def __init__(self, voucherenv, dsn, secret):
self.dsn, self.secret = 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.lockparams = []
self.envelope = BinaryIonParser(voucherenv) self.envelope = BinaryIonParser(voucherenv)

View file

@ -5,6 +5,7 @@
# 2.0 - Python 3 for calibre 5.0 # 2.0 - Python 3 for calibre 5.0
# 2.1 - Some fixes for debugging # 2.1 - Some fixes for debugging
# 2.1.1 - Whitespace!
import os import os
@ -67,7 +68,7 @@ class KFXZipBook:
print("Decrypting KFX DRM voucher: {0}".format(info.filename)) print("Decrypting KFX DRM voucher: {0}".format(info.filename))
for pid in [''] + totalpids: 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): if isinstance(pid, bytes):
pid = pid.decode('ascii') pid = pid.decode('ascii')
for dsn_len,secret_len in [(0,0), (16,0), (16,40), (32,40), (40,0), (40,40)]: 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() voucher.decryptvoucher()
break break
except: except:
traceback.print_exc() traceback.print_exc()
pass pass
else: else:
raise Exception("Failed to decrypt KFX DRM voucher with any key") raise Exception("Failed to decrypt KFX DRM voucher with any key")

View file

@ -3,7 +3,7 @@ from __future__ import (unicode_literals, division, absolute_import,
print_function) print_function)
__license__ = 'GPL v3' __license__ = 'GPL v3'
__version__ = '7.2.0' __version__ = '7.2.1'
__docformat__ = 'restructuredtext en' __docformat__ = 'restructuredtext en'
##################################################################### #####################################################################
@ -20,7 +20,7 @@ except NameError:
PLUGIN_NAME = 'Obok DeDRM' PLUGIN_NAME = 'Obok DeDRM'
PLUGIN_SAFE_NAME = PLUGIN_NAME.strip().lower().replace(' ', '_') PLUGIN_SAFE_NAME = PLUGIN_NAME.strip().lower().replace(' ', '_')
PLUGIN_DESCRIPTION = _('Removes DRM from Kobo kepubs and adds them to the library.') 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]) PLUGIN_VERSION = '.'.join([str(x) for x in PLUGIN_VERSION_TUPLE])
HELPFILE_NAME = PLUGIN_SAFE_NAME + '_Help.htm' HELPFILE_NAME = PLUGIN_SAFE_NAME + '_Help.htm'
PLUGIN_AUTHORS = 'Anon' PLUGIN_AUTHORS = 'Anon'