def finish_enroll():
try:
WebDriverWait(I._driver,30).until(EC.presence_of_element_located((By.ID,"MainContent_btnContinue")))
button = I._driver.find_element_by_id("MainContent_btnContinue")
button.click()
except Exception as e:
log("Finish_enroll error --- Error: {}".format(e))
def _get_limits():
try:
WebDriverWait(I._driver, 30).until(EC.presence_of_element_located((By.CLASS_NAME,"enrollment-limit")))
limits = I._driver.find_elements_by_class_name("enrollment-limit")
return [int(i.text) for i in limits]
except Exception as e:
log("get_limits error --- Error: {}".format(e))
return []
def _get_enrolled():
try:
WebDriverWait(I._driver, 30).until(EC.presence_of_element_located((By.CLASS_NAME,"enrolled-currently")))
enrolled = I._driver.find_elements_by_class_name("enrolled-currently")
return [int(e.text) for e in enrolled]
except Exception as e:
log("get_enrolled error --- Error: {}".format(e))
return []
def _get_sections(section):
try:
WebDriverWait(I._driver, 30).until(EC.presence_of_element_located((By.XPATH,'.//td[@class="section-number"]/*[1]')))
sections = I._driver.find_elements_by_xpath('.//td[@class="section-number"]/*[1]')
return [int(s.text) for s in sections]
except Exception as e:
log("get_sections error with passed in section: {} --- Error: {}".format(section, e))
return list()
def _get_planner_links():
try:
WebDriverWait(I._driver, 30).until(EC.presence_of_element_located((By.XPATH,'.//a[contains(@title, "to add") and contains(@title, "to your planner")]')))
links = I._driver.find_elements_by_xpath('.//a[contains(@title, "to add") and contains(@title, "to your planner")]')
return links
except Exception as e:
log("get_planner_link error --- Error: {}".format(e))
return []
def _login(netid, password):
try:
WebDriverWait(I._driver, 30).until(EC.presence_of_element_located((By.ID,"netid")))
netid_input = I._driver.find_element_by_id("netid")
netid_input.send_keys(netid)
pass_input = I._driver.find_element_by_id("pswd")
pass_input.send_keys(password)
pass_input.send_keys(Keys.RETURN)
except Exception as e:
log("_login error with passed in netid: {} , password first letter: {} --- Error: {}".format(netid, password[0], e))
def enroll_in_course(subject, course_number, section):
try:
if section == None:
section = ""
xp = ".//tr[contains(.,'{}') and contains(.,'{}') and contains(number(./td/a/text()),'{}')]/td/a[contains(@title,'Enroll')]".format(subject, course_number, section)
WebDriverWait(I._driver,30).until(EC.presence_of_element_located((By.XPATH,xp)))
button = I._driver.find_element_by_xpath(xp)
button.click()
I.finish_enroll()
except Exception as e:
log("enroll_in_course error with passed in subject: {}, course: {}, section: {} --- Error: {}".format(subject, course_number, section, e))
def check_for_space_in_sections(section):
courses = I._get_course_data(section)
for c in courses:
if section == None or c.section == section:
if c.enrolled < c.limit:
I.found_planner_link = c.link
log("SUCCESS: Found room in section {}.".format(c.section))
message_limit = 1400
if len(I.available_sections) <= message_limit:
I.available_sections+="{} {} (Section: {}) Opening: {}/{} Enrolled\n".format(Course.subject, Course.course_number, c.section, c.enrolled, c.limit)
else:
I.available_sections+="And more...\n"
return
def search_for_course(term, subject, course_number = "*"):
Course.subject = subject
Course.course_number = course_number
try:
WebDriverWait(I._driver, 30).until(EC.presence_of_element_located((By.ID,"MainContent_ddlTerm")))
select = Select(I._driver.find_element_by_id("MainContent_ddlTerm"))
select.select_by_visible_text(term)
except Exception as e:
log("search_for_course Dropdown Select Term error with passed in term: {}, subject: {}, course_number: {} --- Error: {}".format(term, subject, course_number, e))
try:
WebDriverWait(I._driver, 30).until(EC.presence_of_element_located((By.ID,"MainContent_ddlSubject")))
select = Select(I._driver.find_element_by_id("MainContent_ddlSubject"))
select.select_by_value(subject.upper())
except Exception as e:
log("search_for_course Dropdown Select error with passed in term: {}, subject: {}, course_number: {} --- Error: {}".format(term, subject,course_number, e))
return
try:
field = I._driver.find_element_by_id("MainContent_txtCourseNumber")
field.clear()
field.send_keys(course_number)
except Exception as e:
log("search_for_course Field Fill error with passed in subject: {}, course_number: {} --- Error: {}".format(subject,course_number, e))
return
try:
button = I._driver.find_element_by_id("MainContent_btnSubmit")
button.click()
except Exception as e:
log("search_for_course Button Click error with passed in subject: {}, course_number: {} --- Error: {}".format(subject, course_number, e))
return
def _get_course_data(section):
sections = I._get_sections(section)
enrolled = I._get_enrolled()
limits = I._get_limits()
links = I._get_planner_links()
if not(len(sections) == len(enrolled) == len(limits) == len(links)):
log("get_section_data data has uneven lengths. sections: {}, enrolled: {}, limits: {}, links: {}".format(len(sections), len(enrolled), len(limits), len(links)))
return
courses = list()
for i in range(len(sections)):
courses.append(Course(sections[i], enrolled[i], limits[i], links[i]))
if len(courses) == 0:
log("get_section_data no courses found.")
return courses