Produits Entr'ouvert » Authentic 2

    use_in_migrations = True

class AttributeValueManager(managers.QueryManager.from_queryset(AttributeValueQuerySet)):

    use_in_migrations = True

from django.db import migrations

import authentic2.managers

class Migration(migrations.Migration):

    dependencies = [

        ('authentic2', '0042_api_client'),

    ]

    operations = [

        migrations.AlterModelManagers(

            name='attribute',

            managers=[

                ('all_objects', authentic2.managers.AttributeManager()),

                ('objects', authentic2.managers.AttributeManager(disabled=False)),

            ],

        ),

        migrations.AlterModelManagers(

            name='attributevalue',

            managers=[

                ('all_objects', authentic2.managers.AttributeValueManager()),

                ('objects', authentic2.managers.AttributeValueManager(attribute__disabled=False)),

            ],

        ),

    ]

import phonenumbers

from django.conf import settings

from django.db import migrations

def migrate_phone_field(apps, schema_editor):

    Attribute = apps.get_model('authentic2', 'Attribute')

    AttributeValue = apps.get_model('authentic2', 'AttributeValue')

    User = apps.get_model(*settings.AUTH_USER_MODEL.split('.', maxsplit=1))

    ContentType = apps.get_model('contenttypes', 'ContentType')

    user_ct = ContentType.objects.get_for_model(User)

    try:

        phone_attr = Attribute.objects.get(name='phone')

    except Attribute.DoesNotExist:

        return

    for atv in AttributeValue.objects.filter(attribute=phone_attr):

        pn = None

        try:

            pn = phonenumbers.parse(atv.content)

        except phonenumbers.NumberParseException:

            try:

                pn = phonenumbers.parse(

                    atv.content,

                    settings.PHONE_COUNTRY_CODES[settings.DEFAULT_COUNTRY_CODE],

                )

            except phonenumbers.NumberParseException:

                continue

        number_e164 = phonenumbers.format_number(pn, phonenumbers.PhoneNumberFormat.E164)

        atv.content = number_e164

        atv.save()

        if not atv.content_type == user_ct:

            continue

        try:

            user = User.objects.get(id=atv.object_id)

        except User.DoesNotExist:

            # stale attribute value, should not happen as it is deleted on user deletion

            continue

        if hasattr(user, 'phone'):

            user.phone = number_e164

            user.save()

class Migration(migrations.Migration):

    dependencies = [

        migrations.swappable_dependency(settings.AUTH_USER_MODEL),

        ('custom_user', '0032_auto_20220919_1230'),

        ('authentic2', '0043_attribute_alter_model_managers'),

    ]

    operations = [

        migrations.RunPython(migrate_phone_field, reverse_code=migrations.RunPython.noop),

    ]

from django.conf import settings

def test_migration_authentic2_0044_migrate_phone_field(transactional_db, migration):

    old_apps = migration.before([('authentic2', '0043_attribute_alter_model_managers')])

    Attribute = old_apps.get_model('authentic2', 'Attribute')

    AttributeValue = old_apps.get_model('authentic2', 'AttributeValue')

    User = old_apps.get_model(*settings.AUTH_USER_MODEL.split('.', maxsplit=1))

    ContentType = old_apps.get_model('contenttypes', 'ContentType')

    at = Attribute.objects.create(

        name='phone',

        label='Phone number',

        kind='phone_number',

    )

    user = User.objects.create(

        email='john.doe@example.com',

        username='john.doe',

        phone='12345',

    )

    user2 = User.objects.create(

        email='billy.smith@example.com',

        username='billy.smith',

        phone='abcdef',  # erroneous phone number

    )

    AttributeValue.objects.create(

        attribute=at,

        content_type=ContentType.objects.get_for_model(User),

        object_id=user.id,

        content='12345',

    )

    AttributeValue.objects.create(

        attribute=at,

        content_type=ContentType.objects.get_for_model(User),

        object_id=user2.id,

        content='abcdef',

    )

    new_apps = migration.apply([('authentic2', '0044_migrate_phone_field')])

    User = new_apps.get_model('custom_user', 'User')

    assert User.objects.get(username='john.doe').phone == '+3312345'

    assert User.objects.get(username='billy.smith').phone == 'abcdef'

    Attribute = new_apps.get_model('authentic2', 'Attribute')

    AttributeValue = new_apps.get_model('authentic2', 'AttributeValue')

    ContentType = new_apps.get_model('contenttypes', 'ContentType')

    at = Attribute.objects.get(name='phone')

    atv = AttributeValue.objects.get(

        attribute=at,

        content_type=ContentType.objects.get_for_model(User),

        object_id=user.id,

    )

    assert atv.content == '+3312345'

    atv2 = AttributeValue.objects.get(

        attribute=at,

        content_type=ContentType.objects.get_for_model(User),

        object_id=user2.id,

    )

    assert atv2.content == 'abcdef'