generate.py

#!/usr/bin/env python

"""
Generates a csv file named 'data1.csv' with random values to test scoring.
"""

import sys
import csv
import random
from random import randint as rand


def writer(out_file):
    """
    param:
        out_file:
            the output filename for the csv


    """
            
    teams = []
    for number in range(1,2):
        for number in range(1,50):
            teams.append([str(number)])
        for number in range(1,50):
            teams.append([str(number)])
    #print teams
    for item in teams:
        #print item, '\n'
        item.append(rand(1,100))
        for number in range(1,11):
            item.append(rand(1,10))
        item.append(random.choice('yn'))

    #print teams

    data = csv.writer(out_file, delimiter=',')

    for item in teams:
        data.writerow(item)


if __name__ == "__main__":
        out_file = open('data1.csv', 'w')
        writer(out_file)