voteCounter.py 文件源码

def parse_args():
    global should_draw
    parser = argparse.ArgumentParser()
    parser.add_argument('input')
    parser.add_argument("-e", "--perc_elim", nargs='?', type=int, const=-1, default=20, help='Percentage of contestants eliminated, set to negative number to specify number of contestants')
    parser.add_argument("-t", "--num_gold", nargs='?', type=int, const=5, default=1, help='Number of contestants to place in gold highlighting')
    parser.add_argument('-i', '--omit_image', action='store_false', help='Use this flag to not draw image')
    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',encoding=encoding) 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)
    should_draw=args.omit_image
    top_number = args.num_gold #chart coloring ranges
    elim_number=0

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

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