def get_team_ownerships(self, obj): request = self.context.get('request') teams = [] if obj.user_type == USER_TYPE_ORG and request: teams = obj.team_ownership.all() return TeamMembershipOwnershipListSerializer(teams, many=True, context={ "request": request }).data
def get_team_memberships(self, obj): request = self.context.get('request') teams = [] if obj.user_type == USER_TYPE_ATHLETE and request: teams = obj.athleteuser.team_membership.all() if obj.user_type == USER_TYPE_COACH and request: teams = obj.coachuser.team_membership.all() return TeamMembershipOwnershipListSerializer(teams, many=True, context={ "request": request }).data