Python 3 fix for epubtest.py that detects version of DRM used

pull/1383/head
Apprentice Harper 3 years ago
parent 5ec9c98a0b
commit 45038cc77b

@ -169,12 +169,12 @@ def getfiledata(file, zi):
def encryption(infile):
# returns encryption: one of Unencrypted, Adobe, B&N and Unknown
encryption = "Unknown"
encryption = "Error When Checking."
try:
with open(infile,'rb') as infileobject:
bookdata = infileobject.read(58)
# Check for Zip
if bookdata[0:0+2] == "PK":
if bookdata[0:0+2] == b"PK":
foundrights = False
foundencryption = False
inzip = zipfile.ZipFile(infile,'r')
@ -198,7 +198,10 @@ def encryption(infile):
def main():
argv=unicode_argv()
print(encryption(argv[1]))
if len(argv) < 2:
print("Give an ePub file as a parameter.")
else:
print(encryption(argv[1]))
return 0
if __name__ == "__main__":

Loading…
Cancel
Save