Phill Spiess

# Our imports. Nothing new or interesting here
import sys, time, os
from mechanize import Browser
from BeautifulSoup import BeautifulSoup
 
## The url for the logon page
LOGON_URL = 'http://webdiplomacy.net/logon.php'
 
## The url for the board you want to watch. You'll need to put in the game ID
BOARD_URL = 'http://webdiplomacy.net/board.php?gameID='
 
## Webdiplomacy login information
USERNAME= ''
PASSWORD= ''
 
## Define a simple class to hold the status of a country
class country:
    def __init__(self, cid, status, online):
        self.countryID = cid
        self.status = status
        self.online = online
 
## Our main function
def fetch():
    # Create a browser object. This is a Mechanize class that simulates
    # a fully featured browser
    br = Browser()
 
    # Tell our browser to open a URL. In this case, navigate to the logon page
    br.open(LOGON_URL)
 
    # Tell our browser to navigate to the first (and only) form on the page
    br.select_form(nr=0)
 
    # This form has two input boxes, populate them with our username and password
    br['loginuser'] = USERNAME
    br['loginpass'] = PASSWORD
 
    # Click the submit button, and ignore the response
    br.submit()
 
    # Now that we are cookied (we've logged in) navigate to the board to watch
    br.open(BOARD_URL)
 
    # And tell our browser to load the page, storing the response we download in resp
    resp = br.reload()
 
    # Response is the full web response, we just care about the HTML portion, so save it off
    html = resp.read();
 
    # Now, we need to create a BeautifulSoup object. A BeautifulSoup object is a
    # tree based way to navigate an html page. So we can get the  tag
    # and navigate through its children  and  tags. (For example)
    # We will be using it to search for specific things on the page, like the ready icon.
    soup = BeautifulSoup(''.join(html))
 
    # We begin by looking for a div, whose id is chatboxtabs. soup.find() returns the first
    # result
    chatboxtabs = soup.find('div', {'id':'chatboxtabs'})
 
    # Set up an empty dictionary to store the status of each country using the ID as the key
    onlineDict = {}
 
    # The empty list of countries to return
    returnList = []
    # The country doing the checking is not in the list of chatboxes, so we have to record
    # its status separately
    missingID = 1
 
    # For each anchor (a) tag inside our div
    for chatbox in chatboxtabs('a'):
        # If the anchor tag has a span within it
        if(chatbox('span') != []):
            # Then prune out the country id. These spans have a class that looks like:
            # "country 3 memberstatusplaying"
            # All we care about is the integer in the middle
            theID = chatbox('span')[0]['class'].replace("country", "").split(" ")[0]
 
            # If the ID we read out is not the next id in order,
            # increment it. (This means missingID will stop at the scipter's country ID)
            if (theID == str(missingID)):
                missingID += 1
 
            # If the country is online, then there is an image next to their name
            # Store the status for that country
            if (len(chatbox('img')) > 0):
                onlineDict[theID] = True
            else:
                onlineDict[theID] = False
 
    # Store the status of the scripting country
    onlineDict[str(missingID)] = True
 
    # Now we look for their readiness status. Each country has a td with a class
    # 'memberLeftSide'. We grab them all with soup.findAll
    details = soup.findAll('td', {'class':'memberLeftSide'})
 
    # Iterate through all the td elements
    for status in details:
        # There are a few we can skip because they have no images
        if (status.img != None):
            # The country ID is grabbed, just like the anchor tags above
            countryID = status.span.contents[2]['class']
            countryID = countryID.replace("country", "")
            countryID = countryID.split(" ")[0]
 
            # Get the alternate text for the image
            statStr = status.img['alt']
            statInt = 2 # Assume 'bad' status
 
            # Set our status integer based on the alt text
            if (statStr == "Not received"):
                statInt = 1
            elif (statStr == "Ready"):
                statInt = 4
            elif (statStr == "Completed"):
                statInt = 3
 
            # Create a country object and store it in our return list
            returnList.append(country(countryID, statInt, onlineDict[countryID]))
 
    # Return our country list
    return returnList