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 argparse
|
||||||
import os
|
|
||||||
import urllib3
|
|
||||||
import logging
|
|
||||||
import json
|
import json
|
||||||
|
import logging
|
||||||
|
import os
|
||||||
|
|
||||||
|
import urllib3
|
||||||
|
|
||||||
from kindle_download_helper.config import (
|
from kindle_download_helper.config import (
|
||||||
DEFAULT_OUT_DIR,
|
|
||||||
DEFAULT_SESSION_FILE,
|
|
||||||
DEFAULT_OUT_DEDRM_DIR,
|
DEFAULT_OUT_DEDRM_DIR,
|
||||||
|
DEFAULT_OUT_DIR,
|
||||||
DEFAULT_OUT_EPUB_DIR,
|
DEFAULT_OUT_EPUB_DIR,
|
||||||
|
DEFAULT_SESSION_FILE,
|
||||||
)
|
)
|
||||||
|
from kindle_download_helper.kindle import Kindle
|
||||||
|
|
||||||
logger = logging.getLogger("kindle")
|
logger = logging.getLogger("kindle")
|
||||||
fh = logging.FileHandler(".error_books.log")
|
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