SylvainDe / ComicBookMaker

class ButterSafe(GenericListableComic):

    """Class to retrieve Butter Safe comics."""

    name = 'butter'

    long_name = 'ButterSafe'

    url = 'http://buttersafe.com'

    get_url_from_archive_element = get_href

    comic_link_re = re.compile('^%s/([0-9]*)/([0-9]*)/([0-9]*)/.*' % url)

    @classmethod

    def get_archive_elements(cls):

        archive_url = urljoin_wrapper(cls.url, 'archive/')

        return reversed(get_soup_at_url(archive_url).find_all('a', href=cls.comic_link_re))

    @classmethod

    def get_comic_info(cls, soup, link):

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

        url = cls.get_url_from_archive_element(link)

        title = link.string

        year, month, day = [int(s) for s in cls.comic_link_re.match(url).groups()]

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

        assert img['alt'] == title

        return {

            'title': title,

            'day': day,

            'month': month,

            'year': year,

            'img': [img['src']],

        }

class CalvinAndHobbes(GenericComic):

    """Class to retrieve Calvin and Hobbes comics."""

    # Also on http://www.gocomics.com/calvinandhobbes/

class AbstruseGoose(GenericListableComic):

    """Class to retrieve AbstruseGoose Comics."""

    name = 'abstruse'

    long_name = 'Abstruse Goose'

    url = 'http://abstrusegoose.com'

    get_url_from_archive_element = get_href

    comic_url_re = re.compile('^%s/([0-9]*)$' % url)

    comic_img_re = re.compile('^%s/strips/.*' % url)

    @classmethod

    def get_archive_elements(cls):

        archive_url = urljoin_wrapper(cls.url, 'archive')

        return get_soup_at_url(archive_url).find_all('a', href=cls.comic_url_re)

    @classmethod

    def get_comic_info(cls, soup, archive_elt):

        comic_url = cls.get_url_from_archive_element(archive_elt)

        num = int(cls.comic_url_re.match(comic_url).groups()[0])

        return {

            'num': num,

            'title': archive_elt.string,

            'img': [soup.find('img', src=cls.comic_img_re)['src']]

        }

class PhDComics(GenericNavigableComic):

    """Class to retrieve PHD Comics."""

    name = 'phd'

class PoorlyDrawnLines(GenericListableComic):

    """Class to retrieve Poorly Drawn Lines comics."""

    # Also on http://pdlcomics.tumblr.com

    name = 'poorlydrawn'

    long_name = 'Poorly Drawn Lines'

    url = 'https://www.poorlydrawnlines.com'

    _categories = ('POORLYDRAWN', )

    get_url_from_archive_element = get_href

    @classmethod

    def get_comic_info(cls, soup, link):

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

        imgs = soup.find('div', class_='post').find_all('img')

        assert len(imgs) <= 1

        return {

            'img': [i['src'] for i in imgs],

            'title': imgs[0].get('title', "") if imgs else "",

        }

    @classmethod

    def get_archive_elements(cls):

        archive_url = urljoin_wrapper(cls.url, 'archive')

        url_re = re.compile('^%s/comic/.' % cls.url)

        return reversed(get_soup_at_url(archive_url).find_all('a', href=url_re))

class LoadingComics(GenericNavigableComic):

    """Class to retrieve Loading Artist comics."""

    name = 'loadingartist'

    long_name = 'Loading Artist'