Add created_by to SessionToken

Author: lnagel

rest_framework_sso/migrations/0004_sessiontoken_created_by.py

@@ -0,0 +1,22 @@
+# Generated by Django 3.2.16 on 2022-12-15 12:46
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('rest_framework_sso', '0003_sessiontoken_client_id'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='sessiontoken',
+            name='created_by',
+            field=models.ForeignKey(null=True, blank=True, on_delete=django.db.models.deletion.CASCADE, related_name='+', to=settings.AUTH_USER_MODEL, verbose_name='created by'),
+        ),
+        migrations.RunSQL("UPDATE rest_framework_sso_sessiontoken SET created_by_id = user_id"),
+    ]

rest_framework_sso/models.py

@@ -32,6 +32,7 @@ class SessionToken(models.Model):
     ip_address = models.GenericIPAddressField(null=True, blank=True, db_index=True)
     user_agent = models.CharField(max_length=1000, blank=True)
     last_used_at = models.DateTimeField(null=True, blank=True, db_index=True)
+    created_by = models.ForeignKey(to=AUTH_USER_MODEL, null=True, blank=True, related_name="+", on_delete=models.CASCADE, verbose_name=_("created by"))
     created_at = models.DateTimeField(auto_now_add=True, db_index=True)
     revoked_at = models.DateTimeField(null=True, blank=True, db_index=True)
 

rest_framework_sso/views.py

@@ -80,7 +80,7 @@ def post(self, request, *args, **kwargs):
             .first()
         )
         if session_token is None:
-            session_token = SessionToken(user=user, client_id=client_id)
+            session_token = SessionToken(user=user, client_id=client_id, created_by=user)
         session_token.update_attributes(request=request)
         session_token.save()
         payload = create_session_payload(session_token=session_token, user=user)
@@ -112,7 +112,7 @@ def post(self, request, *args, **kwargs):
                 SessionToken.objects.active().filter(user=request.user).with_user_agent(request=request).first()
             )
             if session_token is None:
-                session_token = SessionToken(user=request.user)
+                session_token = SessionToken(user=request.user, created_by=request.user)
 
         session_token.update_attributes(request=request)
         session_token.save()