dosage/dosagelib/plugins/studiokhimera.py

# SPDX-License-Identifier: MIT
# Copyright (C) 2019-2020 Tobias Gruetzmacher
# Copyright (C) 2019-2020 Daniel Ring
from ..scraper import _ParserScraper
from .common import _WordPressScraper


class StudioKhimera(_ParserScraper):
    imageSearch = '//figure[@class="gallery-item"]//img/@data-src'
    prevSearch = '//a[@rel="prev"]'

    def __init__(self, name, sub, last=None, adult=False, fixNames=False):
        super(StudioKhimera, self).__init__('StudioKhimera/' + name)

        self.baseUrl = 'https://%s.studiokhimera.com/' % sub
        self.stripUrl = self.baseUrl + '%s/'
        self.url = self.baseUrl + 'category/comicChapter/?latest'

        self.multipleImagesPerStrip = True

        if last:
            self.last = True
            self.url = self.stripUrl % last
            self.endOfLife = True

        if adult:
            self.adult = True

    def starter(self):
        # Retrieve list of chapter links
        chapterPage = self.getPage(self.baseUrl + 'archive/')
        self.chapters = chapterPage.xpath('//main//a/@href')
        self.firstStripUrl = self.chapters[0]
        return self.chapters[-1]

    def getPrevUrl(self, url, data):
        # Select previous chapter from list
        index = [i for i, ch in enumerate(self.chapters) if ch == url][0]
        if index == 0:
            return None
        return self.chapters[index - 1]

    @classmethod
    def getmodules(cls):
        return (
            cls('Eorah', 'eorah'),
            cls('Mousechievous', 'mousechievous'),
        )


class UberQuest(_WordPressScraper):
    name = 'StudioKhimera/UberQuest'
    stripUrl = 'https://uberquest.studiokhimera.com/comic/page/%s/'
    url = stripUrl % 'latest'
    firstStripUrl = stripUrl % 'cover'
    imageSearch = '//div[@class="prj--comic-image"]/img'
    prevSearch = '//uq-image-button[d:class("prj--comic-control-prev")]'

    def namer(self, imageUrl, pageUrl):
        # Fix inconsistent filenames
        filename = imageUrl.rsplit('/', 1)[-1]
        filename = filename.replace('Page', 'UberQuest')
        filename = filename.replace('UberQuest01.', 'UberQuest001.')
        filename = filename.replace('UberQuest98.', 'UberQuest098.')
        filename = filename.replace('UberQuest99.', 'UberQuest099.')
        return filename