gestioCOF/kfet/migrations/0059_create_generic.py

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

from django.db import migrations, models

from kfet.auth import KFET_GENERIC_TRIGRAMME, KFET_GENERIC_USERNAME


def setup_kfet_generic_user(apps, schema_editor):
    """
    Setup models instances for the kfet generic account.

    Username and trigramme are retrieved from kfet.auth.__init__ module.
    Other data are registered here.

    See also setup_kfet_generic_user from kfet.auth.utils module.
    """
    User = apps.get_model("auth", "User")
    CofProfile = apps.get_model("gestioncof", "CofProfile")
    Account = apps.get_model("kfet", "Account")

    user, _ = User.objects.update_or_create(
        username=KFET_GENERIC_USERNAME,
        defaults={"first_name": "Compte générique K-Fêt"},
    )
    profile, _ = CofProfile.objects.update_or_create(user=user)
    account, _ = Account.objects.update_or_create(
        cofprofile=profile, defaults={"trigramme": KFET_GENERIC_TRIGRAMME}
    )


class Migration(migrations.Migration):

    dependencies = [("kfet", "0058_delete_genericteamtoken")]

    operations = [migrations.RunPython(setup_kfet_generic_user)]