jyapayne/safaribooks
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added credential parser

Browse source
This commit is contained in:
Lorenzo Di Fuccia 2018-05-21 21:42:08 +02:00
commit 095cd6e3b2
1 changed files with 55 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

95
safaribooks.py
View file

@ -280,7 +280,7 @@ class SafariBooks:
else: else:
self.display.info("Logging into Safari Books Online...", state=True) self.display.info("Logging into Safari Books Online...", state=True)
self.do_login(*[c.replace("'", "").replace('"', "") for c in args.cred]) self.do_login(*args.cred)
if not args.no_cookies: if not args.no_cookies:
json.dump(self.cookies, open(COOKIES_FILE, "w")) json.dump(self.cookies, open(COOKIES_FILE, "w"))
@ -402,6 +402,20 @@ class SafariBooks:
return response return response
@staticmethod
def parse_cred(cred):
if ":" not in cred:
return False
sep = cred.index(":")
new_cred = ["", ""]
new_cred[0] = cred[:sep].strip("'").strip('"')
if "@" not in new_cred[0]:
return False
new_cred[1] = cred[sep + 1:]
return new_cred
def do_login(self, email, password): def do_login(self, email, password):
response = self.requests_provider(self.BASE_URL) response = self.requests_provider(self.BASE_URL)
if response == 0: if response == 0:
@ -962,48 +976,49 @@ class SafariBooks:
# MAIN # MAIN
arguments = argparse.ArgumentParser(prog="safaribooks.py", if __name__ == "__main__":
description="Download and generate an EPUB of your favorite books" arguments = argparse.ArgumentParser(prog="safaribooks.py",
" from Safari Books Online.", description="Download and generate an EPUB of your favorite books"
add_help=False, " from Safari Books Online.",
allow_abbrev=False) add_help=False,
allow_abbrev=False)
arguments.add_argument( arguments.add_argument(
"--cred", metavar="<EMAIL:PASS>", default=False, "--cred", metavar="<EMAIL:PASS>", default=False,
help="Credentials used to perform the auth login on Safari Books Online." help="Credentials used to perform the auth login on Safari Books Online."
" Es. ` --cred \"account_mail@mail.com:password01\" `." " Es. ` --cred \"account_mail@mail.com:password01\" `."
) )
arguments.add_argument( arguments.add_argument(
"--no-cookies", dest="no_cookies", action='store_true', "--no-cookies", dest="no_cookies", action='store_true',
help="Prevent your session data to be saved into `cookies.json` file." help="Prevent your session data to be saved into `cookies.json` file."
) )
arguments.add_argument( arguments.add_argument(
"--no-kindle", dest="no_kindle", action='store_true', "--no-kindle", dest="no_kindle", action='store_true',
help="Remove some CSS rules that block overflow on `table` and `pre` elements." help="Remove some CSS rules that block overflow on `table` and `pre` elements."
" Use this option if you're not going to export the EPUB to E-Readers like Amazon Kindle." " Use this option if you're not going to export the EPUB to E-Readers like Amazon Kindle."
) )
arguments.add_argument( arguments.add_argument(
"--preserve-log", dest="log", action='store_true', help="Leave the `info_XXXXXXXXXXXXX.log`" "--preserve-log", dest="log", action='store_true', help="Leave the `info_XXXXXXXXXXXXX.log`"
" file even if there isn't any error." " file even if there isn't any error."
) )
arguments.add_argument("--help", action="help", default=argparse.SUPPRESS, help='Show this help message.') arguments.add_argument("--help", action="help", default=argparse.SUPPRESS, help='Show this help message.')
arguments.add_argument( arguments.add_argument(
"bookid", metavar='<BOOK ID>', "bookid", metavar='<BOOK ID>',
help="Book digits ID that you want to download. You can find it in the URL (X-es):" help="Book digits ID that you want to download. You can find it in the URL (X-es):"
" `https://www.safaribooksonline.com/library/view/book-name/XXXXXXXXXXXXX/`" " `https://www.safaribooksonline.com/library/view/book-name/XXXXXXXXXXXXX/`"
) )
args_parsed = arguments.parse_args() args_parsed = arguments.parse_args()
if args_parsed.cred: if args_parsed.cred:
cred = args_parsed.cred.split(":") cred = SafariBooks.parse_cred(args_parsed.cred)
if len(cred) != 2 or "@" not in cred[0]: if not cred:
arguments.error("invalid credential: %s" % args_parsed.cred) arguments.error("invalid credential: %s" % args_parsed.cred)
args_parsed.cred = cred args_parsed.cred = cred
else: else:
if args_parsed.no_cookies: if args_parsed.no_cookies:
arguments.error("invalid option: `--no-cookies` is valid only if you use the `--cred` option") arguments.error("invalid option: `--no-cookies` is valid only if you use the `--cred` option")
SafariBooks(args_parsed) SafariBooks(args_parsed)