SylvainDe / ComicBookMaker

class HorovitzNew(HorovitzComics):

    """Class to retrieve Horovitz new comics."""

    name = 'horovitznew'

    long_name = 'Horovitz New'

    link_re = re.compile('^/comics/new/([0-9]+)$')

class HorovitzClassic(HorovitzComics):

    """Class to retrieve Horovitz classic comics."""

    name = 'horovitzclassic'

    long_name = 'Horovitz Classic'

    link_re = re.compile('^/comics/classic/([0-9]+)$')

class GenericGoComic(GenericNavigableComic):

    """Generic class to handle the logic common to comics from gocomics.com."""

    _categories = ('GOCOMIC', )

    @classmethod

    def get_first_comic_link(cls):

        """Get link to first comics."""

        return get_soup_at_url(cls.url).find('a', class_='fa btn btn-outline-default btn-circle fa-backward sm ')

    get_navi_link = get_link_rel_next

    get_url_from_link = join_cls_url_to_href

    @classmethod

    def get_comic_info(cls, soup, link):

        """Get information about a particular comics."""

        short_url_re = re.compile('^%s/\\?p=([0-9]*)' % cls.url)

        short_url = cls.get_url_from_link(soup.find('link', rel='shortlink'))

        num = int(short_url_re.match(short_url).groups()[0])

        imgs = soup.find('div', id='comic').find_all('img')

        assert len(imgs) == 1

        title = imgs[0]['alt']

        title2 = imgs[0]['title']

        return {

            'short_url': short_url,

            'title': title,

            'title2': title2,

            'img': [urljoin_wrapper(cls.url, i['src']) for i in imgs],

            'num': num,

        }