From 16bac45f023f6c53da4848e856ce59aab3ae3638 Mon Sep 17 00:00:00 2001 From: ricterz Date: Sun, 22 Sep 2024 12:30:55 +0800 Subject: [PATCH] generate html viewer automatically after download #342 --- nhentai/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nhentai/utils.py b/nhentai/utils.py index d1fa5fe..c95e8a7 100644 --- a/nhentai/utils.py +++ b/nhentai/utils.py @@ -77,8 +77,8 @@ def parse_doujinshi_obj( filename = './doujinshi' + file_type already_downloaded = False - if doujinshi_obj is not None: - doujinshi_dir = os.path.join(output_dir, doujinshi_obj.filename) + doujinshi_dir = os.path.join(output_dir, doujinshi_obj.filename) + if doujinshi_obj is not None and file_type != '.html': if os.path.exists(doujinshi_dir + file_type): already_downloaded = True elif file_type != '': @@ -96,7 +96,7 @@ def parse_doujinshi_obj( def generate_html(output_dir='.', doujinshi_obj=None, template='default'): - doujinshi_dir, filename, already_downloaded = parse_doujinshi_obj(output_dir, doujinshi_obj) + doujinshi_dir, filename, already_downloaded = parse_doujinshi_obj(output_dir, doujinshi_obj, '.html') if already_downloaded: logger.info(f'Skipped download: {doujinshi_dir} already exists') return