135 lines
4.3 KiB
Python
135 lines
4.3 KiB
Python
from rest_framework.decorators import api_view, action
|
|
from rest_framework.response import Response
|
|
from rest_framework import status, views, viewsets
|
|
from django.contrib.auth import authenticate
|
|
from django.shortcuts import redirect
|
|
from django.contrib.auth import logout
|
|
from rest_framework_simplejwt.tokens import RefreshToken
|
|
from rest_framework.permissions import IsAuthenticated, IsAuthenticatedOrReadOnly
|
|
|
|
from system.accounts.models import User
|
|
from system.accounts.serializers import UserSerializer, ProfileSerializer
|
|
|
|
|
|
@api_view(['POST', ])
|
|
def account_registration(request):
|
|
try:
|
|
user_data = request.data.get('user')
|
|
|
|
serializer = UserSerializer(data=user_data)
|
|
serializer.is_valid(raise_exception=True)
|
|
serializer.save()
|
|
return Response({"user": serializer.data}, status=status.HTTP_201_CREATED)
|
|
|
|
except Exception:
|
|
return Response(status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
@api_view(['POST', ])
|
|
def account_login(request):
|
|
try:
|
|
user_data = request.data.get('user')
|
|
user = authenticate(email=user_data['email'], password=user_data['password'])
|
|
serializer = UserSerializer(user)
|
|
jwt_token = RefreshToken.for_user(user)
|
|
serializer_data = serializer.data
|
|
serializer_data['token'] = str(jwt_token.access_token)
|
|
response_data = {
|
|
"user": serializer_data,
|
|
}
|
|
return Response(response_data, status=status.HTTP_202_ACCEPTED)
|
|
|
|
except Exception:
|
|
return Response(status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
class UserView(views.APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get(self, request, format=None):
|
|
user = self.request.user
|
|
serializer = UserSerializer(user)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
def put(self, request, format=None, pk=None):
|
|
user = self.request.user
|
|
user_data = request.data.get('user')
|
|
|
|
user.email = user_data['email']
|
|
user.bio = user_data['bio']
|
|
user.image = user_data['image']
|
|
user.save()
|
|
|
|
serializer = UserSerializer(user)
|
|
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
|
|
class ProfileDetailView(viewsets.ModelViewSet):
|
|
queryset = User.objects.all()
|
|
serializer_class = ProfileSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
lookup_field = 'username'
|
|
http_method_names = ['get', 'post', 'delete']
|
|
|
|
def get_permissions(self):
|
|
if self.action == 'list':
|
|
return [IsAuthenticatedOrReadOnly(), ]
|
|
return super().get_permissions()
|
|
|
|
def list(self, request, username=None, *args, **kwargs):
|
|
try:
|
|
profile = User.objects.get(username=username)
|
|
serializer = self.get_serializer(profile)
|
|
return Response({"profile": serializer.data})
|
|
|
|
except Exception:
|
|
return Response({"errors": {
|
|
"body": [
|
|
"Invalid User"
|
|
]
|
|
}})
|
|
|
|
@action(detail=True, methods=['post', 'delete'])
|
|
def follow(self, request, username=None, *args, **kwargs):
|
|
if request.method == 'POST':
|
|
|
|
profile = self.get_object()
|
|
follower = request.user
|
|
if profile == follower:
|
|
return Response({"errors": {
|
|
"body": [
|
|
"Invalid follow Request"
|
|
]
|
|
}}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
profile.followers.add(follower)
|
|
serializer = self.get_serializer(profile)
|
|
return Response({"profile": serializer.data})
|
|
|
|
elif request.method == 'DELETE':
|
|
|
|
profile = self.get_object()
|
|
follower = request.user
|
|
if profile == follower:
|
|
return Response({"errors": {
|
|
"body": [
|
|
"Invalid follow Request"
|
|
]
|
|
}}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
if not profile.followers.filter(pk=follower.id).exists():
|
|
return Response({"errors": {
|
|
"body": [
|
|
"Invalid follow Request"
|
|
]
|
|
}}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
profile.followers.remove(follower)
|
|
serializer = self.get_serializer(profile)
|
|
return Response({"profile": serializer.data})
|
|
|
|
|
|
def signout(request):
|
|
logout(request)
|
|
return redirect("main") |