256 lines
9.6 KiB
Python
256 lines
9.6 KiB
Python
from novelCiwei import NovelCiwei
|
|
from db import CwmDb
|
|
from config import Config
|
|
from key import import_keys
|
|
from booksnew import BooksNew
|
|
from crypto import decrypt
|
|
from os.path import dirname
|
|
from os import makedirs
|
|
from epub import EpubFile
|
|
from utils import ask_choice
|
|
import json
|
|
from random import choice
|
|
|
|
|
|
key_imported = False
|
|
key_force_imported = False
|
|
|
|
|
|
def get_key(db: CwmDb, cfg: Config, chapter_id: int):
|
|
global key_imported
|
|
keys = db.get_key(chapter_id)
|
|
if len(keys) == 0:
|
|
if key_imported:
|
|
raise ValueError('The key is not found.')
|
|
else:
|
|
import_keys(cfg.key, db)
|
|
key_imported = True
|
|
keys = db.get_key(chapter_id)
|
|
if len(keys) == 0:
|
|
raise ValueError('The key is not found.')
|
|
return keys
|
|
|
|
|
|
def try_decrypt(db: CwmDb, cfg: Config, content, chapter_id: int):
|
|
global key_force_imported
|
|
keys = get_key(db, cfg, chapter_id)
|
|
for key in keys:
|
|
try:
|
|
return decrypt(content, key).decode()
|
|
except Exception:
|
|
pass
|
|
if not key_force_imported:
|
|
import_keys(cfg.key, db, True)
|
|
key_force_imported = True
|
|
return try_decrypt(db, cfg, content, chapter_id)
|
|
raise ValueError('Failed to decrypt the content.')
|
|
|
|
|
|
def export_chapter(ncw: NovelCiwei, db: CwmDb, cfg: Config, bn: BooksNew,
|
|
chapter_id: int):
|
|
chapter = ncw.get_chapter(chapter_id)
|
|
book_id = int(chapter['book_id'])
|
|
raw_content = bn.get_chapter(book_id, chapter_id)
|
|
content = try_decrypt(db, cfg, raw_content, chapter_id)
|
|
filename = cfg.get_export_chapter(chapter)
|
|
d = dirname(filename)
|
|
makedirs(d, exist_ok=True)
|
|
with open(filename, 'w', encoding='UTF-8') as f:
|
|
f.write(chapter['chapter_title'] + '\n')
|
|
f.write(content)
|
|
|
|
|
|
def export_book(ncw: NovelCiwei, db: CwmDb, cfg: Config, bn: BooksNew,
|
|
book_id: int):
|
|
book = ncw.get_book_in_shelf(book_id)
|
|
if book is None:
|
|
book = ncw.get_book_in_readhistory(book_id)
|
|
if book is None:
|
|
raise ValueError('The book is not found.')
|
|
if cfg.export_txt:
|
|
txt_filename = cfg.get_export_book(book, 'txt')
|
|
d = dirname(txt_filename)
|
|
makedirs(d, exist_ok=True)
|
|
txt = open(txt_filename, 'w', encoding='UTF-8')
|
|
if cfg.export_epub:
|
|
try:
|
|
epub = EpubFile(cfg, cfg.get_export_book(book, 'epub'))
|
|
epub.set_book(book)
|
|
except Exception as e:
|
|
if cfg.export_txt:
|
|
txt.close()
|
|
raise e
|
|
try:
|
|
chapters = ncw.get_chapter_with_bookid(book_id)
|
|
divisions = ncw.get_divisions_with_bookid(book_id)
|
|
maps = {}
|
|
count = 0
|
|
for chapter in chapters:
|
|
division_id = chapter['division_id']
|
|
if division_id in maps:
|
|
maps[division_id].append(chapter)
|
|
else:
|
|
maps[division_id] = [chapter]
|
|
for division in divisions:
|
|
division_name = division['division_name']
|
|
division_id = division['division_id']
|
|
is_linear = db.get_mark(division_id)
|
|
if cfg.export_txt:
|
|
txt.write(f"第{division['division_index']}卷 {division_name}\n")
|
|
if division['description']:
|
|
txt.write(division['description'] + '\n\n')
|
|
if cfg.export_epub and division['description']:
|
|
print('TODO: add division description to epub.')
|
|
if division_id not in maps:
|
|
continue
|
|
chapter_index = 1
|
|
for chapter in maps[division_id]:
|
|
chapter_id = chapter['chapter_id']
|
|
chapter_title = chapter['chapter_title']
|
|
if chapter['is_download']:
|
|
raw_content = bn.get_chapter(book_id, chapter_id)
|
|
content = try_decrypt(db, cfg, raw_content, chapter_id)
|
|
if cfg.export_txt:
|
|
txt.write(f"第{chapter_index}章 {chapter_title}\n")
|
|
txt.write(content + '\n\n')
|
|
if cfg.export_epub:
|
|
epub.add_chapter(chapter, content, division_name,
|
|
is_linear)
|
|
count += 1
|
|
elif cfg.export_nodownload:
|
|
if cfg.export_txt:
|
|
txt.write(f"第{chapter_index}章 {chapter_title} (未下载)\n\n") # noqa: E501
|
|
if cfg.export_epub:
|
|
epub.add_nodownload_chapter(chapter, division_name,
|
|
is_linear)
|
|
chapter_index += 1
|
|
print(f'Exported {count} chapters.')
|
|
finally:
|
|
if cfg.export_txt:
|
|
txt.close()
|
|
if cfg.export_epub:
|
|
epub.save_epub_file()
|
|
|
|
|
|
def export_all(ncw: NovelCiwei, db: CwmDb, cfg: Config, bn: BooksNew):
|
|
books = ncw.get_all_books()
|
|
for book in books:
|
|
book_id = int(book[0])
|
|
try:
|
|
export_book(ncw, db, cfg, bn, book_id)
|
|
except Exception as e:
|
|
print(f'Failed to export book {book_id}: {e}')
|
|
|
|
|
|
class ExportCli:
|
|
def __init__(self, ncw: NovelCiwei, db: CwmDb, cfg: Config, bn: BooksNew):
|
|
self.ncw = ncw
|
|
self.db = db
|
|
self.cfg = cfg
|
|
self.bn = bn
|
|
self.action = None
|
|
self.shelf = None
|
|
self.shelfs = None
|
|
self.book = None
|
|
self.book_id = None
|
|
self.division_id = None
|
|
self.fns = [self.ask_action]
|
|
|
|
def ask_action(self):
|
|
self.action = ask_choice(self.cfg, [], '请选择要导出的类型：', extra=[
|
|
('b', '整本书', 'book'), ('c', '单章', 'chapter'), ('a', '所有书', 'all'),
|
|
('m', '标记为非线性卷', 'mark')])
|
|
if self.action == 'all':
|
|
export_all(self.ncw, self.db, self.cfg, self.bn)
|
|
return
|
|
self.fns.append(self.ask_shelf)
|
|
|
|
def ask_book(self):
|
|
if self.shelf == 'readhistory':
|
|
books = [json.loads(b['book_info']) for b in self.ncw.get_read_history()] # noqa: E501
|
|
elif self.shelf == 'all':
|
|
bookids = [int(b[0]) for b in self.ncw.get_all_books()]
|
|
books = []
|
|
for bid in bookids:
|
|
book = self.ncw.get_book_in_shelf(bid) or self.ncw.get_book_in_readhistory(bid) # noqa: E501
|
|
if book:
|
|
books.append(book)
|
|
else:
|
|
books = [json.loads(b['book_info']) for b in self.shelfs[self.shelf]] # noqa: E501
|
|
self.book = ask_choice(self.cfg, books, '请选择书：', lambda b: f"{b['book_name']} - {b['author_name']}", [('b', '返回', 'back')]) # noqa: E501
|
|
if self.book == 'back':
|
|
self.fns.append(self.ask_shelf)
|
|
return
|
|
self.book_id = int(self.book['book_id'])
|
|
if self.action == 'book':
|
|
export_book(self.ncw, self.db, self.cfg, self.bn, self.book_id)
|
|
return
|
|
self.fns.append(self.ask_division)
|
|
|
|
def ask_chapter(self):
|
|
chapters = self.ncw.get_chapter_with_bookid_division(self.book_id, self.division_id) # noqa: E501
|
|
chapter = ask_choice(self.cfg, chapters, '请选择章节：', lambda b: b['chapter_title'], [('b', '返回', 'back')]) # noqa: E501
|
|
if chapter == 'back':
|
|
self.fns.append(self.ask_division)
|
|
return
|
|
chapter_id = int(chapter['chapter_id'])
|
|
export_chapter(self.ncw, self.db, self.cfg, self.bn, chapter_id)
|
|
|
|
def ask_division(self):
|
|
divisions = self.ncw.get_divisions_with_bookid(self.book_id)
|
|
extras = [('b', '返回', 'back')]
|
|
if self.action == 'mark':
|
|
extras.append(('q', '退出', 'quit'))
|
|
|
|
def show_division(division):
|
|
name = division['division_name']
|
|
if self.action == 'mark':
|
|
division_id = int(division['division_id'])
|
|
if not self.db.get_mark(division_id):
|
|
name += ' (非线性卷)'
|
|
return name
|
|
division = ask_choice(self.cfg, divisions, '请选择卷：', show_division, extras) # noqa: E501
|
|
if division == 'back':
|
|
self.fns.append(self.ask_book)
|
|
return
|
|
if division == 'quit':
|
|
return
|
|
self.division_id = int(division['division_id'])
|
|
if self.action == 'mark':
|
|
now = self.db.get_mark(self.division_id)
|
|
self.db.set_mark(self.division_id, not now)
|
|
self.fns.append(self.ask_division)
|
|
return
|
|
self.fns.append(self.ask_chapter)
|
|
|
|
def ask_shelf(self):
|
|
books = self.ncw.get_books()
|
|
self.shelfs = {}
|
|
for book in books:
|
|
shelf_id = book['shelf_id']
|
|
if shelf_id in self.shelfs:
|
|
self.shelfs[shelf_id].append(book)
|
|
else:
|
|
self.shelfs[shelf_id] = [book]
|
|
|
|
def show_shelf(shelf: str):
|
|
data = f"{shelf} ({len(self.shelfs[shelf])} 本书)"
|
|
book = json.loads(choice(self.shelfs[shelf])['book_info'])
|
|
data += f"\n书架内有 {book['book_name']} - {book['author_name']}"
|
|
return data
|
|
self.shelf = ask_choice(self.cfg, [i for i in self.shelfs.keys()],
|
|
'请选择书架：', show_shelf,
|
|
[('r', '阅读历史', 'readhistory'),
|
|
('a', '所有书', 'all'),
|
|
('b', '返回', 'back')])
|
|
if self.shelf == 'back':
|
|
self.fns.append(self.ask_action)
|
|
return
|
|
self.fns.append(self.ask_book)
|
|
|
|
def start(self):
|
|
while True:
|
|
if len(self.fns) == 0:
|
|
break
|
|
self.fns.pop(0)()
|