JMR-devandClaude Opus 4.8
838700fbec
The plugins target calibre 5/6 (Python 3) and the Poetry environment
pins Python >=3.8, so the Python 2 fallbacks are dead code. This removes
them throughout both plugins (behaviour on Python 3 is unchanged):
- Drop all `from __future__` imports.
- Collapse `if sys.version_info[0] == 2: ... else: ...` blocks to their
Python 3 branch (ineptpdf, mobidedrm, kindlekey, kgenpids, alfcrypto,
erdr2pml, ineptepub, obok, and the various unpad() helpers, etc.).
- Replace `_winreg` import fallbacks with plain `import winreg`, and
delete the py2-only adobekey_winreg_unicode module (now unreferenced).
- Drop py2 name shims: `unicode`/`unichr`, `.iteritems()`,
`from StringIO import StringIO`, `htmlentitydefs` fallback, and the
Windows CommandLineToArgvW dance in unicode_argv (py3 sys.argv is
already Unicode on every platform).
- Remove the "Calibre < 5" (py2) bugfix block from the compat header.
Verified: every .py file in both plugins still byte-compiles.
Scope: the maintained DeDRM_plugin and Obok_plugin only. The archival
standalone scripts under Other_Tools/ are left as historical snapshots.
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
72 lines
2.8 KiB
Python
72 lines
2.8 KiB
Python
# vim:fileencoding=UTF-8:ts=4:sw=4:sta:et:sts=4:ai
|
|
|
|
__license__ = 'GPL v3'
|
|
__docformat__ = 'restructuredtext en'
|
|
|
|
import os, sys
|
|
import binascii, hashlib, re, string
|
|
|
|
class legacy_obok(object):
|
|
def __init__(self):
|
|
self._userkey = ''
|
|
|
|
@property
|
|
def get_legacy_cookie_id(self):
|
|
if self._userkey != '':
|
|
return self._userkey
|
|
self._userkey = self.__oldcookiedeviceid()
|
|
return self._userkey
|
|
|
|
def __bytearraytostring(self, bytearr):
|
|
wincheck = re.match('@ByteArray\\((.+)\\)', bytearr)
|
|
if wincheck:
|
|
return wincheck.group(1)
|
|
return bytearr
|
|
|
|
def plist_to_dictionary(self, filename):
|
|
from subprocess import Popen, PIPE
|
|
from plistlib import readPlistFromString
|
|
'Pipe the binary plist through plutil and parse the xml output'
|
|
with open(filename, 'rb') as f:
|
|
content = f.read()
|
|
args = ['plutil', '-convert', 'xml1', '-o', '-', '--', '-']
|
|
p = Popen(args, stdin=PIPE, stdout=PIPE)
|
|
p.stdin.write(content)
|
|
out, err = p.communicate()
|
|
return readPlistFromString(out)
|
|
|
|
def __oldcookiedeviceid(self):
|
|
'''Optionally attempt to get a device id using the old cookie method.
|
|
Must have winreg installed on Windows machines for successful key retrieval.'''
|
|
wsuid = ''
|
|
pwsdid = ''
|
|
try:
|
|
if sys.platform.startswith('win'):
|
|
import winreg
|
|
regkey_browser = winreg.OpenKey(winreg.HKEY_CURRENT_USER, 'Software\\Kobo\\Kobo Desktop Edition\\Browser')
|
|
cookies = winreg.QueryValueEx(regkey_browser, 'cookies')
|
|
bytearrays = cookies[0]
|
|
elif sys.platform.startswith('darwin'):
|
|
prefs = os.path.join(os.environ['HOME'], 'Library/Preferences/com.kobo.Kobo Desktop Edition.plist')
|
|
cookies = self.plist_to_dictionary(prefs)
|
|
bytearrays = cookies['Browser.cookies']
|
|
for bytearr in bytearrays:
|
|
cookie = self.__bytearraytostring(bytearr)
|
|
wsuidcheck = re.match("^wsuid=([0-9a-f-]+)", cookie)
|
|
if(wsuidcheck):
|
|
wsuid = wsuidcheck.group(1)
|
|
pwsdidcheck = re.match('^pwsdid=([0-9a-f-]+)', cookie)
|
|
if (pwsdidcheck):
|
|
pwsdid = pwsdidcheck.group(1)
|
|
if (wsuid == '' or pwsdid == ''):
|
|
return None
|
|
preuserkey = string.join((pwsdid, wsuid), '')
|
|
userkey = hashlib.sha256(preuserkey).hexdigest()
|
|
return binascii.a2b_hex(userkey[32:])
|
|
except KeyError:
|
|
print ('No "cookies" key found in Kobo plist: no legacy user key found.')
|
|
return None
|
|
except:
|
|
print ('Error parsing Kobo plist: no legacy user key found.')
|
|
return None
|