Search options

#!/usr/bin/env python

"""
mafiawars.py - A python script to play mafiawars when not on the computer.

Usage:
    mafiawars.py -j jobnumber -J jobnumbers [-u facebook login emal] [-p password] [-l link] [-n] [-c configfile]

requires:
    mechanize
"""

import optparse
import os
import sys
import ConfigParser
from mechanize import Browser

class AccountError(Exception): pass

#Play mafia wars while away
def checkOptions():
    config = os.path.expanduser(os.path.join('~', '.mafiawarsrc'))
    op     = optparse.OptionParser()

    op.add_option('-c', '--config',   dest='confpath', default=config)
    op.add_option('-u', '--username', dest='username', default=None) # facebook login email
    op.add_option('-p', '--password', dest='password', default=None)
    op.add_option('-l', '--link_to_follow',   dest='link')
    op.add_option('-n', '--no-save-conf', dest='save_conf', default=True, action='store_false')
    op.add_option('-j', '--job', dest='job', default=None,
                 help="Job number. Login to mafia wars and inspect the " + \
                  "\"Do Job\" link.  The job number is towards the end \"http:\\\\...&job=24\"")
    op.add_option('-J', '--job_list', dest='job_list', default=None, help="comma separated list of job numbers. (no spaces please)")

    options, args = op.parse_args()
    options.msg = ' '.join(args)

    cp = ConfigParser.RawConfigParser()
    cp.add_section('account')
    cp.add_section('links_to_follow')
    cp.add_section('jobs')
    cp.read([options.confpath])

    try:
        if options.password is None:
            options.password = cp.get('account', 'password')
        if options.username is None:
            options.username = cp.get('account', 'username')
    except ConfigParser.NoOptionError:
        raise AccountError()

    if options.job is None and options.job_list is None:
        print >>sys.stderr, 'No job or jobs specified please use ' + \
                            'with -j or --job_list.\n'
        sys.exit(2)

    if options.username is None or options.password is None:
        raise AccountError() 

    return options, cp

def saveConfig(options, cp):
    if not cp.has_option('account', 'username'):
        cp.set('account', 'username', options.username)
    if not cp.has_option('account', 'password'):
        cp.set('account', 'password', options.password)
    if not cp.has_option('links_to_follow', 'link'):
        cp.set('links_to_follow', 'link', options.link)

    cp.write(file(options.confpath, 'w'))
                  
def main():
    try:
        options, cp = checkOptions()
    except AccountError:
        print >>sys.stderr, 'No account specified. Please run mafiawars once ' + \
                            'with --username and --password\n' + \
                            'to create your ~/.mafiawarsrc.'
        sys.exit(2)

    b = Browser()
    try:
        #log into facebook
        b.open("http://www.facebook.com")
        b.select_form(nr=0)
        b["email"] = options.username
        b["pass"] = options.password
        b.submit()

        # visit the provided link
        if options.link:
            b.open(options.link)
        elif options.job_list:
            for j in options.job_list.split(','):
                url = "http://apps.facebook.com/inthemafia/remote/html_server.php?xw_controller=job&xw_action=dojob&job=%s" % j
                b.open(url)
        else:
            url = "http://apps.facebook.com/inthemafia/remote/html_server.php?xw_controller=job&xw_action=dojob&job=%s" % options.job 
            b.open(url)

        b.open("http://www.facebook.com/logout.php")
        b.close()

        if options.save_conf:
            saveConfig(options, cp)
    except KeyboardInterrupt:
        sys.exit(130)

if __name__ == "__main__":
    main()