SylvainDe / ComicBookMaker

class GenericSquareSpace(GenericNavigableComic):

    """Generic class to retrieve comics using SquareSpace."""

    _categories = ('SQUARESPACE', )

    get_url_from_link = join_cls_url_to_href

    get_first_comic_link = simulate_first_link

    @classmethod

    def get_navi_link(cls, last_soup, next_):

        """Get link to next or previous comic."""

        return last_soup.find('a', id='prevLink' if next_ else 'nextLink')

    @classmethod

    def get_images(cls, soup):

        """Get image URLs for a comic."""

        raise NotImplementedError

    @classmethod

    def get_comic_info(cls, soup, link):

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

        title = soup.find('meta', property='og:title')['content']

        desc = soup.find('meta', property='og:description')['content']

        date_str = soup.find('time', itemprop='datePublished')["datetime"]

        day = string_to_date(date_str, "%Y-%m-%d")

        author = soup.find('a', rel='author').string

        return {

            'title': title,

            'img': cls.get_images(soup),

            'month': day.month,

            'year': day.year,

            'day': day.day,

            'author': author,

            'description': desc,

        }

class AtRandomComics(GenericSquareSpace):

    """Class to retrieve At Random Comics."""

    name = 'atrandom'

class ConsoliaComics(GenericNavigableComic):

    """Class to retrieve Consolia comics."""

    name = 'consolia'

    long_name = 'consolia'

    url = 'https://consolia-comic.com'

    get_url_from_link = join_cls_url_to_href

    @classmethod

    def get_first_comic_link(cls):

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

        return get_soup_at_url(cls.url).find('a', class_='first')

    @classmethod

    def get_navi_link(cls, last_soup, next_):

        """Get link to next or previous comic."""

        return last_soup.find('a', class_='next' if next_ else 'prev')

    @classmethod

    def get_comic_info(cls, soup, link):

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

        title = soup.find('meta', property='og:title')['content']

        date_str = soup.find('time')["datetime"]

        day = string_to_date(date_str, "%Y-%m-%d")

        imgs = soup.find_all('meta', property='og:image')

        return {

            'title': title,

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

            'day': day.day,

            'month': day.month,

            'year': day.year,

        }

class GenericBlogspotComic(GenericNavigableComic):

    """Generic class to retrieve comics from Blogspot."""

    get_first_comic_link = simulate_first_link

class PomComics(GenericNavigableComic):

    """Class to retrieve PomComics."""

    name = 'pom'

    long_name = 'Pom Comics / Piece of Me'

    url = 'http://www.pomcomic.com'

    get_url_from_link = join_cls_url_to_href

    @classmethod

    def get_first_comic_link(cls):

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

        return get_soup_at_url(cls.url).find('a', class_='btn-first')

    @classmethod

    def get_navi_link(cls, last_soup, next_):

        """Get link to next or previous comic."""

        return last_soup.find('a', class_='btn-next' if next_ else 'btn-prev')

    @classmethod

    def get_comic_info(cls, soup, link):

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

        title = soup.find('h1').string

        desc = soup.find('meta', property='og:description')['content']

        tags = soup.find('meta', attrs={'name': 'keywords'})['content']

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

        return {

            'title': title,

            'desc': desc,

            'tags': tags,

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

        }

class CubeDrone(GenericComicNotWorking, GenericNavigableComic):  # Website has changed

    """Class to retrieve Cube Drone comics."""

    name = 'cubedrone'