manga-dlp/mangadexdlp/main.py

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('\n=========================================')
  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{", ".join(manga_chapter_list)}')
    print('=========================================\n')
    exit(0)

  # 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)

  # show chapters to download
  print(f'Chapters selected:\n{", ".join(chapters_to_download)}')
  print('=========================================\n')

  # 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 index of chapter
    chapter_index = next(c for c in manga_chapter_data if c[0] == chapter)
    image_urls = MdUtils.get_img_urls(chapter_index)
    chapter_num = chapter_index[0]
    chapter_name = chapter_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(f'Cant download chapter {chapter_num}. Exiting')
      exit(1)
    else:
      # Done with chapter
      print(f'Successfully downloaded chapter {chapter_num}')

    # make cbz of folder
    if not md_nocbz:
      print('\nCreating .cbz archive')
      try:
        MdUtils.make_archive(chapter_path)
      except:
        print('Could not make cbz archive')
        exit(1)
      else:
        print('Done\n')
        print('------------------------------\n')