# File: courseDataGenerator.py
# Purpose: Randomly chooses effort and value for CS courses at Carleton,
#          writing the output to a CSV file.
#
# Author: Tanya Amert

import random

def generateCourses():
    with open("courseData.csv", 'w') as fout:

        courseNames = ["CS 100", "CS 111", "CS 200", "CS 201",
                       "CS 202", "CS 208", "CS 231", "CS 232",
                       "CS 251", "CS 252", "CS 254", "CS 257",
                       "CS 294", "CS 298", "CS 301", "CS 304",
                       "CS 311", "CS 314", "CS 318", "CS 320",
                       "CS 321", "CS 322", "CS 331", "CS 332",
                       "CS 334", "CS 338", "CS 341", "CS 344",
                       "CS 347", "CS 348", "CS 352", "CS 358",
                       "CS 361", "CS 362", "CS 399", "CS 400"]

        for name in courseNames:
            courseCredits = 6
            effort = random.randint(1,10)
            value = random.randint(1,10)

            if name == "CS 111":
                effort = 2
                value = 10

            if name in ["CS 339", "CS 400"]:
                courseCredits = 3                

            lineEnd = "" if name == courseNames[-1] else "\n"

            print("{0},{1},{2},{3}".format(name, courseCredits, effort, value),
                  file=fout, end=lineEnd)

if __name__ == "__main__":
    generateCourses()