Abilty to get book from just IA url

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)