Setup Accounts and add Authentication on this

.gitignore

@@ -1,4 +1,3 @@
 venv
 env
-accounts
 __pycache__

accounts/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

accounts/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class AccountsConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'accounts'

accounts/models.py

@@ -0,0 +1,3 @@
+from django.db import models
+
+# Create your models here.

accounts/serializers.py

@@ -0,0 +1,43 @@
+from django.contrib.auth.models import User
+from rest_framework import serializers
+from django.contrib.auth import authenticate
+
+class UserSerializer(serializers.ModelSerializer):
+    password = serializers.CharField(write_only=True)
+
+    def create(self, validated_data):
+        user = User.objects.create_user(
+            username=validated_data['username'],
+            password=validated_data['password']
+        )
+        return user
+
+    class Meta:
+        model = User
+        fields = ('id', 'username', 'password')
+
+
+class LoginSerializer(serializers.Serializer):
+    username = serializers.CharField(max_length=150)
+    password = serializers.CharField(max_length=128, write_only=True)
+
+    def validate(self, data):
+        username = data.get('username')
+        password = data.get('password')
+        print(username,password)
+
+        if username and password:
+            user = authenticate(username=username, password=password)
+            print("user",user)
+
+            if user:
+                if user.is_active:
+                    data['user'] = user
+                else:
+                    raise serializers.ValidationError("User is not active.")
+            else:
+                raise serializers.ValidationError("Unable to log in with the provided credentials.")
+        else:
+            raise serializers.ValidationError("Must include 'username' and 'password'.")
+
+        return data

accounts/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

accounts/urls.py

@@ -0,0 +1,9 @@
+from django.urls import path
+from . import views
+from rest_framework.authtoken.views import obtain_auth_token
+
+urlpatterns = [
+    path('register/', views.RegisterView.as_view(), name='register'),
+    path('login', obtain_auth_token, name="Get Token"),
+]
+

accounts/views.py

@@ -0,0 +1,16 @@
+from .serializers import LoginSerializer, UserSerializer
+from rest_framework import status
+from rest_framework.response import Response
+from rest_framework.generics import CreateAPIView
+from rest_framework.authtoken.models import Token
+from django.contrib.auth.models import User
+
+class RegisterView(CreateAPIView):
+    queryset = User.objects.all()
+    serializer_class = UserSerializer  # Create a UserSerializer to handle user registration.
+
+    def perform_create(self, serializer):
+        user = serializer.save()
+        token, _ = Token.objects.get_or_create(user=user)
+        return Response({'token': token.key}, status=status.HTTP_201_CREATED)
+

config/settings.py

@@ -10,16 +10,6 @@ For the full list of settings and their values, see
 https://docs.djangoproject.com/en/5.0/ref/settings/
 """
 
-import os
-import pprint
-pprint.pprint(os.environ)
-
-os.environ[ 'NUMBA_CACHE_DIR' ] = '/tmp/'
-
-# from numba.caching import _UserProvidedCacheLocator
-# print("@@@@@@@@@@@@@",_UserProvidedCacheLocator(lambda x:x, 'string').get_cache_path())
-
-
 
 from pathlib import Path
 
@@ -50,7 +40,19 @@ INSTALLED_APPS = [
     'django.contrib.staticfiles',
     'rest_framework',
     'drf_yasg',
-    'texttovoice'
+    'texttovoice',
+    'accounts',
+    'rest_framework.authtoken'
+]
+
+REST_FRAMEWORK = {
+    'DEFAULT_AUTHENTICATION_CLASSES': [
+        'rest_framework.authentication.TokenAuthentication',
+    ],
+}
+
+AUTHENTICATION_BACKENDS = [
+    'django.contrib.auth.backends.ModelBackend',
 ]
 
 X_FRAME_OPTIONS = 'ALLOW-FROM https://huggingface.co/'

config/urls.py

@@ -22,8 +22,9 @@ from texttovoice.views import TextToSpeechCreateView
 
 urlpatterns = [
     path('admin/', admin.site.urls),
+    path('auth/', include('accounts.urls')),
     path('generate-speech/', TextToSpeechCreateView.as_view(), name='generate-speech-create'),
-     path('swagger/', schema_view.with_ui('swagger', cache_timeout=0), name='schema-swagger-ui'),
+    path('swagger/', schema_view.with_ui('swagger', cache_timeout=0), name='schema-swagger-ui'),
     re_path(r'^swagger(?P<format>\.json|\.yaml)$', schema_view.without_ui(cache_timeout=0), name='schema-json'),
 
 ]

texttovoice/views.py

@@ -5,10 +5,14 @@ from rest_framework import status
 from rest_framework.response import Response
 from rest_framework.generics import CreateAPIView
 from TTS.api import TTS
+from rest_framework.authentication import TokenAuthentication
+from rest_framework.permissions import IsAuthenticated
 from .serializers import TextToSpeechSerializer
 
 class TextToSpeechCreateView(CreateAPIView):
     serializer_class = TextToSpeechSerializer
+    authentication_classes = [TokenAuthentication]  # Apply token authentication
+    permission_classes = [IsAuthenticated]  # Require authentication for this view
 
     def create(self, request, *args, **kwargs):
         serializer = self.get_serializer(data=request.data)