import random
import string
import requests

def get_image(boards, nsfw=False):
    """
    get_image(boards, nsfw=False)

    Returns a URL to an image on reddit from a random board in the boards list
    as long as it is hosted on one of the domains in the domains list
    """

    if isinstance(boards, list):
        boards = random.choice(boards)

    domains = [
        'cdn.awwni.m',
        'gfycat.com',
        'i.imgur.com',
        'i.redd.it',
        'i.reddituploads.com',
        'i3.kym-cdn.com',
        'imgur.com',
        'media.giphy.com',
    ]
    print("chose %s" % boards)
    request_string = "https://reddit.com/r/{}/random.json".format(boards)

    if not nsfw:
        # Append this header to the request. Tells the API to only return SFW results
        request_string += '?obey_over18=true'

    # Spoof our user agent with each request so we dont get rate limited
    random_user_agent = ''.join(
        random.SystemRandom().choice(
            string.ascii_uppercase + string.digits
        ) for _ in range(5)
    )

    response = requests.get(
        request_string,
        headers = {'User-agent':random_user_agent}
    ).json()[0]['data']['children'][0]

    if response['data']['domain'] not in domains:
        # If we dont find an approved domain, re-try the request
        return get_image(boards, nsfw=nsfw)

    return response['data']['url'].replace('http://', 'https://')