grade6.py

####################################
# grade6.py                        #
#                                  #
# Brahm Gardner                    #
# Jason Hirschhorn                 #
# Doug Lloyd                       #
#                                  #
# Last revised: 2016-10-31 18:10   #
####################################

import os, check50

def main():
    
    # list of files this script catches
    recreated_files = ['caesar', 'crack', 'credit', 'greedy', 'mario', 'vigenere']
    new_files = ['smile', 'tweets']
    
    # message
    print("Automatically grading Problem Set 6...")
    
    # set the root to be the current directory
    rootdir = './'

    # get a list of all student directories in the root
    directories = [d for d in os.listdir(rootdir) if os.path.isdir(os.path.join(rootdir, d))]
    
    # create a results directory if you don't already have one
    try:
        os.stat(rootdir + "results")
    except:
        os.mkdir(rootdir + "results") 
        
    # iterate over those directories
    for directory in directories:
        
        # ignore the results directory
        if directory == "results" or directory == ".git" or directory[0] == "_":
            continue
            
        # ignore empty directories
        if not os.listdir(directory):
            continue
            
        # start a file for this student's check50 output
        f = open(rootdir + "results/" + directory + ".txt", "w")
        f.write("+------------------+\n\n")
        
        # breadcrumbs
        print("Now grading student {}...".format(directory))
        
        # run all possible checks on the remakes
        check50.mario("{}{}".format(rootdir, directory), f)
        check50.greedy("{}{}".format(rootdir, directory), f)
        check50.credit("{}{}".format(rootdir, directory), f)
        check50.caesar("{}{}".format(rootdir, directory), f)
        check50.vigenere("{}{}".format(rootdir, directory), f)
        check50.crack(f)
        
        # run checks on smile and tweets
        check50.smile("{}{}".format(rootdir, directory + "/sentiments"), f)
        twitter = {'key': 'TODO', 
                   'secret': 'TODO'}
        check50.tweets("{}{}".format(rootdir, directory + "/sentiments"), f, twitter)
        
        f.close()
        
    print("...grading complete!")
    print("Student reports are now ready in the results/ directory.")

# execute process
if __name__ == "__main__":
    main()