Ivan Schaller
9f6bbbb9a5
All checks were successful
continuous-integration/drone/push Build is passing
113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
import pdb
|
|
import argparse
|
|
from time import sleep
|
|
import requests
|
|
import shutil
|
|
from pathlib import Path
|
|
import mangadexdlp.api as MdApi
|
|
import mangadexdlp.utils as MdUtils
|
|
import mangadexdlp.downloader as MdDownloader
|
|
import mangadexdlp.sqlite as MdSqlite
|
|
|
|
|
|
|
|
def mangadex_dlp(md_url='',md_chapters=None,md_dest='downloads',md_lang='en',md_list_chapters=False,md_nocbz=False):
|
|
'''Download Mangas from Mangadex.org\n
|
|
|
|
Args:\n
|
|
url (str) -- Manga URL to Download. No defaults\n
|
|
chapter (str/int) -- Chapters to download "all" for every chapter available. Defaults to none\n
|
|
dest (str) -- Folder to save mangas to. Defaults to "downloads"\n
|
|
lang (str) -- Language to download chapters in. Defaults to "en" -> english\n
|
|
list (bool) -- If it should only list all available chapters. Defaults to False\n
|
|
nocbz (bool) -- If the downloaded images should not be packed into a .cbz archive. Defaults to false\n
|
|
|
|
Returns:\n
|
|
nothing\n
|
|
'''
|
|
# check if md_list_chapters is true, if not check if chapters to download were specified
|
|
if not md_list_chapters and md_chapters == None:
|
|
# no chapters to download were given
|
|
print(f'You need to specify one or more chapters to download. To see all chapters use "--list"')
|
|
exit(1)
|
|
|
|
# get uuid and manga name of url
|
|
manga_uuid = MdApi.get_manga_uuid(md_url)
|
|
manga_title = MdApi.get_manga_title(manga_uuid, md_lang)
|
|
|
|
print(f'Manga Name: {manga_title}')
|
|
print(f'UUID: {manga_uuid}')
|
|
|
|
# get chapters
|
|
manga_chapter_data = MdApi.get_manga_chapters(manga_uuid, md_lang)
|
|
# [0][0] = Chapter number/oneshot
|
|
# [0][1] = Chapter UUID
|
|
# [0][2] = Chapter Hash
|
|
# [0][3] = Chapter Name
|
|
# [0][4] = Chapter Data
|
|
|
|
# crate chapter list
|
|
manga_chapter_list = []
|
|
for chap in manga_chapter_data:
|
|
chapter_number = chap[0]
|
|
manga_chapter_list.append(chapter_number)
|
|
|
|
# list chapters if md_list_chapters is true
|
|
if md_list_chapters:
|
|
print(f'Available Chapters:\n{manga_chapter_list}')
|
|
|
|
# check chapters to download if it not all
|
|
chapters_to_download = []
|
|
if md_chapters.lower() == 'all':
|
|
chapters_to_download = manga_chapter_list
|
|
else:
|
|
chapters_to_download = MdUtils.get_chapter_list(md_chapters)
|
|
|
|
# create manga folder
|
|
manga_path = Path(f'{md_dest}/{manga_title}')
|
|
manga_path.mkdir(parents=True, exist_ok=True)
|
|
|
|
# main download loop
|
|
for chapter in chapters_to_download:
|
|
# get list of image urls
|
|
list_index = manga_chapter_data.index(chapter)
|
|
image_urls = MdUtils.get_img_urls(manga_chapter_data[list_index])
|
|
chapter_num = manga_chapter_data[list_index][0]
|
|
chapter_name = manga_chapter_data[list_index][3]
|
|
|
|
# filename for chapter
|
|
if chapter_name == None and chapter_num == 'Oneshot':
|
|
chapter_filename = 'Oneshot'
|
|
elif chapter_name == None:
|
|
chapter_filename = f'Ch. {chapter_num}'
|
|
else:
|
|
chapter_filename = f'Ch. {chapter_num} - {chapter_name}'
|
|
|
|
# create chapter folder
|
|
chapter_path = manga_path / chapter_filename
|
|
chapter_path.mkdir(parents=True, exist_ok=True)
|
|
|
|
# download images
|
|
print(f'Downloading Chapter {chapter_num}')
|
|
print(f'DEBUG: Downloading Chapter {chapter}')
|
|
try:
|
|
MdDownloader.download_chapter(image_urls, chapter_path)
|
|
except:
|
|
print('Cant download chapter. Exiting')
|
|
exit(1)
|
|
else:
|
|
# Done with chapter
|
|
print(f'--Done with Chapter {chapter_num}')
|
|
|
|
# make cbz of folder
|
|
if not md_nocbz:
|
|
print('Creating .cbz archive')
|
|
try:
|
|
MdUtils.make_archive(chapter_path, manga_path)
|
|
except:
|
|
print('Could not make cbz archive')
|
|
exit(1)
|
|
else:
|
|
print('Done')
|
|
|
|
|