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