# -*- 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)]