voteCounter.py 文件源码

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('input')
    parser.add_argument("-e", "--perc_elim", nargs='?', const=5, default=5)
    parser.add_argument("-t", "--num_gold", nargs='?', const=5, default=5)
    args = parser.parse_args()
    path = args.input

    votes = convert(path)
    prompt = open('./twows/{}/prompt.txt'.format(path),'r').read().split('\n')[0]
    scores = []
    twowers=set()

    with open('./twows/{}/responses.csv'.format(path),'r') as csvfile:#read responses
        reader = csv.reader(csvfile)
        for row in reader:
            #scoredata format [twower, response, votes/mean, count, boost, final, stdev, votegraph]
            name = simplify(row[0])
            twowers.add(name)
            try:
                scores.append([name,row[1],[],0,int(row[2]),0,0,0,[0 for i in range(10)]])
            except:
                scores.append([name,row[1],[],0,0,0,0,0,[0 for i in range(10)]])

    twowers = list(twowers)
    twower_count = len(twowers)

    top_number = int(args.num_gold) #chart coloring ranges
    elim_number=0

    if int(args.perc_elim) < 0:
        elim_number = -int(args.perc_elim)
    else:
        elim_number = round(int(args.perc_elim)*len(twowers)/100)

    return (path, prompt, scores, votes, twowers, twower_count, top_number, elim_number)