From 4c95a0599691867c80f259c8c09954d5f955dae3 Mon Sep 17 00:00:00 2001 From: ciheanyi Date: Thu, 14 Apr 2022 18:50:02 -0400 Subject: [PATCH] Updated mentorship_relation.py functions to use enum class --- app/api/dao/mentorship_relation.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/app/api/dao/mentorship_relation.py b/app/api/dao/mentorship_relation.py index b18c9de1f..5b8fe99b9 100644 --- a/app/api/dao/mentorship_relation.py +++ b/app/api/dao/mentorship_relation.py @@ -136,12 +136,12 @@ def list_mentorship_relations(user_id=None, state=None): message: A message corresponding to the completed action; success if all relationships of a given user are listed, failure if otherwise. """ # To check if the entered 'state' is valid. - valid_states = ["PENDING", "ACCEPTED", "REJECTED", "CANCELLED", "COMPLETED"] - def isValidState(rel_state): - if rel_state in valid_states: - return True - return False + try: + MentorshipRelationState[rel_state] + except KeyError: + return False + return True user = UserModel.find_by_id(user_id) all_relations = user.mentor_relations + user.mentee_relations