main.python

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time
import SSCMail

server = SSCMail.initializeEmail()

browser = webdriver.PhantomJS()
# action_chains = ActionChains(browser)
browser.implicitly_wait(1)
browser.get('https://ssc.adm.ubc.ca/sscportal/servlets/SRVAcademicRecord?context=html')

username = browser.find_element_by_id("username")
password = browser.find_element_by_id("password")

username.send_keys('totallylegitusername')
password.send_keys('hahahahahano', Keys.ENTER)

time.sleep(2)


classesToCheck = ['APSC101', 'PHYS158', 'PHYS170','MATH101']
length = 28
while(len(classesToCheck)>0):
    browser.refresh()
    time.sleep(2)
    _2016W = browser.find_elements_by_xpath("//a[@id='ui-id-3']")

    # Gotta reaaaally make sure we clicked it
    if(len(_2016W)):
        _2016W[0].click()
        _2016W[0].click()
        _2016W[0].click()
        _2016W[0].click()
        _2016W[0].click()
    else:
        usernameL = browser.find_elements_by_id("username")
        passwordL = browser.find_elements_by_id("password")
        if len(usernameL) and len(passwordL):
            usernameL[0].send_keys('cjasper')
            passwordL[0].send_keys('A\/\/esomesauce515', Keys.ENTER)

            time.sleep(2)
        else:
            print("Element not found, is SSC down?")
        continue

    newCourse = 0
    row = 6
    courseList = []
    marksList = []
    averageList = []
    while(row < 12):
        out = browser.find_element_by_xpath("//tr[@id='row-2016W-" + str(row) + "']").text
        split = out.split()

        courseName = split[1] + split[2]
        mark = split[5]
        if(len(split)>=9):
            average = split[8]
        else:
            average = '0'


        if(mark=='0'):
            print("Marks are not out for " + courseName)
        else:
            if(courseName in classesToCheck):
                newCourse = 1
                classesToCheck.remove(courseName)
                courseName = courseName + '!'
                print("New mark out!")

            print("You got " + mark + " for " + courseName)

        courseList.append(courseName)
        marksList.append(mark)
        averageList.append(average)

        row += 1

    if(newCourse == 1):
        newCourse = 0
        message = "Course Mark Average"
        for course, mark, average in zip(courseList,marksList,averageList):
            message = message + "\n" + course + " " + mark + " " + average

        SSCMail.sendEmail(server, message)

browser.save_screenshot('final.png')