支持读取历史记录

This commit is contained in:
2024-03-09 21:15:23 +08:00
parent f980b45703
commit 8b2fc97f4e
2 changed files with 12 additions and 2 deletions

View File

@@ -134,8 +134,12 @@ def export(ncw: NovelCiwei, db: CwmDb, cfg: Config, bn: BooksNew):
book = json.loads(choice(shelfs[shelf])['book_info']) book = json.loads(choice(shelfs[shelf])['book_info'])
data += f"\n书架内有 {book['book_name']} - {book['author_name']}" data += f"\n书架内有 {book['book_name']} - {book['author_name']}"
return data return data
shelf = ask_choice(cfg, [i for i in shelfs.keys()], '请选择书架:', show_shelf) shelf = ask_choice(cfg, [i for i in shelfs.keys()], '请选择书架:', show_shelf,
books = [json.loads(b['book_info']) for b in shelfs[shelf]] [('r', '阅读历史', 'readhistory')])
if shelf == 'readhistory':
books = [json.loads(b['book_info']) for b in ncw.get_read_history()]
else:
books = [json.loads(b['book_info']) for b in shelfs[shelf]]
book = ask_choice(cfg, books, '请选择书:', lambda b: f"{b['book_name']} - {b['author_name']}") # noqa: E501 book = ask_choice(cfg, books, '请选择书:', lambda b: f"{b['book_name']} - {b['author_name']}") # noqa: E501
book_id = int(book['book_id']) book_id = int(book['book_id'])
if action == 'book': if action == 'book':

View File

@@ -42,3 +42,9 @@ class NovelCiwei:
'SELECT * FROM catalog1 WHERE chapter_id = ?;', [str(chapter_id)]) 'SELECT * FROM catalog1 WHERE chapter_id = ?;', [str(chapter_id)])
cur.row_factory = sqlite3.Row cur.row_factory = sqlite3.Row
return cur.fetchone() return cur.fetchone()
def get_read_history(self):
cur = self._db.execute(
'SELECT * FROM read_history ORDER BY readtime DESC;')
cur.row_factory = sqlite3.Row
return cur.fetchall()