frenos/imgload

import re

from requests_html import HTMLSession

from imgload.img_hosts.ImageHost import ImageHost


class PostimgCC(ImageHost):

    def __init__(self):
        super(ImageHost, self).__init__()
        self.base_url = "postimg.cc"
        self.regex = r'\A(http*(s)://postimg.cc/)+.'

    def can_handle(self, link):
        result = re.search(self.regex, link)
        if result is None:
            return False
        return True

    def get_link(self, link, session: HTMLSession):
        resp = session.get(link)

        resp.html.render()
        if resp.status_code is not 200:
            return None

        url = resp.html.xpath("//img[contains(@id, 'main-image')]")
        if url is not None:
            return url[0].attrs.get('src')

        return None