diff --git a/DeDRM_plugin/kgenpids.py b/DeDRM_plugin/kgenpids.py index 9800254..103b77b 100644 --- a/DeDRM_plugin/kgenpids.py +++ b/DeDRM_plugin/kgenpids.py @@ -174,14 +174,14 @@ def pidFromSerial(s, l): # Parse the EXTH header records and use the Kindle serial number to calculate the book pid. def getKindlePids(rec209, token, serialnum): + if isinstance(serialnum,str): + serialnum = serialnum.encode('utf-8') + if rec209 is None: return [serialnum] pids=[] - if isinstance(serialnum,str): - serialnum = serialnum.encode('utf-8') - # Compute book PID pidHash = SHA1(serialnum+rec209+token) bookPID = encodePID(pidHash)