token api activate

api/views.py

@@ -29,14 +29,14 @@ def NewSnapshot(request):
         return JsonResponse({'error': 'Invalid request method'}, status=400)
 
     # Authentication
-    # auth_header = request.headers.get('Authorization')
+    auth_header = request.headers.get('Authorization')
-    # if not auth_header or not auth_header.startswith('Bearer '):
+    if not auth_header or not auth_header.startswith('Bearer '):
-    #     return JsonResponse({'error': 'Invalid or missing token'}, status=401)
+        return JsonResponse({'error': 'Invalid or missing token'}, status=401)
 
-    # token = auth_header.split(' ')[1]
+    token = auth_header.split(' ')[1]
-    # tk = Token.objects.filter(token=token).first()
+    tk = Token.objects.filter(token=token).first()
-    # if not tk:
+    if not tk:
-    #     return JsonResponse({'error': 'Invalid or missing token'}, status=401)
+        return JsonResponse({'error': 'Invalid or missing token'}, status=401)
 
     # Validation snapshot
     try:

user/management/commands/add_user.py

@@ -1,6 +1,9 @@
+from uuid import uuid4
+
 from django.core.management.base import BaseCommand
 from django.contrib.auth import get_user_model
 from user.models import Institution
+from api.models import  Token
 
 
 User = get_user_model()
@@ -28,3 +31,6 @@ class Command(BaseCommand):
         )
         self.u.set_password(self.password)
         self.u.save()
+        token = uuid4()
+        Token.objects.create(token=token, owner=self.u)
+        print(f"TOKEN: {token}")