now_please.py

from selenium import webdriver
from selenium.webdriver import ActionChains
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
import getpass3, sys, socket, os, re
import time

# boolean to indicate the beginning or the end of the class
_sessStart = False 

#open and login to messier
def handle_ruman():
    browser.get('http://ruman.slc/')

    text_password_ruman = browser.find_element_by_name('password')
    text_password_ruman.send_keys(ruman_password + Keys.RETURN)

    #wait for the error message that may be appears
    try:
        try:
            wait = WebDriverWait(browser, 5)
            wait.until(EC.text_to_be_present_in_element((By.ID, 'error-msg'), ' '))
            err = browser.find_element_by_id('error-msg').get_attribute('innerHTML')
            if err != "":
                print "[Ruman.slc]", err, "Exiting..."
                return False
        except Exception as e:
            pass
        browser.find_element_by_css_selector("body").send_keys(Keys.CONTROL, 'a')
        wait = WebDriverWait(browser, 8)
        wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR, '.computer')))

        for act in action_sets:
            btnComp = browser.find_element_by_css_selector(".computer")
            actionChains.move_to_element(btnComp).context_click().perform()
            browser.find_element_by_css_selector(".context-menu-search input").send_keys(act, Keys.DOWN, Keys.ENTER, Keys.ENTER);
            time.sleep(7)
        
    except Exception as e:
        pass

    return True

def handle_messier():
    if _sessStart:
        browser.get('http://messier.slc/')

        text_username = browser.find_element_by_id('user')
        text_username.send_keys(username)
        text_password = browser.find_element_by_id('pass')
        text_password.send_keys(password + Keys.RETURN)

        #wait for the error message that may be appears
        try:
            wait = WebDriverWait(browser, 5)
            wait.until(EC.text_to_be_present_in_element((By.ID, 'msg'), 'not'))
            err = browser.find_element_by_id('msg').get_attribute('innerHTML')
            if err != "":
                print "[Messier]", err + ".", "Exiting..."
                return False
        except:
            pass

        wait = WebDriverWait(browser, 10)
        wait.until(EC.presence_of_element_located((By.ID, 'announcement-container')))

        browser.get('http://messier.slc/Home.aspx#P=Active%20Job')

        wait = WebDriverWait(browser, 30)
        wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'rowHover')))

        task = browser.find_element_by_class_name('Deadline')
        task.click()

        wait = WebDriverWait(browser, 20)
        wait.until(EC.invisibility_of_element_located((By.CSS_SELECTOR, '#loadmsg')))
        # browser.execute_script("document.getElementById('ActiveTeaching_grid_view').style['display'] = 'block';")
        browser.find_element_by_id('view_mode_grid').click()
        browser.find_element_by_id('view_mode_grid').send_keys(Keys.END)

        autoRefresh = browser.find_element_by_id('Auto_Refresh')
        autoRefresh.click()

        return True

if __name__ == '__main__':
    if len(sys.argv) > 1 and sys.argv[1] == '-sq':
        if len(sys.argv) < 3:
            print "Usage: python2 {} -sq [start|end]_[general|quiz|final]_session.txt [--wakeup]".format(sys.argv[0])
            exit()
        try:
            f = open(sys.argv[2], 'r')
            action_sets = [x.strip() for x in f.read().split(',')]
            if len(sys.argv) == 4 and sys.argv[3] == '--wakeup':
                action_sets.insert(0, 'Wake Up')
            if sys.argv[2].split('_')[0] == 'start':
                _sessStart = True
        except Exception as e:
            print e
            exit()

    os.system("cls")

    print "Hi! You're teaching at room", socket.gethostname()[2:5] + "."

    username = raw_input("Username for Messier: ")
    password = getpass3.getpass("Password for Messier: ")
    ruman_password = getpass3.getpass("Password for Ruman.slc: ")

    os.system("cls")

    templates = []
    
    while True:
        choose = raw_input("Choose (1) a template or (2) write your own Ruman sequence(s)? [1..2]:")
        if choose == "1":
            i = 1
            for file in os.listdir(os.getcwd()):
                if file.endswith(".txt"):
                    templates.append(file.split('.')[0])
                    print "{}. {}".format(i, file)
                    i = i + 1
            pick = raw_input("Anything? ")
            f = open(templates[int(pick) - 1] + '.txt', 'r')
            action_sets = [x.strip() for x in f.read().split(',')]
            if templates[int(pick) - 1].split('_')[0] == 'start':
                _sessStart = True
            break
        elif choose == "2":
            sequences = raw_input("Input Ruman sequence(s) (separated with comma (,): ")
            action_sets = [x.strip() for x in sequences.split(',')]
            break

    browser = webdriver.Chrome("chromedriver_win32/chromedriver.exe")
    browser.fullscreen_window()
    actionChains = ActionChains(browser)

    if not handle_ruman() or not handle_messier():
        browser.close()
        exit()

    if not _sessStart:
        log_off = raw_input("Log off this computer? [y|n]:")
        if log_off == "y":
            os.system("shutdown /l /f")


    # handle_messier()
    
    print "Done. Happy Teaching!"

'''
From 18-1 with love <3
Project Handlab by AO18-1
https://github.com/chrisandoryan/Handlab
'''