Subversion Repositories Remote Hare Voting

Rev

Rev 63 | Rev 67 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed 

#!/p/python-2.5.1/bin/python

import optparse
import cgi
import string
import sys

import parseinput
import tally

fHTML = 1
formaction = string.split(sys.argv[0],"/")[-1]

def emitPreHTML(formaction):
    emitHTML("""Content-type: text/html\n
<html>
<body>
<title>Hare Ballot Calculation</title>
    <h1>Hare Ballot Calculation</h1>
    <form enctype="multipart/form-data" method=POST action=%s>
        <table>
        <tr><td>CSV file of ballots:
                <td><input type=file size=70 name=name>
        </table>
        <p>
        <input type=submit name="run1" value="Run 1 Time">
        <input type=submit name="run500" value="Run 500 Times">
        <input type=submit name="runparallel" value="Run All Ways">
    </form>
    <p>
Input file format is given
<a href="http://www.bikmort.com/wiki/index.php?title=Hare_Voting_Procedure">here</a>.
Little error checking is done, so be sure to review the trace output.
The source code is available <a href="%s?fetch=yes">here</a>.
<p>
<small>%s</small>
<hr>
<pre>
""" % (formaction, formaction, "$Id: vote.py 64 2007-10-15 15:48:23Z jtkorb $"))
    return

def emitPostHTML():
    emitHTML("""</pre>
</body>
</html>
""")
    return

def emitHTML(s):
    global fHTML
    if fHTML: print s
    return

def doone(nWinners, ballots):
    tally.traceLevel = 1
    tally.fParallel = 0
    results = tally.dotally(nWinners, ballots)

def domany(nTimes, nWinners, ballots):
    tally.traceLevel = 0
    tally.fParallel = 0
    summary = {}
    for i in range(nTimes):
        results = str(tally.dotally(nWinners, ballots))
        if not summary.has_key(results):
            summary[results] = 0
        summary[results] = summary[results] + 1
    tally.trace(0, "\nSUMMARY AFTER %d TIMES" % nTimes)
    l = summary.items()
    l.sort(lambda x,y: cmp(y[1], x[1]))
    for pair in l:
        tally.trace(0, "%3d %s" % (pair[1], pair[0]))

def doparallel(nWinners, ballots):
    tally.traceLevel = 0
    tally.fParallel = 1
    results = tally.dotally(nWinners, ballots)

def doweb():
    emitPreHTML(formaction)
    form = cgi.FieldStorage()

    if not form:
        emitHTML("Enter file name and press a button to see results.")
    elif form.has_key("fetch"):
        filename = string.replace(formaction, "vote.py", "tally.py")
        source = open(filename).read()
        print cgi.escape(source)
    elif not form.has_key("name"):
        emitHTML("Error: no file received")
    elif not form["name"].filename:
        emitHTML("Error: file name missing")
    else:
        name = form["name"].filename
        contents = form["name"].value

        tally.trace(0, "INPUT FILE")
        tally.trace(0, contents)

        nWinners, ballots = parseinput.parsestring(contents)
        if form.has_key("run1"):
            doone(nWinners, ballots)
        elif form.has_key("run500"):
            domany(500, nWinners, ballots)
        elif form.has_key("runparallel"):
            doparallel(nWinners, ballots)
        else:
            print "UNEXPECTED SUBMIT BUTTON: %s" % form
                
    emitPostHTML()
    return

def main():
    usage = "usage: %prog [options] filename.csv"
    parser = optparse.OptionParser(usage)
    parser.add_option("-m", "--multiple", dest="multi", action="store_true",
                      help="run process multiple (1000) times")
    parser.add_option("-p", "--parallel", dest="para", action="store_true",
                      help="run process in parallel universes")
    (options, args) = parser.parse_args()

    if len(args) == 0:
        doweb()
    else:
        nWinners, ballots = parseinput.parsefile(args[0])
        if options.multi:
            domany(1000, nWinners, ballots)
        elif options.para:
            doparallel(nWinners, ballots)
        else:
            doone(nWinners, ballots)

if __name__=='__main__':
    main()