Black

Author: lnagel

rest_framework_sso/migrations/0004_sessiontoken_created_by.py

@@ -9,14 +9,21 @@ class Migration(migrations.Migration):
 
     dependencies = [
         migrations.swappable_dependency(settings.AUTH_USER_MODEL),
-        ('rest_framework_sso', '0003_sessiontoken_client_id'),
+        ("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'),
+            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,7 +32,14 @@ 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_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)
 

tests/settings.py

@@ -4,15 +4,10 @@
 import os
 
 DEBUG_PROPAGATE_EXCEPTIONS = True
-SECRET_KEY = 'drf-sso'
-DATABASES = {
-    'default': {
-        'ENGINE': 'django.db.backends.sqlite3',
-        'NAME': ':memory:'
-    }
-}
+SECRET_KEY = "drf-sso"
+DATABASES = {"default": {"ENGINE": "django.db.backends.sqlite3", "NAME": ":memory:"}}
 REST_FRAMEWORK_SSO = {
-    'KEY_STORE_ROOT': os.path.join(os.path.dirname(__file__), 'keys'),
-    'PRIVATE_KEYS': {'test-issuer': ['test-2048.pem', 'test-1024.pem']},
-    'PUBLIC_KEYS': {'test-issuer': ['test-2048.pem', 'test-1024.pem']},
+    "KEY_STORE_ROOT": os.path.join(os.path.dirname(__file__), "keys"),
+    "PRIVATE_KEYS": {"test-issuer": ["test-2048.pem", "test-1024.pem"]},
+    "PUBLIC_KEYS": {"test-issuer": ["test-2048.pem", "test-1024.pem"]},
 }