Abilty to get book from just IA url
This commit is contained in:
bipinkrish
parent
b6a000b7b9
commit
d4999b9c48
4 changed files with 231 additions and 51 deletions
110
DeGourou.py
110
DeGourou.py
|
|
@ -7,34 +7,43 @@ 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
|
||||
from setup.ia import SESSION_FILE, manage_login, get_book, return_book
|
||||
|
||||
def main(acsmFile, login, outputFilename):
|
||||
|
||||
# user login
|
||||
if login:
|
||||
if not exists("account"):
|
||||
mkdir("account")
|
||||
loginAndGetKey()
|
||||
exit(0)
|
||||
|
||||
def loginADE(email, password):
|
||||
if email is None or password is None:
|
||||
print("Email or Password cannot be empty")
|
||||
print()
|
||||
return
|
||||
if not exists('account'): mkdir('account')
|
||||
loginAndGetKey(email, password)
|
||||
print()
|
||||
|
||||
def loginIA(email,password):
|
||||
if email is None or password is None:
|
||||
print("Email or Password cannot be empty")
|
||||
print()
|
||||
return
|
||||
manage_login(email,password)
|
||||
print()
|
||||
|
||||
def main(acsmFile, outputFilename):
|
||||
if not exists('account'): mkdir('account')
|
||||
|
||||
# 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)
|
||||
return
|
||||
|
||||
# download
|
||||
encryptedFile = downloadFile(acsmFile)
|
||||
|
|
@ -49,7 +58,7 @@ def main(acsmFile, login, outputFilename):
|
|||
else:
|
||||
print("File format not supported")
|
||||
print()
|
||||
exit(1)
|
||||
return
|
||||
|
||||
remove(encryptedFile)
|
||||
if outputFilename is None:
|
||||
|
|
@ -60,20 +69,69 @@ def main(acsmFile, login, outputFilename):
|
|||
print(tempName)
|
||||
print()
|
||||
|
||||
def handle_IA(url,format):
|
||||
if not exists(SESSION_FILE):
|
||||
print("Login to InternetArchive first or give ACSm file as input")
|
||||
return
|
||||
acsmFile = get_book(url,format)
|
||||
if acsmFile is None:
|
||||
print("Could not get Book, try using ACSm file as input")
|
||||
return
|
||||
main(acsmFile,None)
|
||||
remove(acsmFile)
|
||||
if(return_book(url) is None):
|
||||
print("Please return it yourself")
|
||||
|
||||
|
||||
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)")
|
||||
parser = argparse.ArgumentParser(description="Download and Decrypt an encrypted PDF or EPUB file.")
|
||||
parser.add_argument("-f", type=str, nargs='?', default=None, help="path to the ACSM file")
|
||||
parser.add_argument("-u", type=str, nargs='?', default=None, help="book url from InternetArchive")
|
||||
parser.add_argument("-t", type=str, nargs='?', default='pdf', help="book file type/format/extension for book url (defaults to PDF)")
|
||||
parser.add_argument("-o", type=str, nargs='?', default=None, help="output file name")
|
||||
parser.add_argument("-la", action="store_true", help="login to your ADE account.")
|
||||
parser.add_argument("-li", action="store_true", help="login to your InternetArchive.")
|
||||
parser.add_argument("-e", type=str, nargs='?', default=None, help="email/username")
|
||||
parser.add_argument("-p", type=str, nargs='?', default=None, help="password")
|
||||
parser.add_argument("-lo", action="store_true", help="logout from all")
|
||||
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)
|
||||
# Logout
|
||||
if args.lo:
|
||||
from shutil import rmtree
|
||||
rmtree("account")
|
||||
mkdir('account')
|
||||
print()
|
||||
print("Logout Sucessfull")
|
||||
print()
|
||||
|
||||
main(args.file, args.login, args.output)
|
||||
# ADE login
|
||||
elif args.la:
|
||||
print()
|
||||
print("chose login for ADE")
|
||||
loginADE(args.e, args.p)
|
||||
|
||||
# IA login
|
||||
elif args.li:
|
||||
print()
|
||||
print("chose login for InternetArchive")
|
||||
loginIA(args.e, args.p)
|
||||
|
||||
# Book url
|
||||
elif args.u:
|
||||
print()
|
||||
if not args.t in ['pdf','epub']:
|
||||
print("only PDF and EPUB are supported")
|
||||
else:
|
||||
handle_IA(args.u, args.t)
|
||||
print()
|
||||
|
||||
# check for default value
|
||||
elif args.f == None:
|
||||
if exists("URLLink.acsm"):
|
||||
args.f = "URLLink.acsm"
|
||||
main(args.f, args.output)
|
||||
else: parser.print_help()
|
||||
|
||||
else:
|
||||
main(args.f, args.output)
|
||||
|
|
|
|||
|
|
@ -1,3 +1,4 @@
|
|||
pycryptodomex==3.17
|
||||
oscrypto==1.3.0
|
||||
lxml==4.9.2
|
||||
requests
|
||||
135
setup/ia.py
Normal file
135
setup/ia.py
Normal file
|
|
@ -0,0 +1,135 @@
|
|||
from os import path, mkdir
|
||||
import requests
|
||||
import random
|
||||
import string
|
||||
import pickle
|
||||
|
||||
|
||||
SESSION_FILE = 'account/session.pkl'
|
||||
session = None
|
||||
if path.exists(SESSION_FILE):
|
||||
with open(SESSION_FILE, 'rb') as f: session = pickle.load(f)
|
||||
|
||||
|
||||
# print error
|
||||
def display_error(response, message):
|
||||
print(message)
|
||||
print(response)
|
||||
print(response.text)
|
||||
|
||||
|
||||
# login and format
|
||||
def format_data(content_type, fields):
|
||||
data = ""
|
||||
for name, value in fields.items():
|
||||
data += f"--{content_type}\x0d\x0aContent-Disposition: form-data; name=\"{name}\"\x0d\x0a\x0d\x0a{value}\x0d\x0a"
|
||||
data += content_type+"--"
|
||||
return data
|
||||
|
||||
def login(email, password):
|
||||
session = requests.Session()
|
||||
session.get("https://archive.org/account/login")
|
||||
content_type = "----WebKitFormBoundary"+"".join(random.sample(string.ascii_letters + string.digits, 16))
|
||||
|
||||
headers = {'Content-Type': 'multipart/form-data; boundary='+content_type}
|
||||
data = format_data(content_type, {"username":email, "password":password, "submit_by_js":"true"})
|
||||
|
||||
response = session.post("https://archive.org/account/login", data=data, headers=headers)
|
||||
if "bad_login" in response.text:
|
||||
print("[-] Invalid credentials!")
|
||||
return None
|
||||
elif "Successful login" in response.text:
|
||||
print("[+] Successful login")
|
||||
return session
|
||||
else:
|
||||
display_error(response, "[-] Error while login:")
|
||||
return None
|
||||
|
||||
|
||||
# get book
|
||||
def loan(book_id):
|
||||
global session
|
||||
if not session:
|
||||
with open(SESSION_FILE, 'rb') as f: session = pickle.load(f)
|
||||
data = {
|
||||
"action": "grant_access",
|
||||
"identifier": book_id
|
||||
}
|
||||
response = session.post("https://archive.org/services/loans/loan/searchInside.php", data=data)
|
||||
data['action'] = "browse_book"
|
||||
response = session.post("https://archive.org/services/loans/loan/", data=data)
|
||||
|
||||
if response.status_code == 400 :
|
||||
if response.json()["error"] == "This book is not available to borrow at this time. Please try again later.":
|
||||
print("This book doesn't need to be borrowed")
|
||||
return session
|
||||
else :
|
||||
display_error(response, "Something went wrong when trying to borrow the book.")
|
||||
return None
|
||||
|
||||
data['action'] = "create_token"
|
||||
response = session.post("https://archive.org/services/loans/loan/", data=data)
|
||||
|
||||
if "token" in response.text:
|
||||
print("[+] Successful loan")
|
||||
return session
|
||||
else:
|
||||
display_error(response, "Something went wrong when trying to borrow the book, maybe you can't borrow this book.")
|
||||
return None
|
||||
|
||||
# acsm file
|
||||
def get_acsmfile(bookid,format="pdf"):
|
||||
global session
|
||||
if not session:
|
||||
with open(SESSION_FILE, 'rb') as f: session = pickle.load(f)
|
||||
|
||||
response = session.get(f"https://archive.org/services/loans/loan/?action=media_url&format={format}&redirect=1&identifier={bookid}")
|
||||
|
||||
if response.status_code == 200:
|
||||
with open(f"{bookid}.acsm","w") as af: af.write(response.text)
|
||||
return f"{bookid}.acsm"
|
||||
else:
|
||||
display_error(response, "Something went wrong when trying to get ACSM")
|
||||
return None
|
||||
|
||||
|
||||
# return the book
|
||||
def return_loan(book_id):
|
||||
global session
|
||||
if not session:
|
||||
with open(SESSION_FILE, 'rb') as f: session = pickle.load(f)
|
||||
data = {
|
||||
"action": "return_loan",
|
||||
"identifier": book_id
|
||||
}
|
||||
response = session.post("https://archive.org/services/loans/loan/", data=data)
|
||||
if response.status_code == 200 and response.json()["success"]:
|
||||
print("[+] Book returned")
|
||||
return True
|
||||
else:
|
||||
display_error(response, "Something went wrong when trying to return the book")
|
||||
return None
|
||||
|
||||
|
||||
# manage
|
||||
def manage_login(email,password):
|
||||
global session
|
||||
if not path.exists('account'): mkdir('account')
|
||||
sess = login(email,password)
|
||||
if sess is not None:
|
||||
with open(SESSION_FILE, 'wb') as f: pickle.dump(sess, f)
|
||||
session = sess
|
||||
|
||||
def get_book(url,format):
|
||||
global session
|
||||
bookid = url.split("/")[4]
|
||||
sess = loan(bookid)
|
||||
if sess is not None:
|
||||
with open(SESSION_FILE, 'wb') as f: pickle.dump(sess, f)
|
||||
session = sess
|
||||
return get_acsmfile(bookid,format)
|
||||
return None
|
||||
|
||||
def return_book(url):
|
||||
bookid = url.split("/")[4]
|
||||
return return_loan(bookid)
|
||||
|
|
@ -16,21 +16,7 @@ from decrypt.params import KEYPATH
|
|||
|
||||
#################################################################
|
||||
|
||||
def takeInput():
|
||||
|
||||
global VAR_MAIL
|
||||
global VAR_PASS
|
||||
|
||||
VAR_MAIL = input("Enter Mail: ")
|
||||
VAR_PASS = input("Enter Password: ")
|
||||
|
||||
if VAR_MAIL == "" or VAR_MAIL == "":
|
||||
print("It cannot be empty")
|
||||
print()
|
||||
exit(1)
|
||||
|
||||
|
||||
def loginAndGetKey():
|
||||
def loginAndGetKey(email, password):
|
||||
|
||||
global VAR_MAIL
|
||||
global VAR_PASS
|
||||
|
|
@ -40,31 +26,32 @@ def loginAndGetKey():
|
|||
# acc files
|
||||
if True:
|
||||
|
||||
takeInput()
|
||||
VAR_MAIL = email
|
||||
VAR_PASS = password
|
||||
|
||||
print("Logging in")
|
||||
|
||||
createDeviceKeyFile()
|
||||
|
||||
success = createDeviceFile(True, VAR_VER)
|
||||
if (success is False):
|
||||
print("Error, couldn't create device file.")
|
||||
exit(1)
|
||||
return
|
||||
|
||||
success, resp = createUser(VAR_VER, None)
|
||||
if (success is False):
|
||||
print("Error, couldn't create user: %s" % resp)
|
||||
exit(1)
|
||||
return
|
||||
|
||||
success, resp = signIn("AdobeID", VAR_MAIL, VAR_PASS)
|
||||
|
||||
if (success is False):
|
||||
print("Login unsuccessful: " + resp)
|
||||
exit(1)
|
||||
return
|
||||
|
||||
success, resp = activateDevice(VAR_VER, None)
|
||||
if (success is False):
|
||||
print("Couldn't activate device: " + resp)
|
||||
exit(1)
|
||||
return
|
||||
|
||||
print("Authorized to account " + VAR_MAIL)
|
||||
|
||||
|
|
@ -79,16 +66,15 @@ def loginAndGetKey():
|
|||
success = exportAccountEncryptionKeyDER(filename)
|
||||
if (success is False):
|
||||
print("Couldn't export key.")
|
||||
exit(1)
|
||||
return
|
||||
print("Successfully exported key for account " + VAR_MAIL + " to file " + filename)
|
||||
|
||||
else:
|
||||
print("failed")
|
||||
exit(1)
|
||||
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
exit(1)
|
||||
|
||||
|
||||
print('All Set')
|
||||
print()
|
||||
Loading…
Reference in a new issue