146 lines
5.1 KiB
Python
146 lines
5.1 KiB
Python
import argparse
|
||
from Editer2 import Editer
|
||
import os
|
||
import shutil
|
||
from utils import *
|
||
|
||
def parse_args():
|
||
"""Parse input arguments."""
|
||
parser = argparse.ArgumentParser(description='config')
|
||
parser.add_argument('--book_no', default='0000', type=str)
|
||
parser.add_argument('--volume_no', default='1', type=int)
|
||
parser.add_argument('--no_input', default=False, type=bool)
|
||
args = parser.parse_args()
|
||
return args
|
||
|
||
|
||
def query_chaps(book_no):
|
||
print('未输入卷号，将返回书籍目录信息......')
|
||
editer = Editer(root_path='./out', book_no=book_no)
|
||
print('*******************************')
|
||
print(editer.title, editer.author)
|
||
print('*******************************')
|
||
editer.get_chap_list()
|
||
print('*******************************')
|
||
print('请输入所需要的卷号进行下载（多卷可以用英文逗号分隔或直接使用连字符，详情见说明）')
|
||
|
||
temp_path = ''
|
||
|
||
def delete_tmp():
|
||
print(temp_path)
|
||
if os.path.exists(temp_path):
|
||
shutil.rmtree(temp_path)
|
||
|
||
def download_single_volume(root_path,
|
||
book_no,
|
||
volume_no,
|
||
is_gui=False,
|
||
hang_signal=None,
|
||
progressring_signal=None,
|
||
cover_signal=None,
|
||
edit_line_hang=None):
|
||
|
||
editer = Editer(root_path=root_path, book_no=book_no, volume_no=volume_no)
|
||
print('正在积极地获取书籍信息....')
|
||
success = editer.get_index_url()
|
||
if not success:
|
||
print('书籍信息获取失败')
|
||
return
|
||
print(editer.title + '-' + editer.volume['book_name'], editer.author)
|
||
print('****************************')
|
||
temp_path = editer.temp_path
|
||
if not editer.is_buffer():
|
||
editer.check_volume(is_gui=is_gui, signal=hang_signal, editline=edit_line_hang)
|
||
print('正在下载文本....')
|
||
print('*********************')
|
||
editer.get_text()
|
||
print('*********************')
|
||
editer.buffer()
|
||
else:
|
||
print('检测到文本文件，直接下载插图')
|
||
editer.buffer()
|
||
|
||
|
||
print('正在下载插图.....................................')
|
||
editer.get_image(is_gui=is_gui, signal=progressring_signal)
|
||
|
||
print('正在编辑元数据....')
|
||
editer.get_cover(is_gui=is_gui, signal=cover_signal)
|
||
editer.get_toc()
|
||
editer.get_content()
|
||
editer.get_epub_head()
|
||
|
||
print('正在生成电子书....')
|
||
epub_file = editer.get_epub()
|
||
print('生成成功！', f'电子书路径【{epub_file}】')
|
||
|
||
|
||
def downloader_router(root_path,
|
||
book_no,
|
||
volume_no,
|
||
is_gui=False,
|
||
hang_signal=None,
|
||
progressring_signal=None,
|
||
cover_signal=None,
|
||
edit_line_hang=None):
|
||
is_multi_chap = False
|
||
if len(book_no)==0:
|
||
print('请检查输入是否完整正确！')
|
||
return
|
||
elif volume_no == '':
|
||
query_chaps(book_no)
|
||
return
|
||
elif volume_no.isdigit():
|
||
volume_no = int(volume_no)
|
||
if volume_no<=0:
|
||
print('请检查输入是否完整正确！')
|
||
return
|
||
elif "-" in volume_no:
|
||
start, end = map(str, volume_no.split("-"))
|
||
if start.isdigit() and end.isdigit() and int(start)>0 and int(start)<int(end):
|
||
volume_no_list = list(range(int(start), int(end) + 1))
|
||
is_multi_chap = True
|
||
else:
|
||
print('请检查输入是否完整正确！')
|
||
return
|
||
elif "," in volume_no:
|
||
volume_no_list = [num for num in volume_no.split(",")]
|
||
if all([num.isdigit() for num in volume_no_list]):
|
||
volume_no_list = [int(num) for num in volume_no_list]
|
||
is_multi_chap = True
|
||
else:
|
||
print('请检查输入是否完整正确！')
|
||
return
|
||
else:
|
||
print('请检查输入是否完整正确！')
|
||
return
|
||
if is_multi_chap:
|
||
for volume_no in volume_no_list:
|
||
download_single_volume(root_path, book_no, volume_no, is_gui, hang_signal, progressring_signal, cover_signal, edit_line_hang)
|
||
print('所有下载任务都已经完成！')
|
||
else:
|
||
download_single_volume(root_path, book_no, volume_no, is_gui, hang_signal, progressring_signal, cover_signal, edit_line_hang)
|
||
|
||
if __name__=='__main__':
|
||
args = parse_args()
|
||
download_path = os.path.join(os.path.expanduser('~'), 'Downloads')
|
||
|
||
if args.no_input:
|
||
downloader_router(root_path='out', book_no=args.book_no, volume_no=args.volume_no)
|
||
else:
|
||
while True:
|
||
args.book_no = input('请输入书籍号：')
|
||
args.volume_no = input('请输入卷号(查看目录信息不输入直接按回车，下载多卷请使用逗号分隔或者连字符-)：')
|
||
downloader_router(root_path='out', book_no=args.book_no, volume_no=args.volume_no)
|
||
# args.book_no = '2931'
|
||
# args.volume_no = '7'
|
||
# downloader_router(root_path='out', book_no=args.book_no, volume_no=args.volume_no)
|
||
# exit(0)
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|