80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
from setup.login_account import loginAndGetKey
|
|
from setup.fulfill import downloadFile
|
|
|
|
from decrypt.decodePDF import decryptPDF
|
|
from decrypt.decodeEPUB import decryptEPUB
|
|
|
|
import argparse
|
|
from os import mkdir, remove, rename
|
|
from os.path import exists
|
|
from sys import exit
|
|
|
|
from setup.params import FILE_DEVICEKEY, FILE_DEVICEXML, FILE_ACTIVATIONXML
|
|
from decrypt.params import KEYPATH
|
|
from setup.data import createDefaultFiles
|
|
|
|
def main(acsmFile, login, outputFilename):
|
|
|
|
# user login
|
|
if login:
|
|
if not exists("account"):
|
|
mkdir("account")
|
|
loginAndGetKey()
|
|
exit(0)
|
|
|
|
# setting up the account and keys
|
|
if not (exists(FILE_ACTIVATIONXML) and exists(FILE_DEVICEXML) and exists(FILE_DEVICEKEY) and exists(KEYPATH)):
|
|
if not exists("account"):
|
|
mkdir("account")
|
|
createDefaultFiles()
|
|
|
|
print()
|
|
|
|
# cheek for file existance
|
|
if not exists(acsmFile):
|
|
print(f"{acsmFile} file does not exist")
|
|
print()
|
|
exit(1)
|
|
|
|
# download
|
|
encryptedFile = downloadFile(acsmFile)
|
|
print(encryptedFile)
|
|
print()
|
|
|
|
# decrypt
|
|
if encryptedFile.endswith(".pdf"):
|
|
decryptedFile = decryptPDF(encryptedFile)
|
|
elif encryptedFile.endswith(".epub"):
|
|
decryptedFile = decryptEPUB(encryptedFile)
|
|
else:
|
|
print("File format not supported")
|
|
print()
|
|
exit(1)
|
|
|
|
remove(encryptedFile)
|
|
if outputFilename is None:
|
|
tempName = encryptedFile
|
|
else:
|
|
tempName = outputFilename
|
|
rename(decryptedFile, tempName)
|
|
print(tempName)
|
|
print()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Download and Decrypt an encrypted PDF or EPUB file. It uses Dummy account for ADE, you can overide using --login")
|
|
parser.add_argument("file", type=str, nargs='?', default=None, help="Path to the ACSM file")
|
|
parser.add_argument("-l", "--login", action="store_true", help="Login to your ADE account. (optional)")
|
|
parser.add_argument("-o", "--output", type=str, default=None, help="Output file name. (optional)")
|
|
args = parser.parse_args()
|
|
|
|
# check for default value
|
|
if args.file == None:
|
|
if exists("URLLink.acsm"):
|
|
args.file = "URLLink.acsm"
|
|
else:
|
|
parser.print_help()
|
|
exit(0)
|
|
|
|
main(args.file, args.login, args.output)
|