95 lines
3.6 KiB
Python
95 lines
3.6 KiB
Python
from rest_framework import status
|
|
from rest_framework.viewsets import ViewSet
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
|
|
from django.shortcuts import get_object_or_404
|
|
from django.core.validators import validate_email
|
|
from django.core.exceptions import ValidationError
|
|
|
|
from api.auth.serializers import UserResponseSerializer
|
|
from api.models import UserProfile
|
|
from api.utils.decorators import handle_exceptions
|
|
from routes.models import Route
|
|
from .serializers import RouteSerializer
|
|
|
|
class UserDataView(ViewSet):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def initial(self, request, *args, **kwargs):
|
|
try:
|
|
super().initial(request, *args, **kwargs)
|
|
except Exception as e:
|
|
print(f"Authentication error: {e}")
|
|
raise
|
|
|
|
@action(detail=False, methods=['get'])
|
|
@handle_exceptions
|
|
def user_data(self, request):
|
|
user = request.user
|
|
|
|
try:
|
|
user_data = UserResponseSerializer(user).data
|
|
return Response(user_data, status=status.HTTP_200_OK)
|
|
except UserProfile.DoesNotExist:
|
|
return Response(
|
|
{"error": "User profile not found"},
|
|
status=status.HTTP_404_NOT_FOUND
|
|
)
|
|
|
|
class AccountActionsView(ViewSet):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
@action(detail=False, methods=['patch'])
|
|
@handle_exceptions
|
|
|
|
def change_data_main_tab(self, request):
|
|
"""Обновление данных на главной странице аккаунта"""
|
|
|
|
user = request.user
|
|
user_profile = get_object_or_404(UserProfile, user=user)
|
|
|
|
# обновляем данные пользователя
|
|
if 'firstName' in request.data:
|
|
user.first_name = request.data['firstName']
|
|
if 'lastName' in request.data:
|
|
user.last_name = request.data['lastName']
|
|
if 'email' in request.data:
|
|
email = request.data['email']
|
|
validate_email(email) # handle_exceptions обработает ValidationError
|
|
user.email = email
|
|
user.username = email
|
|
|
|
# обновляем номер телефона
|
|
if 'phone_number' in request.data:
|
|
phone = request.data['phone_number']
|
|
if phone:
|
|
if len(phone) < 13: # +375XXXXXXXXX
|
|
raise ValidationError("Номер телефона слишком короткий")
|
|
|
|
if len(phone) > 18:
|
|
raise ValidationError("Номер телефона слишком длинный")
|
|
|
|
# проверка на уникальность
|
|
if UserProfile.objects.filter(phone_number=phone).exclude(user=user).exists():
|
|
raise ValidationError("Этот номер телефона уже используется")
|
|
|
|
user_profile.phone_number = phone
|
|
|
|
# сохраняем изменения
|
|
user.save()
|
|
user_profile.save()
|
|
|
|
return Response({
|
|
"message": "Данные успешно обновлены",
|
|
"user": UserResponseSerializer(user).data
|
|
}, status=status.HTTP_200_OK)
|
|
|
|
@action(detail=False, methods=['get'])
|
|
@handle_exceptions
|
|
def user_routes(self, request):
|
|
user = request.user
|
|
routes = Route.objects.filter(owner=user)
|
|
return Response(RouteSerializer(routes, many=True).data, status=status.HTTP_200_OK)
|