manga-dlp/mangadlp/api/mangadex.py

import requests
import re
import mangadlp.utils as MUtils


class Mangadex():

  # api information
  api_base_url = 'https://api.mangadex.org'
  img_base_url = 'https://uploads.mangadex.org'


  # get infos to initiate class
  def __init__(self, manga_url, manga_lang):
    self.manga_url = manga_url
    self.manga_lang = manga_lang
    self.manga_uuid = self.get_manga_uuid()
    self.manga_title = self.get_manga_title(self.manga_uuid)
    self.manga_chapter_data = self.get_manga_chapters(self.manga_uuid)


  # get the uuid for the manga
  def get_manga_uuid(self):
    # isolate id from url
    uuid_regex = re.compile('[a-z0-9]{8}-[a-z0-9]{4}-[a-z0-9]{4}-[a-z0-9]{4}-[a-z0-9]{12}')
    # check for new mangadex id
    if uuid_regex.search(self.manga_url):
      manga_uuid = uuid_regex.search(self.manga_url)[0]
    else:
      print('No valid uuid found')
      exit(1)
    # check if the manga exists
    try:
      req = requests.get(f'{self.api_base_url}/manga/{manga_uuid}')
    except:
      print('Error. Maybe the MangaDex API is down?')
      exit(1)
    else:
      # check mangadex status
      response = req.json()['result']
      if not response == 'ok':
        print('Manga not found')
        exit(1)

      return manga_uuid


  # get the title of the manga (and fix the filename)
  def get_manga_title(self, manga_uuid):
    req = requests.get(f'{self.api_base_url}/manga/{manga_uuid}')
    api_resp = req.json()
    try:
      title = api_resp['data']['attributes']['title'][self.manga_lang]
    except:
      # search in alt titles
      try:
        alt_titles = {}
        for title in api_resp['data']['attributes']['altTitles']:
          alt_titles.update(title)
        title = alt_titles[self.manga_lang]
      except: # no title on requested language found
        print('Chapter in requested language not found.')
        exit(1)

    return MUtils.fix_name(title)


  # get all chapter data for further parsing
  def get_manga_chapters(self, manga_uuid):
    content_ratings = 'contentRating[]=safe&contentRating[]=suggestive&contentRating[]=erotica&contentRating[]=pornographic'
    chap_data_list = []
    req = requests.get(f'{self.api_base_url}/manga/{manga_uuid}/feed?limit=0&translatedLanguage[]={self.manga_lang}&{content_ratings}')
    try:
      total = req.json()['total']
    except:
      print('Error retrieving the chapters list. Did you specify a valid language code?')
      exit(1)
    if total == 0:
      print('No chapters available to download!')
      exit(0)
    last_chap = ['', '']
    offset = 0
    while offset < total: # if more than 500 chapters
      req = requests.get(f'{self.api_base_url}/manga/{manga_uuid}/feed?order[chapter]=asc&order[volume]=asc&limit=500&translatedLanguage[]={self.manga_lang}&offset={offset}&{content_ratings}')
      for chapter in req.json()['data']:
        chap_num = chapter['attributes']['chapter']
        chap_vol = chapter['attributes']['volume']
        chap_uuid = chapter['id']
        chap_hash = chapter['attributes']['hash']
        chap_data = chapter['attributes']['data']
        chap_name = chapter['attributes']['title']
        if not chap_name == None:
          chap_name = MUtils.fix_name(chap_name)
        # check if the chapter is external (cant download them)
        chap_external = chapter['attributes']['externalUrl']
        # name chapter "oneshot" if there is no chapter number
        if chap_external == None and chap_num == None:
          # check for duplicates
          if last_chap[0] == chap_vol and last_chap[1] == chap_num:
            continue
          chap_data_list.append([chap_vol, 'Oneshot', chap_uuid, chap_hash, chap_name, chap_data])
        # else add chapter number
        elif chap_external == None:
          # check for duplicates
          if last_chap[0] == chap_vol and last_chap[1] == chap_num:
            continue
          chap_data_list.append([chap_vol, chap_num, chap_uuid, chap_hash, chap_name, chap_data])
        last_chap = [chap_vol, chap_num]
      offset += 500

    return chap_data_list


  def get_chapter_index(self, chapter, forcevol):
    # get index of chapter
    if forcevol:
      chapter_index = next(c for c in self.manga_chapter_data if f'{c[0]}:{c[1]}' == chapter)
    else:
      chapter_index = next(c for c in self.manga_chapter_data if c[1] == chapter)

    return chapter_index


  # create list of chapters
  def create_chapter_list(self, chapter_data, forcevol):
    chapter_list = []
    for chap in chapter_data:
      volume_number = chap[0]
      chapter_number = chap[1]
      if forcevol:
        chapter_list.append(f'{volume_number}:{chapter_number}')
      else:
        chapter_list.append(chapter_number)

    return chapter_list


  # get list of image urls
  def get_img_urls(self, images, chapter_hash):
    img_urls = []
    for img in images:
      img_urls.append(f'{self.img_base_url}/data/{chapter_hash}/{img}')

    return img_urls