Merge pull request #7 from LedruRollin/add-authentication

Add authentication

Author: LedruRollin

requirements.txt

@@ -0,0 +1,2 @@
+
+from .token_serializer import AugmentedTokenObtainPairSerializer

tests/__init__.py renamed to src/auth/__init__.py

@@ -0,0 +1,11 @@
+
+from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
+
+
+class AugmentedTokenObtainPairSerializer(TokenObtainPairSerializer):
+
+    @classmethod
+    def get_token(cls, user):
+        token_repr = super().get_token(user)
+        token_repr['username'] = user.username
+        return token_repr

src/auth/serializers/__init__.py

@@ -11,6 +11,7 @@
 """
 
 import os
+from datetime import timedelta
 from pathlib import Path
 
 # Build paths inside the project like this: BASE_DIR / 'subdir'.
@@ -42,9 +43,10 @@
     'django.contrib.messages',
     'django.contrib.staticfiles',
     'rest_framework',
+    'rest_framework_simplejwt',
     'corsheaders',
-    'drf_spectacular',
     'search_targets',
+    'drf_spectacular',
 ]
 
 MIDDLEWARE = [
@@ -139,7 +141,13 @@
     'DEFAULT_PAGINATION_CLASS': 'search_targets.paginations.PaginationWithTotalCount',
     'PAGE_SIZE': 20,
     'DEFAULT_SCHEMA_CLASS': 'drf_spectacular.openapi.AutoSchema',
-    'TEST_REQUEST_DEFAULT_FORMAT': 'json'
+    'TEST_REQUEST_DEFAULT_FORMAT': 'json',
+    'DEFAULT_AUTHENTICATION_CLASSES': ('rest_framework_simplejwt.authentication.JWTAuthentication',),
+}
+
+SIMPLE_JWT = {
+    "ACCESS_TOKEN_LIFETIME": timedelta(days=1),
+    "TOKEN_OBTAIN_SERIALIZER": "auth.serializers.AugmentedTokenObtainPairSerializer",
 }
 
 # Custom media URL

src/auth/serializers/token_serializer.py

@@ -19,6 +19,7 @@
 from django.contrib import admin
 from django.urls import path, include
 from rest_framework import routers
+from rest_framework_simplejwt.views import TokenObtainPairView, TokenRefreshView
 
 from search_targets import views as search_target_views
 from drf_spectacular.views import SpectacularAPIView, SpectacularSwaggerView
@@ -33,6 +34,8 @@
     path('admin/', admin.site.urls),
     path('api-auth/', include('rest_framework.urls')),
     path('api/', include(router.urls)),
+    path('api/token/', TokenObtainPairView.as_view(), name='token_obtain_pair'),
+    path('api/token/refresh/', TokenRefreshView.as_view(), name='token_refresh'),
     path('api/schema/', SpectacularAPIView.as_view(), name='schema'),
     path('api/schema/swagger-ui/', SpectacularSwaggerView.as_view(url_name='schema'), name='swagger-ui'),
 ]

src/roback/settings.py

@@ -7,7 +7,7 @@ class SearchTargetAdmin(admin.ModelAdmin):
     list_display = ("search_text", "insertion_date", "insertion_time")
 
 class MediaAdmin(admin.ModelAdmin):
-    list_display = ("search_target", "type", "file", "name")
+    list_display = ("search_target", "type", "file_path", "name")
 
 class URLAdmin(admin.ModelAdmin):
     list_display = ("url",)

src/roback/urls.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.1 on 2024-10-10 13:04
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('search_targets', '0018_alter_searchtarget_insertion_time'),
+    ]
+
+    operations = [
+        migrations.RenameField(
+            model_name='media',
+            old_name='file',
+            new_name='file_path',
+        ),
+    ]

src/search_targets/admin.py

@@ -36,5 +36,5 @@ class Media(Model):
     """ Model for a media (image, video, gif, etc)"""
     search_target = ForeignKey(SearchTarget, on_delete=CASCADE, related_name="media")
     name = TextField()
-    file = FileField(upload_to=get_media_path)
+    file_path = FileField(upload_to=get_media_path)
     type = TextField(choices=MediaType.choices, default='')

src/search_targets/migrations/0019_rename_file_media_file_path.py

@@ -1,10 +1,16 @@
 
-from rest_framework.serializers import ModelSerializer
+from rest_framework.serializers import ModelSerializer, FilePathField, SerializerMethodField
 
 from search_targets.models import Media
 
 
 class MediaSerializer(ModelSerializer):
     class Meta:
         model = Media
-        fields = '__all__'
+        fields = (
+            "id",
+            "name",
+            "file_path",
+            "type",
+            "search_target",
+        )