authentik/passbook/outposts/migrations/0009_fix_missing_token_identifier.py

# Generated by Django 3.1.2 on 2020-10-17 14:26

from django.apps.registry import Apps
from django.db import migrations
from django.db.backends.base.schema import BaseDatabaseSchemaEditor


def fix_missing_token_identifier(apps: Apps, schema_editor: BaseDatabaseSchemaEditor):
    User = apps.get_model("passbook_core", "User")
    Token = apps.get_model("passbook_core", "Token")
    from passbook.outposts.models import Outpost

    for outpost in (
        Outpost.objects.using(schema_editor.connection.alias).all().only("pk")
    ):
        user_identifier = outpost.user_identifier
        user = User.objects.get(username=user_identifier)
        tokens = Token.objects.filter(user=user)
        for token in tokens:
            if token.identifier != outpost.token_identifier:
                token.identifier = outpost.token_identifier
                token.save()


class Migration(migrations.Migration):

    dependencies = [
        ("passbook_core", "0014_auto_20201018_1158"),
        ("passbook_outposts", "0008_auto_20201014_1547"),
    ]

    operations = [
        migrations.RunPython(fix_missing_token_identifier),
    ]