def testSection(self):
self.sq = SectionQuery()
self.sq.connect()
crn = '31359'
cursor1 = self.sq.get_section_cursor_crn(crn)
cursor2 = self.sq.get_cursor({'crn': crn})
self.assertEqual(cursor1.count(), 1)
self.assertEqual(cursor1.count(), cursor2.count())
self.assertEqual(cursor1[0], cursor2[0])
self.sq.disconnect()
def post(self):
myClass = json.loads(request.data)
print(myClass)
#initiate object to get details of specific classs
query = CourseQuery()
query.connect()
info = query.get_course_JSON(myClass['sub'], myClass['subId'])
del info['_id']
info['class_title'] = info['title']
del info['title']
query.disconnect()
subjectIds = ast.literal_eval(json.dumps(info))
print(subjectIds)
#create object to query the times of the specific course
section = SectionQuery()
sectionList = []
section.connect()
for crn in info['crns']:
cursor = section.get_section_cursor_crn(crn)
courseSect = cursor[0]
del courseSect['_id']
courseSect['class_start'] = courseSect['start']
del courseSect['start']
# courseSection = ast.literal_eval(json.dumps(courseSect))
sectionList.append(courseSect)
section.disconnect()
print(sectionList)
return jsonify({'success':True, 'classInfo': info, 'times':sectionList})
def testSearchTime(self):
self.cq = CourseQuery()
self.sq = SectionQuery()
self.cq.connect()
self.sq.connect()
start_num = 1000
end_num = 1400
list = self.cq.search_for_course_with_time_cursor(subject_code = 'CS', start_time = start_num, end_time = end_num)
for json in list:
boolean = False
start_end_list
for crn in json['crns']:
section_cursor = self.sq.get_section_cursor_crn(crn)
if section_cursor.count() > 0:
section_json = section_cursor[0]
if 'start_num' in section_json and 'end_num' in section_json:
start_end_list.append((section_json['start_num'], section_json['end_num']))
for pair in start_end_list:
if start_num < pair[0] and pair[1] < end_num:
boolean = True
break
self.assertTrue(boolean)
self.cq.disconnect()
self.sq.disconnect()
class Test(unittest.TestCase):
def testProf(self):
self.pq = ProfQuery()
self.pq.connect()
self.assertEqual(self.pq.get_prof_cursor_first_initial('L', 'Angrave').count(), 1)
self.assertEqual(self.pq.get_prof_cursor('Lawrence', 'Angrave')[0], self.pq.get_prof_cursor_first_initial('L', 'Angrave')[0])
self.assertEqual(self.pq.get_prof_cursor('Lawrence', 'Angrave').count(), self.pq.get_prof_cursor_first_initial('L', 'Angrave').count())
self.pq.disconnect()
def testCourse(self):
self.cq = CourseQuery()
self.cq.connect()
subject_code = "CS"
id_num = "125"
cursor1 = self.cq.get_cursor({'code':subject_code, 'course_id':id_num})
cursor2 = self.cq.get_course_cursor(subject_code, id_num)
self.assertEqual(cursor1.count(), 1)
self.assertEqual(cursor1.count(), cursor2.count())
self.assertEqual(cursor1[0], cursor2[0])
cursor1 = self.cq.get_cursor({'code':subject_code})
cursor2 = self.cq.get_courses_cursor_subject(subject_code)
self.assertEqual(cursor1.count(), cursor2.count())
self.assertEqual(cursor1[0], cursor2[0])
self.assertEqual(cursor1[cursor1.count()-1], cursor2[cursor2.count()-1])
self.cq.disconnect()
def testSection(self):
self.sq = SectionQuery()
self.sq.connect()
crn = '31359'
cursor1 = self.sq.get_section_cursor_crn(crn)
cursor2 = self.sq.get_cursor({'crn': crn})
self.assertEqual(cursor1.count(), 1)
self.assertEqual(cursor1.count(), cursor2.count())
self.assertEqual(cursor1[0], cursor2[0])
self.sq.disconnect()
def testSearchGPA(self):
self.cq = CourseQuery()
self.cq.connect()
min_gpa = 3
cursor = self.cq.search_for_course_cursor(min_gpa = min_gpa)
for json in cursor:
self.assertGreaterEqual(json['gpa'], min_gpa)
self.cq.disconnect()
def testSearchProfRating(self):
self.cq = CourseQuery()
self.cq.connect()
prof_rating = 4.5
cursor = self.cq.search_for_course_cursor(min_prof_rating = prof_rating)
for json in cursor:
self.assertGreaterEqual(json['prof_rating'], min_prof_rating)
self.cq.disconnect()
def testSearchCreditHour(self):
self.cq = CourseQuery()
self.cq.connect()
credit_hour = 4
cursor = self.cq.search_for_course_cursor(credit_hours = credit_hour)
for json in cursor:
self.assertIsNotNone(set([credit_hour, str(credit_hour)]).intersection(set(json['credit_hours'])))
self.cq.disconnect()
def testSearchCreditHours(self):
self.cq = CourseQuery()
self.cq.connect()
credit_hours = [4, 6, '4', '6']
credit_hours_set = set(credit_hours)
cursor = self.cq.search_for_course_cursor(credit_hours = credit_hours)
for json in cursor:
self.assertIsNotNone(credit_hours_set.intersection(set(json['credit_hours'])))
self.cq.disconnect()
def testSearchTime(self):
self.cq = CourseQuery()
self.sq = SectionQuery()
self.cq.connect()
self.sq.connect()
start_num = 1000
end_num = 1400
list = self.cq.search_for_course_with_time_cursor(subject_code = 'CS', start_time = start_num, end_time = end_num)
for json in list:
boolean = False
start_end_list
for crn in json['crns']:
section_cursor = self.sq.get_section_cursor_crn(crn)
if section_cursor.count() > 0:
section_json = section_cursor[0]
if 'start_num' in section_json and 'end_num' in section_json:
start_end_list.append((section_json['start_num'], section_json['end_num']))
for pair in start_end_list:
if start_num < pair[0] and pair[1] < end_num:
boolean = True
break
self.assertTrue(boolean)
self.cq.disconnect()
self.sq.disconnect()
from pprint import PrettyPrinter
if __name__ == '__main__':
pp = PrettyPrinter()
client_courses = CourseQuery()
client_courses.connect()
pp.pprint(client_courses.get_course_JSON('CS', '125'))
#client_courses.set_database_name('professors')
#client_courses.set_collection('professors')
#print(client_courses.get_cursor({})[1])
client_courses.disconnect()
section_query = SectionQuery()
section_query.connect()
print(section_query.get_cursor({})[0])
print(section_query.client[section_query.db_name][section_query.collection_name].distinct('days_of_week'))
print('HERE')
section_query.disconnect()
prof_query = ProfQuery()
prof_query.connect()
print(prof_query.get_cursor({})[0])
prof_query.disconnect()
table_query = CourseTableQuery({}, 20)
table_query.connect()
print(table_query.get_table_page_JSON_list(1))
table_query.disconnect()
