authentik.providers.oauth2.migrations.0022_remove_accesstoken_session_id_and_more

  1# Generated by Django 5.0.9 on 2024-10-23 13:38
  2
  3from hashlib import sha256
  4import django.db.models.deletion
  5from django.db import migrations, models
  6from django.apps.registry import Apps
  7from django.db.backends.base.schema import BaseDatabaseSchemaEditor
  8from authentik.lib.migrations import progress_bar
  9
 10
 11def migrate_session(apps: Apps, schema_editor: BaseDatabaseSchemaEditor):
 12    AuthenticatedSession = apps.get_model("authentik_core", "authenticatedsession")
 13    AuthorizationCode = apps.get_model("authentik_providers_oauth2", "authorizationcode")
 14    AccessToken = apps.get_model("authentik_providers_oauth2", "accesstoken")
 15    RefreshToken = apps.get_model("authentik_providers_oauth2", "refreshtoken")
 16    db_alias = schema_editor.connection.alias
 17
 18    print(f"\nFetching session keys, this might take a couple of minutes...")
 19    session_ids = {}
 20    for session in progress_bar(AuthenticatedSession.objects.using(db_alias).all()):
 21        session_ids[sha256(session.session_key.encode("ascii")).hexdigest()] = session.session_key
 22    for model in [AuthorizationCode, AccessToken, RefreshToken]:
 23        print(
 24            f"\nAdding session to {model._meta.verbose_name}, this might take a couple of minutes..."
 25        )
 26        for code in progress_bar(model.objects.using(db_alias).all()):
 27            if code.session_id_old not in session_ids:
 28                continue
 29            code.session = (
 30                AuthenticatedSession.objects.using(db_alias)
 31                .filter(session_key=session_ids[code.session_id_old])
 32                .first()
 33            )
 34            code.save()
 35
 36
 37class Migration(migrations.Migration):
 38
 39    dependencies = [
 40        ("authentik_core", "0039_source_group_matching_mode_alter_group_name_and_more"),
 41        ("authentik_providers_oauth2", "0021_oauth2provider_encryption_key_and_more"),
 42    ]
 43
 44    operations = [
 45        migrations.RenameField(
 46            model_name="accesstoken",
 47            old_name="session_id",
 48            new_name="session_id_old",
 49        ),
 50        migrations.RenameField(
 51            model_name="authorizationcode",
 52            old_name="session_id",
 53            new_name="session_id_old",
 54        ),
 55        migrations.RenameField(
 56            model_name="refreshtoken",
 57            old_name="session_id",
 58            new_name="session_id_old",
 59        ),
 60        migrations.AddField(
 61            model_name="accesstoken",
 62            name="session",
 63            field=models.ForeignKey(
 64                default=None,
 65                null=True,
 66                on_delete=django.db.models.deletion.SET_DEFAULT,
 67                to="authentik_core.authenticatedsession",
 68            ),
 69        ),
 70        migrations.AddField(
 71            model_name="authorizationcode",
 72            name="session",
 73            field=models.ForeignKey(
 74                default=None,
 75                null=True,
 76                on_delete=django.db.models.deletion.SET_DEFAULT,
 77                to="authentik_core.authenticatedsession",
 78            ),
 79        ),
 80        migrations.AddField(
 81            model_name="devicetoken",
 82            name="session",
 83            field=models.ForeignKey(
 84                default=None,
 85                null=True,
 86                on_delete=django.db.models.deletion.SET_DEFAULT,
 87                to="authentik_core.authenticatedsession",
 88            ),
 89        ),
 90        migrations.AddField(
 91            model_name="refreshtoken",
 92            name="session",
 93            field=models.ForeignKey(
 94                default=None,
 95                null=True,
 96                on_delete=django.db.models.deletion.SET_DEFAULT,
 97                to="authentik_core.authenticatedsession",
 98            ),
 99        ),
100        migrations.RunPython(migrate_session),
101        migrations.RemoveField(
102            model_name="accesstoken",
103            name="session_id_old",
104        ),
105        migrations.RemoveField(
106            model_name="authorizationcode",
107            name="session_id_old",
108        ),
109        migrations.RemoveField(
110            model_name="refreshtoken",
111            name="session_id_old",
112        ),
113    ]