mirror of
https://github.com/yihong0618/Kindle_download_helper.git
synced 2025-11-22 07:59:04 +08:00
fix: file path
This commit is contained in:
yihong0618
329
cli.py
329
cli.py
@@ -1,329 +0,0 @@
|
||||
import argparse
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
|
||||
import urllib3
|
||||
|
||||
from kindle_download_helper.config import (
|
||||
DEFAULT_OUT_DEDRM_DIR,
|
||||
DEFAULT_OUT_DIR,
|
||||
DEFAULT_OUT_EPUB_DIR,
|
||||
DEFAULT_SESSION_FILE,
|
||||
)
|
||||
from kindle_download_helper.kindle import Kindle
|
||||
|
||||
logger = logging.getLogger("kindle")
|
||||
fh = logging.FileHandler(".error_books.log")
|
||||
fh.setLevel(logging.ERROR)
|
||||
logger.addHandler(fh)
|
||||
|
||||
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
||||
|
||||
|
||||
# download selected books for cli
|
||||
def download_selected_books(kindle, options):
|
||||
# get all books and get the default device
|
||||
print("Getting all books, please wait...")
|
||||
books = kindle.get_all_books(filetype=options.filetype)
|
||||
device = kindle.find_device()
|
||||
|
||||
# print all books
|
||||
for idx, book in enumerate(books):
|
||||
print(
|
||||
"Index: "
|
||||
+ "{:>5d}".format(idx + 1)
|
||||
+ " | Title: "
|
||||
+ book["title"]
|
||||
+ " | asin: "
|
||||
+ book["asin"]
|
||||
)
|
||||
|
||||
# download loop
|
||||
while True:
|
||||
# get the indices of the books to download
|
||||
indices = input(
|
||||
"Input the index of books you want to download, split by space (q to quit, l to list books).\n"
|
||||
).split()
|
||||
|
||||
# if input "q", quit
|
||||
# if input "l", list all books again
|
||||
if indices[0] == "q":
|
||||
break
|
||||
elif indices[0] == "l":
|
||||
for idx, book in enumerate(books):
|
||||
print(
|
||||
"Index: "
|
||||
+ "{:>5d}".format(idx + 1)
|
||||
+ " | Title: "
|
||||
+ book["title"]
|
||||
+ " | asin: "
|
||||
+ book["asin"]
|
||||
)
|
||||
continue
|
||||
|
||||
# decode the indices
|
||||
downlist = []
|
||||
flag = True
|
||||
for idx in indices:
|
||||
if idx.isnumeric() == False:
|
||||
if ":" in idx:
|
||||
# if is not a number, and ":" in it, then it is a range
|
||||
# decode the range
|
||||
idx_begin, idx_end = [int(i) for i in idx.split(":")]
|
||||
# append the range to downlist
|
||||
extend_list = [i for i in range(idx_begin - 1, idx_end)]
|
||||
downlist.extend(extend_list)
|
||||
else:
|
||||
# if is not a number, and no ":" in it, then it is an error
|
||||
print("Input error, please input numbers!!!")
|
||||
flag = False
|
||||
break
|
||||
else:
|
||||
# if is a number, then append it to downlist
|
||||
downlist.append(int(idx) - 1)
|
||||
if not flag:
|
||||
continue
|
||||
|
||||
# remove the duplicate indices
|
||||
downlist = list(set(downlist))
|
||||
|
||||
# check if the indices are valid
|
||||
if max(downlist) >= len(books) or min(downlist) < 0:
|
||||
print(
|
||||
"Input error, please input numbers between 1 and "
|
||||
+ str(len(books))
|
||||
+ "!!!"
|
||||
)
|
||||
continue
|
||||
|
||||
# print the books to download
|
||||
for idx in downlist:
|
||||
print(
|
||||
"Index: "
|
||||
+ "{:>5d}".format(idx + 1)
|
||||
+ " | Title: "
|
||||
+ books[idx]["title"]
|
||||
+ " | asin: "
|
||||
+ books[idx]["asin"]
|
||||
)
|
||||
print("Downloading " + str(len(downlist)) + " books:")
|
||||
|
||||
# ask if to continue
|
||||
while True:
|
||||
flag = input("Continue? (y/n)")
|
||||
if flag == "y" or flag == "n":
|
||||
break
|
||||
else:
|
||||
print("Input error, please input y or n")
|
||||
if flag == "n":
|
||||
continue
|
||||
|
||||
# download the books
|
||||
for i, idx in enumerate(downlist):
|
||||
print(
|
||||
"Downloading "
|
||||
+ str(i + 1)
|
||||
+ "/"
|
||||
+ str(len(downlist))
|
||||
+ " "
|
||||
+ books[idx]["title"]
|
||||
+ " ..."
|
||||
)
|
||||
kindle.download_one_book(books[idx], device, idx, filetype=options.filetype)
|
||||
print("Download finished.")
|
||||
|
||||
|
||||
def main():
|
||||
logger.setLevel(os.environ.get("LOGGING_LEVEL", "INFO"))
|
||||
|
||||
logger.addHandler(logging.StreamHandler())
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("csrf_token", help="amazon or amazon cn csrf token", nargs="?")
|
||||
|
||||
cookie_group = parser.add_mutually_exclusive_group()
|
||||
cookie_group.add_argument(
|
||||
"--cookie", dest="cookie", default="", help="amazon or amazon cn cookie"
|
||||
)
|
||||
cookie_group.add_argument(
|
||||
"--cookie-file", dest="cookie_file", default="", help="load cookie local file"
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--cn",
|
||||
dest="domain",
|
||||
action="store_const",
|
||||
const="cn",
|
||||
default="com",
|
||||
help="if your account is an amazon.cn account",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--jp",
|
||||
dest="domain",
|
||||
action="store_const",
|
||||
const="jp",
|
||||
default="com",
|
||||
help="if your account is an amazon.co.jp account",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--de",
|
||||
dest="domain",
|
||||
action="store_const",
|
||||
const="de",
|
||||
default="com",
|
||||
help="if your account is an amazon.de account",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--uk",
|
||||
dest="domain",
|
||||
action="store_const",
|
||||
const="uk",
|
||||
default="com",
|
||||
help="if your account is an amazon.co.uk account",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--resume-from",
|
||||
dest="index",
|
||||
type=int,
|
||||
default=1,
|
||||
help="resume from the index if download failed",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--cut-length",
|
||||
dest="cut_length",
|
||||
type=int,
|
||||
default=100,
|
||||
help="truncate the file name",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-o", "--outdir", default=DEFAULT_OUT_DIR, help="dwonload output dir"
|
||||
)
|
||||
parser.add_argument(
|
||||
"-od",
|
||||
"--outdedrmdir",
|
||||
default=DEFAULT_OUT_DEDRM_DIR,
|
||||
help="dwonload output dedrm dir",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-oe",
|
||||
"--outepubmdir",
|
||||
default=DEFAULT_OUT_EPUB_DIR,
|
||||
help="dwonload output epub dir",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-s",
|
||||
"--session-file",
|
||||
default=DEFAULT_SESSION_FILE,
|
||||
help="The reusable session dump file",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--pdoc",
|
||||
dest="filetype",
|
||||
action="store_const",
|
||||
const="PDOC",
|
||||
default="EBOK",
|
||||
help="to download personal documents or ebook",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--resolve_duplicate_names",
|
||||
dest="resolve_duplicate_names",
|
||||
action="store_true",
|
||||
help="Resolve duplicate names files to download",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--readme",
|
||||
dest="readme",
|
||||
action="store_true",
|
||||
help="If you want to generate kindle readme stats",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--dedrm",
|
||||
dest="dedrm",
|
||||
action="store_true",
|
||||
help="If you want to `dedrm` directly",
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--list",
|
||||
dest="list_only",
|
||||
action="store_true",
|
||||
help="just list books/pdoc, not to download",
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--device_sn",
|
||||
dest="device_sn",
|
||||
default="",
|
||||
help="Download file for device with this serial number",
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--mode",
|
||||
dest="mode",
|
||||
default="all",
|
||||
help="Mode of download, all: download all files at once, sel: download selected files",
|
||||
)
|
||||
|
||||
options = parser.parse_args()
|
||||
|
||||
if not os.path.exists(options.outdir):
|
||||
os.makedirs(options.outdir)
|
||||
# for dedrm
|
||||
if not os.path.exists(options.outdedrmdir):
|
||||
os.makedirs(options.outdedrmdir)
|
||||
# for epub
|
||||
if not os.path.exists(options.outepubmdir):
|
||||
os.makedirs(options.outepubmdir)
|
||||
|
||||
kindle = Kindle(
|
||||
options.csrf_token,
|
||||
options.domain,
|
||||
options.outdir,
|
||||
options.outdedrmdir,
|
||||
options.outepubmdir,
|
||||
options.cut_length,
|
||||
session_file=options.session_file,
|
||||
device_sn=options.device_sn,
|
||||
)
|
||||
# other args
|
||||
kindle.to_resolve_duplicate_names = options.resolve_duplicate_names
|
||||
kindle.dedrm = options.dedrm
|
||||
|
||||
if options.cookie_file:
|
||||
with open(options.cookie_file, "r") as f:
|
||||
kindle.set_cookie_from_string(f.read())
|
||||
elif options.cookie:
|
||||
kindle.set_cookie_from_string(options.cookie)
|
||||
else:
|
||||
kindle.is_browser_cookie = True
|
||||
|
||||
if options.list_only:
|
||||
kindle.get_devices()
|
||||
print(
|
||||
json.dumps(
|
||||
kindle.get_all_books(filetype=options.filetype),
|
||||
indent=4,
|
||||
ensure_ascii=False,
|
||||
)
|
||||
)
|
||||
exit()
|
||||
|
||||
if options.readme:
|
||||
# generate readme stats
|
||||
kindle.make_kindle_stats_readme()
|
||||
else:
|
||||
# check the download mode
|
||||
if options.mode == "all":
|
||||
# download all books
|
||||
kindle.download_books(
|
||||
start_index=options.index - 1, filetype=options.filetype
|
||||
)
|
||||
elif options.mode == "sel":
|
||||
# download selected books
|
||||
download_selected_books(kindle, options)
|
||||
else:
|
||||
print("mode error, please input all or sel")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -1,16 +1,17 @@
|
||||
from kindle_download_helper.kindle import Kindle
|
||||
import argparse
|
||||
import os
|
||||
import urllib3
|
||||
import logging
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
|
||||
import urllib3
|
||||
|
||||
from kindle_download_helper.config import (
|
||||
DEFAULT_OUT_DIR,
|
||||
DEFAULT_SESSION_FILE,
|
||||
DEFAULT_OUT_DEDRM_DIR,
|
||||
DEFAULT_OUT_DIR,
|
||||
DEFAULT_OUT_EPUB_DIR,
|
||||
DEFAULT_SESSION_FILE,
|
||||
)
|
||||
from kindle_download_helper.kindle import Kindle
|
||||
|
||||
logger = logging.getLogger("kindle")
|
||||
fh = logging.FileHandler(".error_books.log")
|
||||
|
||||
5
no_kindle.py
Normal file
5
no_kindle.py
Normal file
@@ -0,0 +1,5 @@
|
||||
from kindle_download_helper import no_main
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
no_main()
|
||||
Reference in New Issue
Block a user