moving to docker

This commit is contained in:
2023-05-14 20:28:29 +03:00
parent 6f7219ff7b
commit d5802fa48f
62 changed files with 87 additions and 0 deletions

View File

@ -0,0 +1,55 @@
from rest_framework.authtoken.views import ObtainAuthToken
from rest_framework.authtoken.models import Token
from rest_framework.response import Response
from rest_framework.request import Request
from rest_framework.views import APIView
from django.contrib.auth.models import User
class CustomAuthToken(ObtainAuthToken):
def post(self, request: Request, *args, **kwargs):
serializer = self.serializer_class(data=request.data,
context={'request': request})
serializer.is_valid(raise_exception=True)
user = serializer.validated_data['user']
token, created = Token.objects.get_or_create(user=user)
return Response({
'token': token.key,
'user_id': user.pk,
'username': user.username
})
class RegisterView(APIView):
def get(self, request: Request, *args, **kwargs):
username = request.data.get("username", None)
if not username:
return Response({"message": "Username must not be empty"}, status=406)
if User.objects.filter(username=username).exists():
return Response({"message": "Username already taken"}, status=409)
return Response({"message": "Username availiable"}, status=200)
def post(self, request: Request, *args, **kwargs):
username = request.data.get("username")
if not username:
return Response({"message": "Username must not be empty"}, status=406)
if User.objects.filter(username=username).exists():
return Response({"message": "Username already taken"}, status=409)
password = request.data.get("password", None)
if not password:
return Response({"message": "Password must not be empty"}, status=406)
try:
user = User.objects.create_user(
is_superuser=False,
username=username,
password=password
)
token, created = Token.objects.get_or_create(user=user)
return Response({
'token': token.key,
'user_id': user.pk,
'username': user.username
}, status=200)
except Exception as e:
return Response({"message": "Something went wrong, registration is not completed", "error": e}, status=500)