Miroirs/noDRM_DeDRM_tools
1
0
Fork 0
mirror of https://github.com/noDRM/DeDRM_tools synced 2024-11-16 19:48:10 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Apprentice Harper 2020-11-27 15:49:57 +00:00
parent 5ec9c98a0b
commit 45038cc77b
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
DeDRM_plugin/epubtest.py
View file

@ -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__":