def test_addRetrieveStudent(self):
name = F.addName("Some", "Student")
name1 = F.addName("Some", "Faculty")
faculty = F.addFaculty(name, ["*****@*****.**"], "Lecturer", "sfac")
student = F.addStudent(name, ["*****@*****.**", "*****@*****.**"], "PhD Student", "stud", [faculty])
self.assert_(student.username == "stud")
self.assert_("*****@*****.**" in student.emails and "*****@*****.**" in student.emails)
self.assert_(student.name.first == "Some")
self.assert_(student.name.last == "Student")
self.assert_(student.name.suffix is None)
self.assert_(student.name.middle is None)
self.assert_(faculty.key() in student.advisors)
key = student.key()
student = db.get(key)
self.assert_(student.username == "stud")
self.assert_("*****@*****.**" in student.emails and "*****@*****.**" in student.emails)
self.assert_(student.name.first == "Some")
self.assert_(student.name.last == "Student")
self.assert_(student.name.suffix is None)
self.assert_(student.name.middle is None)
self.assert_(faculty.key() in student.advisors)
def put_person(person):
"""
pulls xml 'name', 'email', 'username', and 'title' or 'status'
from 'person', makes DB Faculty or Student Object
Keyword arguments:
person -- xml person tag
return reference to DB Faculty or Student object
"""
assert person.tag == 'faculty' or person.tag == 'student'
name = put_name(person.find('name'))
username = None
if person.find('username') is not None:
username = person.find('username').text
emails = person.findall('email')
emails = [email.text for email in emails]
if person.tag == 'faculty':
title = get_text(person.find('title'))
person_handle = Faculty3.addFaculty(name, emails, title,
username)
elif person.tag == 'student':
status = get_text(person.find('status'))
person_handle = Faculty3.addStudent(name, emails, status,
username)
return person_handle
def put_person(person):
"""
pulls xml 'name', 'email', 'username', and 'title' or 'status'
from 'person', makes DB Faculty or Student Object
Keyword arguments:
person -- xml person tag
return reference to DB Faculty or Student object
"""
assert person.tag == 'faculty' or person.tag == 'student'
name = put_name(person.find('name'))
username = None
if person.find('username') is not None:
username = person.find('username').text
emails = person.findall('email')
emails = [email.text for email in emails]
if person.tag == 'faculty':
title = get_text(person.find('title'))
person_handle = Faculty3.addFaculty(name, emails, title, username)
elif person.tag == 'student':
status = get_text(person.find('status'))
person_handle = Faculty3.addStudent(name, emails, status, username)
return person_handle
def put_advisees(person_handle, advisees):
for advisee in advisees:
name = put_name(advisee.find('name'))
email = get_text(advisee.find('email'))
status = get_text(advisee.find('status'))
username = advisee.find('username')
if username is not None:
username = get_text(username)
advisee = Faculty3.addStudent(name, [email], status, username, advisors = person_handle)
def put_advisees(person_handle, advisees):
for advisee in advisees:
name = put_name(advisee.find('name'))
email = get_text(advisee.find('email'))
status = get_text(advisee.find('status'))
username = advisee.find('username')
if username is not None:
username = get_text(username)
advisee = Faculty3.addStudent(name, [email],
status,
username,
advisors=person_handle)
