1c61b80731
- Add rules_engine.py with quantitative rules for all 8 workout types - Add quality gate retry loop in generate_single_workout() - Expand calibrate_structure_rules to all 120 combinations (8 types × 5 goals × 3 sections) - Wire WeeklySplitPattern DB records into _pick_weekly_split() - Enforce movement patterns from WorkoutStructureRule in exercise selection - Add straight-set strength support (single main lift, 4-6 rounds) - Add modality consistency check for duration-dominant workout types - Add InjuryStep component to onboarding and preferences - Add sibling exercise exclusion in regenerate and preview_day endpoints - Display generator warnings on dashboard - Expand fix_rep_durations, fix_exercise_flags, fix_movement_pattern_typo - Add audit_exercise_data and check_rules_drift management commands - Add Next.js frontend with dashboard, onboarding, preferences, history pages - Add generator app with ML-powered workout generation pipeline - 96 new tests across 7 test modules Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
132 lines
5.0 KiB
Python
132 lines
5.0 KiB
Python
from .models import RegisteredUser
|
|
from rest_framework.decorators import api_view
|
|
from rest_framework.response import Response
|
|
from rest_framework import status
|
|
from .serializers import RegisteredUserSerializer, CreateRegisteredUserThroughUserSerializer, CreateRegisteredUserSerializer
|
|
from django.contrib.auth.models import User
|
|
from rest_framework.authtoken.models import Token
|
|
from django.contrib.auth import authenticate
|
|
from rest_framework.authentication import TokenAuthentication
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.decorators import authentication_classes
|
|
from rest_framework.decorators import permission_classes
|
|
from django.shortcuts import get_object_or_404
|
|
import json
|
|
|
|
|
|
@api_view(['GET'])
|
|
def all_registered_users(request):
|
|
users = RegisteredUser.objects.all()
|
|
serializer = RegisteredUserSerializer(users, many=True)
|
|
return Response(data=serializer.data, status=status.HTTP_200_OK)
|
|
|
|
|
|
@api_view(['POST'])
|
|
def create_registered_user(request):
|
|
_serializer = CreateRegisteredUserSerializer(data=request.data)
|
|
|
|
if not _serializer.is_valid():
|
|
return Response(_serializer.errors, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
|
|
|
|
email = request.data["email"]
|
|
|
|
if User.objects.filter(email=email):
|
|
return Response({"email": [ "Email in use" ] }, status=status.HTTP_409_CONFLICT)
|
|
|
|
serializer = CreateRegisteredUserThroughUserSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
new_registered_user = serializer.save()
|
|
serializer = RegisteredUserSerializer(new_registered_user, many=False)
|
|
|
|
token = Token.objects.get(user=new_registered_user.user).key
|
|
data = serializer.data
|
|
data["token"] = token
|
|
return Response(data,status=status.HTTP_201_CREATED)
|
|
|
|
return Response(serializer.errors, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
|
|
|
|
|
|
@api_view(['POST'])
|
|
def login_registered_user(request):
|
|
email = request.data.get("email", "").strip()
|
|
password = request.data.get("password", "")
|
|
|
|
# Try authenticating with the input as username first, then by email lookup
|
|
user = authenticate(username=email, password=password)
|
|
if user is None:
|
|
from django.contrib.auth.models import User
|
|
try:
|
|
user_obj = User.objects.get(email=email)
|
|
user = authenticate(username=user_obj.username, password=password)
|
|
except User.DoesNotExist:
|
|
pass
|
|
|
|
if user is not None:
|
|
registered_user = get_object_or_404(RegisteredUser, user=user)
|
|
serializer = RegisteredUserSerializer(registered_user, many=False)
|
|
token = get_object_or_404(Token, user=registered_user.user).key
|
|
data = serializer.data
|
|
data["token"] = token
|
|
return Response(data,status=status.HTTP_200_OK)
|
|
else:
|
|
return Response({"detail": "Invalid email or password"}, status=status.HTTP_404_NOT_FOUND)
|
|
|
|
|
|
@api_view(['POST'])
|
|
@authentication_classes([TokenAuthentication])
|
|
@permission_classes([IsAuthenticated])
|
|
def update_registered_user(request):
|
|
registered_user = RegisteredUser.objects.get(user=request.user)
|
|
|
|
email = request.POST.get("email")
|
|
first_name = request.POST.get("first_name")
|
|
last_name = request.POST.get("last_name")
|
|
image = request.POST.get("image")
|
|
|
|
registered_user.first_name = first_name
|
|
registered_user.last_name = last_name
|
|
registered_user.user.email = email
|
|
registered_user.image = image
|
|
|
|
registered_user.save()
|
|
registered_user.user.save()
|
|
|
|
registered_user = RegisteredUser.objects.get(user=request.user)
|
|
serializer = RegisteredUserSerializer(registered_user, many=False)
|
|
token = Token.objects.get(user=registered_user.user).key
|
|
data = serializer.data
|
|
data["token"] = token
|
|
return Response(data,status=status.HTTP_200_OK)
|
|
|
|
|
|
@api_view(['POST'])
|
|
@authentication_classes([TokenAuthentication])
|
|
@permission_classes([IsAuthenticated])
|
|
def update_password(request):
|
|
current_password = request.POST.get("current_password")
|
|
new_password = request.POST.get("new_password")
|
|
user = request.user
|
|
|
|
success = user.check_password(current_password)
|
|
if success:
|
|
user.set_password(new_password)
|
|
user.save()
|
|
registered_user = RegisteredUser.objects.get(user=request.user)
|
|
serializer = RegisteredUserSerializer(registered_user, many=False)
|
|
token = Token.objects.get(user=registered_user.user).key
|
|
data = serializer.data
|
|
data["token"] = token
|
|
return Response(data,status=status.HTTP_200_OK)
|
|
else:
|
|
return Response(status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
@api_view(['GET'])
|
|
@authentication_classes([TokenAuthentication])
|
|
@permission_classes([IsAuthenticated])
|
|
def refresh(request):
|
|
registered_user = RegisteredUser.objects.get(user=request.user)
|
|
serializer = RegisteredUserSerializer(registered_user, many=False)
|
|
token = get_object_or_404(Token, user=registered_user.user).key
|
|
data = serializer.data
|
|
data["token"] = token
|
|
return Response(data,status=status.HTTP_200_OK) |