run.py

import requests
import random
import string
import argparse

from bs4 import BeautifulSoup as BS
from user_agent import generate_user_agent


def setup_argparse():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-n", "--count", type=int, default=100,
        help="number of images that will be scraped"
    )
    parser.add_argument(
        "-c", "--clear", action="store_true", default=False,
        help="clear saved images"
    )
    return parser.parse_args()


def string_generator(length):
    while True:
        chars_list = string.ascii_lowercase + string.digits
        comb_list = random.choices(chars_list, k=length)
        yield "".join(comb_list)


def get_image_by_string(string):
    url = "https://prnt.sc/" + string
    headers = {"user-agent": generate_user_agent()}

    response = requests.get(url, headers=headers)
    soup = BS(response.text, "html.parser")

    image_html = soup.find("img", class_="screenshot-image")

    if image_html:
        image_url = image_html.get("src")

        if "http" in image_url:
            return image_url


def main():
    """ Main entry point """

    # set up argparse
    args = setup_argparse()

    # scraping
    file_mode = "w" if args.clear else "a"

    print("Target:", args.count)
    print("Clear:", str(args.clear), end="\n")

    with open("images.html", file_mode) as f:
        counter = 0
        str_gen = string_generator(6)

        for some_str in str_gen:
            image = get_image_by_string(some_str)
            if not image:
                continue

            counter += 1

            print(f"{counter}. image:\t{image}")

            f.write(f'<img src="{image}" style="border: 3px solid red">')

            if counter >= args.count:
                break


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        print("\nExiting.")
        exit()