Standalone link grabber

get_links.py

def get_links(browser, tag, logger, media, element):
    # Get image links in scope from tags
    link_elems = element.find_elements_by_tag_name('a')
    sleep(2)
    links = []
    try:
        if link_elems:
            new_links = [link_elem.get_attribute('href') for link_elem in link_elems
                     if link_elem and link_elem.text in media]
            links.extend(new_links)
        else:
            logger.info("'{}' tag does not contain a picture".format(tag[1:] if tag[:1] == '#' else tag))
    except BaseException as e:
        logger.error("link_elems error {}".format(str(e)))
    return links

Click on RAW and copy