From eae42c8eb545c50012f89b777af7ad913f6e069d Mon Sep 17 00:00:00 2001 From: ricterz Date: Wed, 11 Dec 2024 23:57:01 +0800 Subject: [PATCH] fix #356 --- nhentai/downloader.py | 6 +++--- nhentai/parser.py | 1 - 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/nhentai/downloader.py b/nhentai/downloader.py index 3079002..e127bb3 100644 --- a/nhentai/downloader.py +++ b/nhentai/downloader.py @@ -64,8 +64,7 @@ class Downloader(Singleton): filename = filename if filename else os.path.basename(urlparse(url).path) base_filename, extension = os.path.splitext(filename) - digits = int(math.log10(length)) + 1 - filename = base_filename.zfill(digits) + extension + filename = base_filename.zfill(length) + extension save_file_path = os.path.join(self.folder, filename) @@ -153,8 +152,9 @@ class Downloader(Singleton): # Assuming we want to continue with rest of process. return True + digit_length = len(str(len(queue))) coroutines = [ - self._semaphore_download(url, filename=os.path.basename(urlparse(url).path), length=len(queue)) + self._semaphore_download(url, filename=os.path.basename(urlparse(url).path), length=digit_length) for url in queue ] diff --git a/nhentai/parser.py b/nhentai/parser.py index 9609acd..07818ae 100644 --- a/nhentai/parser.py +++ b/nhentai/parser.py @@ -135,7 +135,6 @@ def doujinshi_parser(id_, counter=0): logger.warning(f'Error: {e}, ignored') return None - # print(response) html = BeautifulSoup(response, 'html.parser') doujinshi_info = html.find('div', attrs={'id': 'info'})