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'})