changed getting training data

PhilipSnell · Feb 14, 2022 · 3206cc9 · 3206cc9
1 parent 2fcc1d2
commit 3206cc9
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/api/views.py b/api/views.py
@@ -184,15 +184,16 @@ class TrainingData(APIView):
  def post(self, request):
  # print(request.data["username"]+ "- blah blah")
  email_lookup = request.data["username"]
- user = User.objects.filter(email=email_lookup)
- phases = Phase.objects.filter(user=user[0].id)
+ user = User.objects.get(email=email_lookup)
+ phases = Phase.objects.filter(user=user)
  for phase in phases:
  try:
  activeWeek = phase.weeks.get(isActive=True)
+ trainingData = TrainingEntry.objects.filter(
+ user=user, phase=activeWeek.phase, week=activeWeek.week)
  except:
+ trainingData = None
  print('active week not found')
- trainingData = TrainingEntry.objects.filter(
- user=user[0].id, phase=activeWeek.phase, week=activeWeek.week)
 
  serializer = TrainingSerializer(trainingData, many=True)