submit.py

import selenium
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time
import sys
from openpyxl import load_workbook
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait
from getpass import getpass

def student_submit(drv, assignment, sid, points, comments):
    global wait
    global DELETE_COMMENT
    drv.get('https://canvas.wisc.edu/courses/65854/gradebook/speed_grader?assignment_id={0}#%7B%22student_id%22%3A%22{1}%22%7D'.format(assignment, sid))
    #print 'Please Confirm Login and Load.'
    #a = input()
    view_rubric = wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR, VIEW_RUBRIC_SELECTOR)))
    pts = drv.find_elements_by_css_selector(POINTS_SELECTOR)
    # view_rubric = drv.find_element_by_css_selector(VIEW_RUBRIC_SELECTOR)
    comment = drv.find_element_by_id(COMMENT_ID)
    submit = drv.find_element_by_id(SUBMIT_ID)
    view_rubric.click()
    save = wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR, SAVE_SELECTOR)))
    i = 0
    for x in points:
        pts[i].clear()
        pts[i].send_keys(str(x))
        i = i + 1
    comment_delete = drv.find_elements_by_css_selector(DELETE_ASSIGNMENT_SELECTOR)
    if DELETE_COMMENT:
        for deletes in comment_delete:
            res = deletes.get_attribute('innerHTML')
            print res
            if comments in res:
                deletes.click()
                wait.until(EC.alert_is_present())
                alert = drv.switch_to.alert
                alert.accept()
        
    if comments != None:
        comment.send_keys(comments)
    save.click()
    submit.click()
    # time.sleep(1)

USER_NAME = 'szou28' # Please change it to your own user name.

print 'USER:',USER_NAME
print 'Please enter the password:'
USER_PWD = getpass()
print 'Saved.'
TESTS_NUM = int(raw_input('How many rubrics do we have (a number for example 8): '))
assignment = int(raw_input('Please enter the assignment ID (for example 199326): '))

if len(sys.argv) <= 1:
    print 'please give me the file name(without .xlsx): '
    fileName = raw_input()
else:
    fileName = sys.argv[1]
print 'You are now submitting assignment id = ' + str(assignment) + ' test rubrics number:' + str(TESTS_NUM) + ' filename:' + fileName + '.xlsx'
y = raw_input('confirm? \nPlease enter \'y\' to confirm your submission. \nDon\'t do anything when submitting: ')
if y != 'y':
    exit()

driver = webdriver.Chrome()
driver.get("http://canvas.wisc.edu")
driver.maximize_window()
print 'Login.'
POINTS_SELECTOR = '.criterion_points.span1.no-margin-bottom'
VIEW_RUBRIC_SELECTOR = '.toggle_full_rubric.edit.btn'
COMMENT_ID = 'speedgrader_comment_textarea'
SUBMIT_ID = 'comment_submit_button'
SAVE_SELECTOR = '.save_rubric_button.Button.Button--primary'
DELETE_ASSIGNMENT_SELECTOR = '.delete_comment_link.icon-x'
first_rubric = 1
DELETE_COMMENT = 1
wait = WebDriverWait(driver, 15)
USER_NAME_ID = 'j_username'
USER_PASSWORD_ID = 'j_password'
SUBMIT_LOGIN_NAME = '_eventId_proceed'

SID_COL = 'B'
COM_COL = 'C'
TEST_COL = 'D'


submit = wait.until(EC.element_to_be_clickable((By.NAME, SUBMIT_LOGIN_NAME)))
username = driver.find_element_by_id(USER_NAME_ID)
pwd = driver.find_element_by_id(USER_PASSWORD_ID)
username.send_keys(USER_NAME)
pwd.send_keys(USER_PWD)
submit.click()


print 'loading files from ' + fileName
wb = load_workbook(fileName + '.xlsx')
worksheet = wb[wb.get_sheet_names()[0]]

i = 2
first = 1
while True:
    sid = worksheet[SID_COL + str(i)].value
    if sid == None:
        break
    points = [worksheet[chr(ord(TEST_COL) + t) + str(i)].value for t in range(TESTS_NUM)]
    comments = worksheet[COM_COL + str(i)].value
    print sid
    print points
    print comments
    i = i + 1
    time.sleep(1)
    student_submit(driver, assignment, str(sid), points, comments)