import wikipedia
import tweepy
import random
import urllib.request

def twitter_api():
    twitter_consumer_key = ''
    twitter_consumer_secret = ''

    twitter_access_token = ''
    twitter_access_secret = ''

    auth = tweepy.OAuthHandler(twitter_consumer_key, twitter_consumer_secret)
    auth.set_access_token(twitter_access_token, twitter_access_secret)
    api = tweepy.API(auth)
    return api

def get_pages_and_pictures():
    #get two random wikipedia articles
    random_pages = wikipedia.random(2)
    #init empty list
    pages_and_pictures = []

    for page in random_pages:
        #set picture to first image on page
        picture_url = wikipedia.page(page).images[0]
        #append both page name & picture to our list
        pages_and_pictures.append(page)
        pages_and_pictures.append(picture_url)
        #download image with JPG extenstion
        #attempt to ensure photo is jpg and article has enough images
        while True:
            if picture_url[-4] == '.jpg' and len(wikipedia.page(page).images) > 2:
                urllib.request.urlretrieve(picture_url, page + picture_url[-4:])
                break
            else:
                picture_url = wikipedia.page(wikipedia.random(1)).images[0]

    #return our list
    return pages_and_pictures

def assemble_tweet(tweet_assembly_list):
    tweet_string = ''
    tweet_string += tweet_assembly_list[0] + ', ' + tweet_assembly_list[2]
    return tweet_string

while True:
    try:
        api = twitter_api()

        tweet_list = get_pages_and_pictures()
        final_tweet = assemble_tweet(tweet_list)
        print(final_tweet)
        api.update_status(final_tweet)
        break
    except wikipedia.exceptions.DisambiguationError as e:
        print(e.options)