mirrors/DeDRM_tools
mirrors/DeDRM_tools
1
0
Fork 0
mirror of https://github.com/noDRM/DeDRM_tools.git synced 2024-12-28 20:17:39 +06:00
Code Issues Projects Releases Wiki Activity

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