From 5deffec3be96b7d6550db5296b5efd1a6af7999e Mon Sep 17 00:00:00 2001 From: Chase Geigle Date: Tue, 14 Aug 2018 14:10:15 -0500 Subject: [PATCH] fix(dev): repair is_ta boolean in get_user_classes is_ta appears to be no longer sent in the networks list of 'user.status', so instead we use the 'profs_hash' to check for TA status. --- piazza_api/piazza.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/piazza_api/piazza.py b/piazza_api/piazza.py index 3a57dba..2cc092c 100644 --- a/piazza_api/piazza.py +++ b/piazza_api/piazza.py @@ -74,14 +74,16 @@ def get_user_classes(self): # raw_classes = self.get_user_profile().get('all_classes').values() # Get classes from the user status (includes all classes) - raw_classes = self.get_user_status().get('networks', []) + status = self.get_user_status() + uid = status['id'] + raw_classes = status.get('networks', []) classes = [] for rawc in raw_classes: c = {k: rawc[k] for k in ['name', 'term']} c['num'] = rawc.get('course_number', '') c['nid'] = rawc['id'] - c['is_ta'] = rawc.get('is_ta', False) + c['is_ta'] = uid in rawc['prof_hash'] classes.append(c) return classes