From f980b45703ad7e213bca0c49f620fb80e120f526 Mon Sep 17 00:00:00 2001
From: lifegpc <root@lifegpc.com>
Date: Fri, 8 Mar 2024 22:44:36 +0800
Subject: [PATCH] Bug fix

---
 export.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/export.py b/export.py
index c3476ef..dfca0e9 100644
--- a/export.py
+++ b/export.py
@@ -81,14 +81,17 @@ def export_book(ncw: NovelCiwei, db: CwmDb, cfg: Config, bn: BooksNew,
                 maps[division_id] = [chapter]
         for division in divisions:
             division_name = division['division_name']
+            division_id = division['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['division_id']]:
+            for chapter in maps[division_id]:
                 chapter_id = chapter['chapter_id']
                 chapter_title = chapter['chapter_title']
                 if chapter['is_download']: